Python基础学习：Python连接mysql方法封装

Python基础学习：Python连接mysql普适性方法封装

# 需要下载mysql-connector，mysql-connector是MySQL官方提供的驱动器
# 下载命令：pip install mysql-connector
from mysql import connector


class MysqlConnectDatabase(object):
    # 连接数据库
    def __init__(self, host, port, user, password, db, charset='utf8'):
        self.__host = host
        self.__port = int(port)  # mysql端口
        self.__username = user  # mysql远程连接用户名
        self.__password = password  # mysql远程连接密码
        self.__db = db  # mysql使用的数据库名
        self.__charset = charset  # mysql使用的字符编码,默认为utf8
        try:
            self.__connect_database = connector.connect(host=self.__host, port=self.__port,
                                                        user=self.__username, password=self.__password,
                                                        db=self.__db, charset=self.__charset)
        except:
            print('连接失败')

    # 操作数据库（增删改查等）
    def operate_data(self, sql):
        try:
            cur = self.__connect_database.cursor()
            cur.execute(sql)
            try:
                self.__connect_database.commit() # 增删改
                result1 = ''
                cur.close()
                result2 = 'sql语句："%s"执行成功' % sql
            except Exception:
                try:
                    result1 = cur.fetchall() # 数据库查询返回结果
                    cur.close()
                    result2 = 'sql语句："%s"执行成功' % sql
                except Exception as e:
                    print(e)
                    result1 = ''
                    result2 = 'sql语句："%s"执行失败' % sql
                    self.__connect_database.rollback()
        except Exception as e:
            print('失败', e)
            result1 = ''
            result2 = 'sql语句："%s"执行失败' % sql
            self.__connect_database.rollback()
        if result1 == '':
            return result2
        else:
            return result1, result2

    # 断开数据库连接
    def close_database(self):
        self.__connect_database.close()