[大数据]Python操作mysql

使用mysql官方python驱动

mysql驱动官网下载
安然安装即可

注意

pycharm 创建项目勾选上从全局加载 inherit global site-packages

使用模块

import mysql.connector

创建数据库连接

    # connect(host,port,user,password,database)
    # host 数据库地址
    # port 数据库端口
    # user 用户名
    # password 密码
    # database 库名
    # autocommit 是否自动提交 默认False
    # autocommit 为False时，insert,update,delete等修改操作需要通过con.commit() 才能写入数据库
    config = {
        "host": "xxxx",
        "port": "3306",
        "user": "xxxx",
        "password": "xxxx",
        "database": "sun_riches_test",
        "autocommit": True
    }

    con = mysql.connector.connect(**config)

创建游标

# 数据库操作要在游标上进行
cursor = con.cursor()

执行sql语句

    # 调用游标的execute(sql,params) 执行sql语句 返回的结果存储在游标上
    # params参数填写代表使用预编译,可以防止sql注入&提升使用同一条语句(结构相同，参数不同)多次执行的速度
    # params不为空，要求sql中的参数需要使用占位符%s替代且个数同params中元素个数相同
    # 如：cursor.execute("select name,nickname from admin_users limit %s", (2,))
    # 等价于 cursor.execute("select name,nickname from admin_users limit 2"）

    # 一次性执行多条语句可以使用 cursor.executemany(sql,params)
    # 如：cursor.executemany("update admin_users set name = %s where id = %s", [['陈浩南2', 14], ['张三丰2', 15]])

    cursor.execute("select name,nickname from admin_users limit 5")

    # cursor.fetchone() 获取一条返回结果 返回类型tuple
    # 如:('超级管理员', '')

    # cursor.fetchall() 获取所有返回结果 返回类型是list(tuple)
    # 如:[('超级管理员', ''), ('测试账号', 'test'), ('陈楠', 'THINKMAN'), ('李昕妍', 'Lijie520'), ('许钊', 'gavin')]
    result = cursor.fetchall()

关闭连接

con.close()

事务

    # 使用 start_transaction() 开启事务
    # commit() 提交事务
    # rollback() 回滚事务
    try:
        con.start_transaction()
        cursor.execute("update admin_users set name = 'zsf2' where id = 14")
        con.commit()
    except Exception as e:
        print(e)
        if "con" in dir():
            con.rollback()
    finally:
        if "con" in dir():
            con.close()

注意事项

cursor 中的结果要全部取尽才能进行下一个
cursor.execute()