python连接mysql数据库封装

# 导入相应的库
import MySQLdb

# 定义封装的数据库连接类
class MySQLDB:
    def __init__(self, host, port, user, password, database):
        self.host = host
        self.port = port
        self.user = user
        self.password = password
        self.database = database
        self.connection = None

    def connect(self):
        try:
            self.connection = MySQLdb.connect(host=self.host, port=self.port, user=self.user,
                                              password=self.password, database=self.database)
            print("成功连接到MySQL数据库")
        except MySQLdb.Error as e:
            print("数据库连接失败:", str(e))

    def disconnect(self):
        if self.connection:
            self.connection.close()
            print("成功断开与MySQL数据库的连接")

    def execute_query(self, query):
        result = None
        try:
            cursor = self.connection.cursor()
            cursor.execute(query)
            result = cursor.fetchall()
        except MySQLdb.Error as e:
            print("查询失败:", str(e))
        return result

# 使用示例
if __name__ == '__main__':
    # 创建数据库连接对象
    db = MySQLDB(host='localhost', port=3306, user='root', password='password', database='test')

    # 连接数据库
    db.connect()

    # 执行查询
    result = db.execute_query('SELECT * FROM table_name')

    # 打印查询结果
    if result:
        for row in result:
            print(row)

    # 断开数据库连接
    db.disconnect()