Python 封裝一個操作mysql的類

import pymysql

class MysqlOperation():

    def __init__(self, host, user, pwd, port, db):
        self.host = host
        self.user = user
        self.password = pwd
        self.port = port
        self.dbname = db
        self.conn = None  # 連接
        self.cur = None  # 游標

    def open(self):
        # 創建連接
        self.conn = pymysql.connect(host=self.host, user=self.user, password=self.password, port=self.port,
                                    db=self.dbname, charset='utf8')  # 創建連接
        self.cur = self.conn.cursor()  # 創建游標

    def select(self, sql):
        '''查詢數據'''
        self.cur.execute(sql)  # 查詢數據
        return self.cur.fetchall()  # 獲取結果

    def execute(self, sql):
        '''執行sql'''
        try:
            # 執行SQL語句
            self.cur.execute(sql)
            # 提交事務到數據庫執行
            self.conn.commit()  # 事務是訪問和更新數據庫的一個程序執行單元

        except BaseException as  f:
            print(f)
            self.conn.rollback()

        # 返回受影響行數
        return self.cur.rowcount

    def executemany(self, sql, params):
        '''
        批量插入數據
        :param sql:    插入數據模版, 需要指定列和可替換字符串個數
        :param params:  插入所需數據，列表嵌套元組[(1, '張三', '男'),(2, '李四', '女'),]
        :return:    影響行數
        '''
        try:
            # sql = "INSERT INTO USER VALUES (%s,%s,%s,%s)"  # insert 模版
            # params = [(2, 'fighter01', 'admin', 'sanpang'),
            #           (3, 'fighter02', 'admin', 'sanpang')]  # insert數據，
            self.cur.executemany(sql, params)

        except BaseException as  f:
            print(f)
            self.conn.rollback()

        return self.cur.rowcount

    def __enter__(self):
        self.open()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        '''退出時關閉游標關閉連接'''
        self.cur.close()
        self.conn.close()


obj = MysqlOperation(host='127.0.0.1', user='root', pwd='123456', port=3306, db='wlj_db')  # 對象
with obj as obj:
    result = obj.executemany('''INSERT INTO student(gender, class_id, 分) VALUES ('男', '2', '王五');''')
    print(result)