存测试数据python
有的时候大批量的数据,咱们须要存到数据库中,在测试的时候才能用到,测试的时候就从数据库中读取出来。这点是很是重要的!mysql
存测试结果sql
python2中使用mysqldb,python3中再也不支持,换成pyMysql数据库
1) 安装pyMysql模块:pip install PyMysql函数
2) python链接mysql数据库,须要下载驱动:https://dev.mysql.com/downloads/connector/python/mysqlmysql官方网站提供的驱动版本,目前只支持到python3.4,python版本要与驱动支持的版本匹配才能安装成功测试
3) 安装mysql图形界面Navicat,链接数据库后,方便查看/编辑数据库数据fetch
(可选)安装mysql服务端:https://dev.mysql.com/downloads/mysql/网站
增删改查,因此咱们要学的也是利用python去数据库里面读取数据,进行增删改查操做编码
对数据库进行操做步骤:spa
1. 链接数据库、建立游标
2. 准备好增、删、改、查sql语句
3. 执行sql语句
4. 读取执行结果—执行影响的行
5. 关闭游标、关闭链接(释放链接数)
数据库链接信息以下:
IP地址、端口号、数据库名字、登陆用户名、密码
pymysql中的链接数据库方法:
conn = pymysql.Connect(host, port, db, user, passwd)
cursorclass = pymysql.cursors.DictCursor 指定返回数据为字典形式
建立游标:
cursor = conn.cursor() 每次操做都须要获取游标,才能进行
sql语句:select 字段 from 表名 where 条件语句
fetchall():返回的数据格式是列表形式的
fetchone():返回的数据格式是元组形式的(能够用cursorclass = pymysql.cursors.DictCursor改为字典的形式)
重点看代码,掌握用法
execute中,sql语句的多种方式:
execute(sql语句、参数[可选]):
1. 数据直接写在sql语句中
2. 用格式化的方式
sql语句中:列表和元组 %s代替
字典:%(键名)s代替
参数:为真正的数值
例:
insert_sql = "insert into test1(id, name) values(%s, %s)"
data = [14, 'xiaojian']
#字典
data1 = {'id': 14, 'name': 'xiaojian'}
insert_sql = "insert into test1(id, name) values(%(id)s, %(name)s)"
cursor.execute(insert_sql, data)
提交操做:在对数据库进行任何修改的状况下,都使用commit函数来提交操做
修改包括增长、修改、删除数据等操做
提交修改:
conn.commit()
回滚操做:在对数据库进行修改的过程当中,因任何异常而致使修改中断,咱们都应该使用回滚操做使数据库恢复到修改以前的状态
回滚修改:
conn.rollback()
#引入相关的库 import pymysql
mysql_host = "localhost"
mysql_port = 3306
mysql_db = "xiaozhai"
mysql_user = "root"
mysql_passwd = "123456"
#链接操做:编码格式的指定,默认返回数据类型的指定
conn = pymysql.Connect(host=mysql_host, user=mysql_user, password=mysql_passwd,
db=mysql_db, port=mysql_port, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor)
#获取游标 cur = conn.cursor()
#sql语句——增长数据
sql_insert = "insert into python6(name, sex) values('xiaohua', 'female')"
#执行sql语句
try:
cur.execute(sql_insert)
conn.commit()
except:
conn.rollback()
#关闭链接、关闭游标
cur.close()
conn.close()
运行结果:
#引入相关的库 import pymysql mysql_host = "localhost" mysql_port = 3306 mysql_db = "xiaozhai" mysql_user = "root" mysql_passwd = "123456" #链接操做:编码格式的指定,默认返回数据类型的指定 conn = pymysql.Connect(host=mysql_host, user=mysql_user, password=mysql_passwd, db=mysql_db, port=mysql_port, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor) #获取游标 cur = conn.cursor() #查询语句 sql_select = "select * from python6" #执行 cur.execute(sql_select) #获取查询结果——会获取一条数据 data_a = cur.fetchone() print(data_a) print("=============================") #获取查询结果——获取全部条数据(游标已经到了第二条,从第二条开始读) data_all = cur.fetchall() print(data_all) #关闭链接、关闭游标 cur.close() conn.close()
运行结果
{'id': 1, 'sex': 'male', 'name': 'xiaozhai'} ============================= [{'id': 2, 'sex': 'male', 'name': 'xiaoli'}, {'id': 3, 'sex': 'female', 'name': 'xiaohua'}]
列表形式
#引入相关的库 import pymysql mysql_host = "localhost" mysql_port = 3306 mysql_db = "xiaozhai" mysql_user = "root" mysql_passwd = "123456" #链接操做:编码格式的指定,默认返回数据类型的指定 conn = pymysql.Connect(host=mysql_host, user=mysql_user, password=mysql_passwd, db=mysql_db, port=mysql_port, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor) #获取游标 cur = conn.cursor() #格式化方式插入数据 data_list = ["xiaozhao", "female"] sql_insert = "insert into python6(name, sex) values(%s, %s)" #执行sql语句 try: cur.execute(sql_insert, data_list) conn.commit() except: conn.rollback() #关闭链接、关闭游标 cur.close() conn.close()
运行结果
字典形式
#引入相关的库 import pymysql mysql_host = "localhost" mysql_port = 3306 mysql_db = "xiaozhai" mysql_user = "root" mysql_passwd = "123456" #链接操做:编码格式的指定,默认返回数据类型的指定 conn = pymysql.Connect(host=mysql_host, user=mysql_user, password=mysql_passwd, db=mysql_db, port=mysql_port, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor) #获取游标 cur = conn.cursor() #格式化方式插入数据 data_dict = {"name": "xiaoliu", "sex":"female"} sql_insert = "insert into python6(name, sex) values (%(name)s, %(sex)s)" #执行sql语句 try: cur.execute(sql_insert, data_dict) conn.commit() except: conn.rollback() #关闭链接、关闭游标 cur.close() conn.close()
运行结果
import pymysql class Mysql_Operate: def __init__(self, host, db, user, passwd, port=3306): #链接数据库 try: self.conn = pymysql.Connect(host=host, user=user, password=passwd, db=db, port=port, charset="utf8mb4", cursorclass=pymysql.cursors.DictCursor) #获取游标 self.cur = self.conn.cursor() #没有异常,connect_flag为0 self.connect_flag = 0 except Exception as e: print(e) self.connect_flag = 1 #查找 def select_all_datas(self, select_sql): #查询数据——execute函数 self.cur.execute(select_sql) #获取全部的数据并返回 data_all = self.cur.fetchall() return data_all #更新数据 def update_datas(self, update_sql ): try: self.cur.execute(update_sql) self.conn.commit() return True except: self.conn.rollback() return False #关闭数据库链接 def close_db(self): self.cur.close() self.conn.close() mysql_host = "localhost" mysql_port = 3306 mysql_db = "xiaozhai" mysql_user = "root" mysql_passwd = "123456" ms = Mysql_Operate(mysql_host, mysql_user, mysql_passwd, mysql_db) if ms.connect_flag == 0: pass