自动化（一）使用Python+pandas在数据库取值

如何使用Python在数据库取值

在编写自动化UI脚本亦或是接口脚本时，或多或少都会涉及到数据库取值，下面给大家介绍一个比较好用的数据库取值方式：Python + pandas 。

简单介绍Pandas

pandas是一个强大的分析结构化数据的工具集；它的使用基础是Numpy（提供高性能的矩阵运算）；用于数据挖掘和数据分析，同时也提供数据清洗功能。
话不多说，pandas具体使用规则自行搜索，我们现在开始pyhton代码取值

一、安装pandas

#可以通过CMD进行安装pandas
c:\user\admin>pip install pandas
#验证是否安装成功
c:\user\admin>python>>>import pandas

#如果使用CMD安装报错，我们也可以尝试使用Pycharm进行便捷安装

二、安装sqlalchemy

**#如果直接sqlalchemy的安装，会报错的，因为sqlalchemy依赖了PyMySQL 包 **
#需要先安装PyMySQL
使用指令：pip install PyMySQL

#接下来再安装sqlalchemy便会成功的
使用指令：pip instal sqlalchemy

三、Python代码取值应用

import pandas as pd
from sqlalchemy import create_engine
import pymysql
#数据库链接：
engine = create_engine(‘mysql+pymysql://用户名:数据库密码@主机名:3306/数据库表名?charset=utf8’)
#从数据库取值（此处举例为：从验证码库获取这个手机号申请的所有验证码，并倒序排列）
phone = 110
captcha= pd.read_sql(‘SELECT * FROM t_captcha where phone="’+ str(i)+’" order by postTime desc’,engine)
#验证码提取
captcha = captcha.iloc[0, 0]

至此，我们就通过Python + pandas 完成了从数据库的取值操作，就可以愉快的去填写一些进输入款或者赋值给接口参数，从而完成我们的工作啦~

四、Tips:封装查询

如果你的数据库取值操作很频繁，建议还是将取值方法封装起来吧，这样不但可以大大减少代码量，还方便日常维护
封装示例：

import pandas as pd
from sqlalchemy import create_engine
import pymysql
engine = create_engine('mysql+pymysql://用户名:数据库密码@主机名:3306/数据库表名?charset=utf8')
def getCaptcha(phone):
    captchaList= pd.read_sql('SELECT * FROM t_captcha where phone="' + str(phone) + '" order by 	postTime desc', engine)
    Captcha= captchaList.iloc[0, 1]
    return lastCaptcha