封装pymysql

Posted 2021-01-04 zengsf

from pymysql import connect

class MysqlHelp:
    def __init__(self,database,host = "localhost", user = "zengsf", 
            password = "密码", charset = "utf8", port = 3306):
        self.database = database
        self.host = host
        self.user = user
        self.password = password
        self.charset = charset
        self.port = port

    #创建对象
    def open(self):
        #创建conn
        self.conn = connect(host = self.host, user = self.user,
                            password = self.password, database = self.database,
                            charset = self.charset, port = self.port)
        #创建游标
        self.cur = self.conn.cursor()

    #关闭
    def close(self):
        self.cur.close()
        self.conn.close()

    #执行SQL语句
    def workOn(self, sql, L=[]):
        self.open()
        try:
            self.cur.execute(sql, L)
            self.conn.commit()
        except Exception as e:
            self.conn.rollback()
        self.close()

    #getAll查询方法
    def getAll(self, sql, L=[]):
        self.open()
        self.cur.execute(sql, L)
        result = self.cur.fetchall()
        self.close()
        return result

if __name__ == "__main__":
    #测试 
    mysql = MysqlHelp("shengfen")
#    sql_insert = "insert into sheng(s_name) values(‘美国‘);"
#    mysql.workOn(sql_insert)
    sql_select = "select * from sheng;"
    result = mysql.getAll(sql_select)    
    print(result)