python接口自动化第四篇--python连接数据库（mysql）

对于其它方式的数据库，大同小异，这里以mysql为例子

#连接数据库，这里展示的是pymysql
#安装：pip install pymysql
import pymysql
import DictCursor

#建立连接
from pymysql.cursors import DictCursor

conn = pymysql.connect(host = ''#域名
                       port = #端口号
                       user = ''#数据库登陆的用户名
                       password = ''#数据库登陆的用户名
                       charset = 'utf8'#不能是utf-8
                       database = ''#你要连接的那个数据库
                       cursorclass = DictCursor#r为了查询结果以字典的形式查询出来，如果不穿入这个参数，查询结果是以元祖的形式查询出来
                       )

#游标：采取行动,每一次操作
cursor = conn.cursor()

#发起请求，查询数据
#sql = 'select*from where id= {}'.format(1,2,3,4)#这样的形式不推荐用，因为不安全
sql = 'select*from where id=%s'
cursor.execute(sql,args=[1,])#看源码的参数传入,可以是列表，也可是元组

#获取数据,获取一条记录
res = cursor.fetchone()
print(res)
#获取数据,获取所有记录
res1 = cursor.fetchall()
print(res1)

#断开连接和关闭sql
cursor.close()
conn.close()

 

总结tips:

1、这里的信息需要去和研发或者其他人员获取哈

切记：每一次查询都要断开连接和关闭，不然如果数据过多会把库弄坏