Example #1
0
class HandleMysql:
    def __init__(self):
        self.data = ReadConfig()

    def conn_mysql(self):
        """连接数据库"""
        host = self.data.get_db("mysql", "host")
        user = self.data.get_db("mysql", "user")
        password = self.data.get_db("mysql", "passwd")
        db = self.data.get_db("mysql", "db")
        charset = self.data.get_db("mysql", "charset")
        self.conn = pymysql.connect(host=host,
                                    user=user,
                                    password=password,
                                    db=db,
                                    charset=charset)
        self.cur = self.conn.cursor()

    def execute_sql(self, sql, data):
        """执行操作数据的相关sql"""
        self.conn_mysql()
        self.cur.execute(sql, data)
        self.conn.commit()

    def search(self, sql):
        """执行查询sql"""
        self.conn_mysql()
        self.cur.execute(sql)
        result = self.cur.fetchall()
        self.conn.commit()
        return result

    def dispose(self):
        self.cur.close()
        self.conn.close()