mysql数据库操作的封装

浅浅的花香味﹌ 2022-09-03 15:13 186阅读 0赞

import pymysql
    class Mysql:
        # 初始化
        def __init__(self,ip='localhost',port=3306,user='root',passwd='123456',db='test',charset='utf8'):
            # 创建连接
            self.db=pymysql.connect(host=ip,port=port,user=user,passwd=passwd,database=db,charset=charset)
            # 创建游标
            self.cursor=self.db.cursor()
    
        # 查询操作
        def select(self,sql,match='all'):
            self.cursor.execute(sql)
            if match=='all':
                result=self.cursor.fetchall()
            elif match=='one':
                result=self.cursor.fetchone()
            else:
                result='请输入正确的匹配规则'
            return result
    
        # 修改操作
        def update(self,sql):
            try:
                self.cursor.execute(sql)
            except Exception as e:
                # 回滚
                self.db.rollback()
                print(f'mysqldb error:{ e}')
            # 提交
            self.db.commit()
    
        # 插入操作
        def insert(self,sql):
            try:
                self.cursor.execute(sql)
            except Exception as e:
                self.db.rollback()
                print(f'mysqldb error:{ e}')
            self.db.commit()
    
        # 删除操作
        def delete(self,sql):
            try:
                self.cursor.execute(sql)
            except Exception as e:
                self.db.rollback()
                print(f'mysqldb error:{ e}')
            self.db.commit()
    
        def exit(self):
            # 关闭游标
            self.cursor.close()
            # 关闭数据库连接
            self.db.close()
    if __name__ == '__main__':
        db=Mysql()
        res=db.select('select * from course;',match='one')
        print(res)
        db.exit()