python接口自动化测试二十六：使用pymysql模块链接数据库

Posted 2020-11-14 向前走。

 #!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/5/28 18:51
# @Author  : StalloneYang
# @File    : mysql_test.py
# @desc:


# 连接数据库

import pymysql.cursors


# 连接MySQL数据库
connection = pymysql.connect(host=\'localhost\', port=3306, user=\'yang\', password=\'\', db=\'test_yang\', charset=\'utf8mb4\', cursorclass=pymysql.cursors.DictCursor)

# 通过cursor创建游标
cursor = connection.cursor()

# 执行数据查询
sql = "SELECT `id`, `name`,\'address\' FROM `tb_emp` WHERE NAME = \'杨大大\'"
cursor.execute(sql)

#查询数据库单条数据
result = cursor.fetchone()
print(result)
print(type(result))

print("-----------华丽分割线------------")

# 执行数据查询
sql = "SELECT `id`, `name`,\'address\' FROM `tb_emp`"
cursor.execute(sql)

#查询数据库多条数据
result = cursor.fetchall()
for data in result:
    print(data)

# 关闭数据连接
connection.close()