实现一个MYSQL工具类,实现动态切换数据源,可以创建数据表,可以删除数据表,可以获取数据表数据量...

这是一种可以实现动态切换数据源、创建数据表、删除数据表、获取数据表数据量的 MySQL 工具类的实现方式:

import mysql.connector

class MySQLTool:
    def __init__(self, host, user, password, database=None):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.conn = None
    
    def connect(self):
        """连接数据库"""
        self.conn = mysql.connector.connect(
            host=self.host,
            user=self.user,
            password=self.password,
            database=self.database
        )
    
    def close(self):
        """关闭数据库连接"""
        self.conn.close()
    
    def change_database(self, database):
        """切换数据库"""
        self.database = database
        self.connect()
    
    def create_table(self, table_name, columns):
        """创建数据表"""
        cursor = self.conn.cursor()
        sql = f"CREATE TABLE {table_name} ({', '.join(columns)})"
        cursor.execute(sql)
        cursor.close()
    
    def delete_table(self, table_name):
        """删除数据表"""
        cursor = self.conn.cursor()
        sql = f"DROP TABLE {table_name}"
        cursor.execute(sql)
        cursor.close()
    
    def get_table_count(self, table_name):
        """获取数据表数据量"""
        cursor = self.conn.cursor()
        sql = f"SELECT COUNT(*) FROM {table_name}"
        cursor.execute(sql)
        count = cursor.fetchone()[0]
        cursor.close()
        return count
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值