python接口自动化连接数据库
时间: 2023-12-16 21:28:32 浏览: 29
在Python中进行接口自动化测试时,连接数据库是非常常见的需求。可以使用第三方库pymysql来连接MySQL数据库并进行操作。
首先,需要安装pymysql库。可以使用以下命令来安装:
```shell
pip install pymysql
```
接下来,可以使用以下代码示例来连接数据库并执行相关操作:
```python
import pymysql
# 连接数据库
conn = pymysql.connect(host='localhost', port=3306, user='root', password='password', database='test')
# 创建游标对象
cursor = conn.cursor()
# 执行SQL语句
sql = "SELECT * FROM users"
cursor.execute(sql)
# 获取查询结果
result = cursor.fetchall()
# 打印查询结果
for row in result:
print(row)
# 关闭游标和连接
cursor.close()
conn.close()
```
上述代码中,首先使用`pymysql.connect()`方法连接数据库,需要提供数据库的主机名、端口号、用户名、密码和数据库名等信息。然后,使用`conn.cursor()`方法创建游标对象,通过游标对象可以执行SQL语句。接着,使用`cursor.execute()`方法执行SQL语句,可以是查询语句或者更新语句等。最后,使用`cursor.fetchall()`方法获取查询结果,并通过循环打印结果。最后,使用`cursor.close()`和`conn.close()`方法关闭游标和连接。
需要注意的是,连接数据库的参数需要根据实际情况进行修改,包括主机名、端口号、用户名、密码和数据库名等。