Python3之数据库编程
使用pymysql模块,主要用于Python3.x 连接 Mysql 数据库
请使用pip3 install pymysql安装模块
一、数据库连接对象
pymysql.connect(host='', user='', password='', database='', port=0, db='', charset='') 连接数据库 db连接.begin 开启事务 db连接.close() 关闭连接 db连接.commit() 提交事务到数据库,不提交事务所有操作都不会生效 db连接.cursor() 得到一个可以执行SQL语句并且将结果作为字典返回的游标对象 db连接.rollback() 发生错误时回滚事务 db连接.select_db(db) 选择数据库 db连接.show_warnings() 向mysql发送 "SHOW WARNINGS" 命令 db连接.ping(reconnect=True) 检查连接是否存活,连接不存在会重新发起连接
二、游标对象
cursor.close()
关闭游标
cursor.execute(SQL, args)
执行单条SQL命令,返回受影响的行数
args: 元组,列表,字典类型,为SQL命令传入的参数
executemany(SQL, args)
执行多条SQL语句,返回受影响的行数
args: 为SQL命令传入的参数,二维多元元组或列表,多条SQL语句参数组成的元组或列表,每条SQL语句的参数也是一个元组或列表类型
cursor.fetchone()
返回最近一条SQL查询语句结果的下一行数据组成的列表或元组
cursor.fetchmany(size)
返回最近一条SQL查询语句结果的下size行数据组成的二维多元列表或元组,每行数据也是一个元组或列表
cursor.fetchall()
返回最近一条SQL查询语句结果剩下的所有行数据组成的二维多元列表或元组,每行数据也是一个元组或列表
cursor.rowcount
返回最近一条SQL语句(包括增删改查)影响的总行数
三、示例
1 try...except...finally... 操作数据库
import pymysql #连接数据库 db = pymysql.connect(host='localhost', port=3306, user='root', password='Admin!123', database='test_1', charset='utf8') try: cur = db.cursor() #查询当前数据库的版本 cur.execute('select version();') print('Data version:',cur.fetchone()) #sql语句 sql_drop_table= 'drop table if exists user_info' sql_create_table = 'create table if not exists user_info(\ id int not null auto_increment,\ username varchar(100) not null,\ age int not null,\ password varchar(256) not null,\ sex varchar(10) not null,\ primary key(id))\ engine=Innodb DEFAULT CHARSET=utf8;' sql_insert = 'insert into user_info values(%s, %s, %s, %s, %s);' sql_query = 'select * from user_info where age=%s or sex=%s;' insert_arg = [1, 'Charles', 25, 'Charles!123', 'male'] insert_args = ((2, 'Mark', 24, 'Mark!123', 'male'), (3, 'Linda', 23, 'Linda!123', 'female'), (4, 'Lily', 23, 'Lily!123', 'female'), (5, 'Helen', 23, 'Helen!123', 'female'), (6, 'Fred', 25, 'Fred!123', 'male'), (7, 'Diana', 24, 'Diana!123', 'female')) query_arg = (23, 'female') #执行sql语句 cur.execute(sql_drop_table) cur.execute(sql_create_table) cur.execute(sql_insert, insert_arg) cur.executemany(sql_insert, insert_args) cur.execute(sql_query, query_arg) #返回查询结果的下一行数据 print(cur.fetchone()) #返回查询结果的下两行数据(除去上一行数据) print(cur.fetchmany(2)) #返回查询结果剩余的数据 print(cur.fetchall()) #返回查询结果数据总行数 print(cur.rowcount) #提交 db.commit() except Exception as e: print(e.args) db.rollback() finally: #关闭游标 cur.close() #关闭数据库连接 db.close()
2 with上下文管理
自动回滚事务的上下文管理器：发生异常时，事务将自动回滚；否则，需要手动提交，和关闭mysql 连接对象
import pymysql #连接数据库 db = pymysql.connect(host='localhost', port=3306, user='root', password='Admin!123', database='test_1', charset='utf8') with db.cursor() as cur: #查询当前数据库的版本 cur.execute('select version();') print('Data version:',cur.fetchone()) #sql语句 sql_drop_table= 'drop table if exists user_info' sql_create_table = 'create table if not exists user_info(\ id int not null auto_increment,\ username varchar(100) not null,\ age int not null,\ password varchar(256) not null,\ sex varchar(10) not null,\ primary key(id))\ engine=Innodb DEFAULT CHARSET=utf8;' sql_insert = 'insert into user_info values(%s, %s, %s, %s, %s);' sql_query = 'select * from user_info where age=%s or sex=%s;' insert_arg = [1, 'Charles', 25, 'Charles!123', 'male'] insert_args = ((2, 'Mark', 24, 'Mark!123', 'male'), (3, 'Linda', 23, 'Linda!123', 'female'), (4, 'Lily', 23, 'Lily!123', 'female'), (5, 'Helen', 23, 'Helen!123', 'female'), (6, 'Fred', 25, 'Fred!123', 'male'), (7, 'Diana', 24, 'Diana!123', 'female')) query_arg = (23, 'female') #执行sql语句 cur.execute(sql_drop_table) cur.execute(sql_create_table) cur.execute(sql_insert, insert_arg) cur.executemany(sql_insert, insert_args) cur.execute(sql_query, query_arg) print(cur.fetchone()) print(cur.fetchmany(2)) print(cur.fetchall()) print(cur.rowcount) #提交 db.commit() #关闭数据库连接 db.close()
