python自动化:mysql的各种操作

安装第三方库

pip install mysqlclient

导入第三方库

import MySQLdb

连接数据库

def connectSQL(server_host, server_port, username, passwd, tableName, charset='utf8'):
    """
    连接数据库
    :param server_host: 数据库ip
    :param server_port: 数据库端口，默认3306
    :param username: 数据库用户名
    :param passwd: 数据库密码
    :param tableName: 数据库
    :param charset: 数据库编码
    :return: 连接对象
    """
    try:
        db = MySQLdb.Connect(host=server_host, port=server_port, user=username, password=passwd, db=tableName,
                             charset=charset)
        return db
    except Exception as e:
        return e

执行查询操作

def querySQL(db, sql):
    """
    数据库查询操作
    :param db: 数据库连接对象
    :param sql: sql命令
    :return: 查询结果
    """
    try:
        cursor = db.cursor()  # 创建数据库游标
        cursor.execute(sql)  # 执行sql命令
        result = cursor.fetchall()  # 接收查询到的全部的返回结果
        cursor.close()  # 关闭数据库游标
        return result  # 返回查询结果
    except Exception as e:
        return e

执行插入，删除，更新操作

def operationSQL(db, sql):
    """
    操作数据库
    :param db: 数据库连接
    :param sql: sql命令。可执行插入，删除，更新操作
    :return: 运行结果
    """
    try:
        cursor = db.cursor()  # 创建数据库游标
        cursor.execute(sql)  # 执行sql命令
        db.commit()  # 提交操作
        cursor.close()  # 关闭数据库游标
    except Exception as e:
    	db.rollback()  # 发生错误则回滚
        return e

关闭数据库连接

def closeSQL(db):
    """
    关闭数据库连接
    :param db: 数据库连接
    :return: 运行结果
    """
    try:
        db.close()
    except Exception as e:
        return e