【Python学习】操作Sqllite

#!/usr/bin/env python
# encoding: utf-8
from ea.tests.common import *
from mgr.common.logger import *
from mgr.common.cmd import exec_cmd
import os
import sys
import re
import time
import sqlite3


class agent_db_manage:
    def __init__(self):
        self.conn = None
        self.cursor = None
        self.encode_db_path = r"C:ProgramDataSangforEDRPDFLinksfedomain.db"
        self.LIST_PATH = os.path.dirname(os.path.realpath(__file__))
        workspace = r"C:Jenkinsworkspace"
        for floor in os.listdir(workspace):
            if os.path.exists(os.path.join(workspace, floor, "Swiss-Knife", "tools", "DomainTool")):
                self.workspace = os.path.join(workspace, floor)
                break
        self.toolspace = os.path.join(self.workspace, "Swiss-Knife", "tools")
        self.uncode_path = os.path.join(self.toolspace, "uncodeTool")

    def refesh_db(self, encode_db_path):
        """
        再解密数据库并读取
        encode_db_path : agent上加密的数据库位置
        :return:
        """

        ret = exec_cmd(r"cd %s && uncode.bat %s" % (self.uncode_path, encode_db_path))
        time.sleep(2)
        if self.cursor:
            self.cursor.close()
        if self.conn:
            self.conn.close()

        self.conn = sqlite3.connect(os.path.join(self.uncode_path, "plaintext.db"))
        self.cursor = self.conn.cursor()

    def close(self):
        """
        关闭数据库连接
        :return:
        """
        if self.cursor:
            self.cursor.close()
        if self.conn:
            self.conn.close()

    def search_by_sql(self, sql):
        """
        通过sql语句进行查询
        :param sql:
        :return:
        """
        try:
            self.cursor.execute(sql)
        except Exception:
            self.refesh_db(self.encode_db_path)
            self.cursor.execute(sql)
        return self.cursor.fetchall()


if __name__ == "__main__":
    am = agent_db_manage()
    db_path = r"C:ProgramDataSangforEDRPDFLinksfedomain.db"
    sql = r"select count(*) from process"

    am.refesh_db(db_path)
    ret = am.search_by_sql(sql)
    print ret
    am.close()