实现一个MYSQL工具类，包含判断创建数据表是否存在，创建数据表，删除数据表，获取数据表元数据

比眉伴天荒 2024-03-27 12:19 16阅读 0赞

这是一个示例的 MySQL 工具类的实现：

import mysql.connector
    
    class MySQLUtil:
        def __init__(self, host, user, password, database):
            self.host = host
            self.user = user
            self.password = password
            self.database = database
            self.conn = mysql.connector.connect(
                host=self.host,
                user=self.user,
                password=self.password,
                database=self.database
            )
            self.cursor = self.conn.cursor()
        
        def __enter__(self):
            return self
    
        def __exit__(self, exc_type, exc_val, exc_tb):
            self.conn.close()
    
        def check_table_exists(self, table_name):
            """
            判断数据表是否存在
            """
            sql = "SHOW TABLES"
            self.cursor.execute(sql)
            tables = [self.cursor.fetchall()]
            if (table_name,) in tables:
                return True
            return False
        
        def create_table(self, table_name, columns):
            """
            创建数据表
            """
            column_str = ", ".join([f"{k} {v}" for k, v in columns.items()])
            sql = f"CREATE TABLE {table_name} ({column_str})"
            self.cursor.execute(sql)
            self.conn.commit()
        
        def delete_table(self, table_name):
            """
            删除数据表
            """
            sql = f"DROP TABLE {table_name}"
            self.cursor.execute(sql)
            self.conn.commit()
        
        def get_table_metadata(self, table_name):
            """
            获取数据表元数据
            """
            sql = f"DESCRIBE {table_name}"
            self.cursor.execute(sql)
            metadata = self.cursor.fetchall()
            return metadata

该工具类使用了 Python 的 mysql.connector 库来连接 MySQL 数据库。在初始化时，需要提供 MySQL 服务器的地址、用户名、密码和数据库名称。

该工具类包含四个方法：

*  `check_table_exists`：判断数据表是否存在。
 *  `create_table`：创建数据表。
 *  `delete_table`：