封装Python连接MySQL数据库为工具类

import pymysql


class DBUtil:
    __conn = None
    __cursor = None

    @classmethod
    def __get_conn(cls):  # 获取连接
        if cls.__conn == None:
            cls.__conn = pymysql.connect(host="localhost",
                                         port=3306,
                                         user="root",
                                         password="123456",
                                         database="books")
            return cls.__conn

    @classmethod
    def __get_cursor(cls):  # 获取游标
        if cls.__cursor == None:
            cls.__cursor = cls.__get_conn().cursor()
            return cls.__cursor

    @classmethod
    def __close_cursor(cls):  # 关闭游标
        if cls.__cursor != None:
            cls.__cursor.close()
            cls.__cursor = None

    @classmethod
    def __close_conn(cls):  # 关闭连接
        if cls.__conn != None:
            cls.__conn.close()
            cls.__conn = None

    @classmethod
    def mysql(cls, sql_str):  # 程序入口
        try:
            cursor = cls.__get_cursor()
            cursor.execute(sql_str)
            str = sql_str.split(" ")[0]
            if str == "select":
                return cursor.fetchall()
            else:
                cls.__conn.commit()
                return cursor.rowcount
        except Exception as e:
            cls.__conn.rollback()
            print(e)
        finally:
            cls.__close_cursor()
            cls.__close_conn()

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Python数据封装是将数据库操作相关的代码封装一个模块或类,使得代码结构更加清晰,易于维护和重用。常用的Python数据库封装模块有: 1. PyMySQL一个Python实现的MySQL客户端库,支持Python 2和Python 3。 2. MySQLdb:Python MySQL数据库接口,是Python一个第三方模块,用于连接MySQL数据库。 3. SQLAlchemy:一个Python SQL工具和对象关系映射器,支持多种数据库,包括MySQL、PostgreSQL、SQLite、Oracle等。 4. Django ORM:一个Python Web框架,自带ORM(对象关系映射器),支持多种数据库,包括MySQL、PostgreSQL、SQLite等。 封装后的代码示例: ``` import pymysql class DB: def __init__(self, host, user, password, database): self.conn = pymysql.connect( host=host, user=user, password=password, database=database, charset='utf8' ) def query(self, sql): cursor = self.conn.cursor() cursor.execute(sql) result = cursor.fetchall() cursor.close() return result def insert(self, table, data): cursor = self.conn.cursor() keys = ','.join(data.keys()) values = ','.join(['%s'] * len(data)) sql = 'INSERT INTO %s (%s) VALUES (%s)' % (table, keys, values) cursor.execute(sql, tuple(data.values())) self.conn.commit() cursor.close() def update(self, table, data, condition): cursor = self.conn.cursor() values = ','.join(["%s='%s'" % (k, v) for k, v in data.items()]) sql = 'UPDATE %s SET %s WHERE %s' % (table, values, condition) cursor.execute(sql) self.conn.commit() cursor.close() def delete(self, table, condition): cursor = self.conn.cursor() sql = 'DELETE FROM %s WHERE %s' % (table, condition) cursor.execute(sql) self.conn.commit() cursor.close() def close(self): self.conn.close() ``` 以上代码是一个封装MySQL数据库操作的类,包括查询、插入、更新和删除等常用操作。使用时只需实例化DB类,并调用对应的方法即可完成数据库操作。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值