Python--操作数据库class

Posted 2020-10-24 王思磊

import pymysql


class OpMySql1:  # 经典类
    pass


class OpMySql(object):  # 新式类
    def __init__(self, host, user, password, db, port=3306, charset=‘utf8‘):
        schema = {
            ‘user‘: user,
            ‘host‘: host,
            ‘password‘: password,
            ‘db‘: db,
            ‘port‘: port,
            ‘charset‘: charset
        }
        try:
            self.coon = pymysql.connect(**schema)
        except Exception as e:
            print(‘数据库连接异常！%s‘ % e)
            quit()
        else:  # 没有异常的情况下，建立游标
            self.cur = self.coon.cursor(cursor=pymysql.cursors.DictCursor)

    def execute(self, sql):
        try:
            self.cur.execute(sql)
        except Exception as e:
            print(‘sql语句有错误！%s‘ % e)
            return e
        if sql[:6].upper() == ‘SELECT‘:
            return self.cur.fetchall()
        else:  # 其他sql语句的话
            self.coon.commit()
            return ‘ok‘

    def __del__(self):
        self.cur.close()
        self.coon.close()


ybq = OpMySql(‘211.149.218.16‘, ‘jxz‘, ‘123456‘, db=‘jxz‘)  # 实例化
print(ybq.execute(‘select * from stu;‘))