python操作mysql数据库之pymysql
摘要:
我们经常需要将大量数据保存起来以备后续使用,数据库是一个很好的解决方案。在众多数据库中,MySQL数据库算是入门比较简单、语法比较简单,同时也比较实用的一个。
1. pymysql.connect()连接数据库函数
连接数据库是进行数据库操作的第一步
import pymysql
# 打开数据库连接
conn = pymysql.connect(host='119.23.55.123',port=3309,user = "root",passwd = "Luck123!",db = "learn")
print(conn) #输出连接结果,用来测试连接是否成功
cursor = conn.cursor() #获取游标
cursor.execute("SELECT VERSION()") # 使用 execute() 方法执行 SQL 查询
data = cursor.fetchone() # 使用 fetchone() 方法获取单条数据
print ("Database version : %s " % data)
conn.close() # 关闭数据库连接
输出结果:
<pymysql.connections.Connection object at 0x0050B2B0>
Database version : 5.7.29
2. 创建表操作
使用cursor.execute(sql),执行已编辑好的SQL即可
import pymysql
# 打开数据库连接
conn = pymysql.connect(host='119.23.55.123',port=3309,user = "root",passwd = "Luck123!",db = "qianli")
cursor = conn.cursor() #创建一个游标对象 cursor
# 使用 execute() 方法执行 SQL,如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS emp")
# 使用预处理语句创建表
sql = """CREATE TABLE emp (
ename CHAR(20) NOT NULL,
age INT,
sex CHAR(1),
income FLOAT )"""
#执行sql语句
cursor.execute(sql)
# 关闭数据库连接
conn.close()
3. 操作数据
操作数据库分为插入数据,更新数据和删除数据,其操作相似。
import pymysql
# 打开数据库连接
conn = pymysql.connect(host='119.23.55.123',port=3309,user = "root",passwd = "Luck123!",db = "qianli")
# 使用cursor()方法获取操作游标
cursor = conn.cursor()
# SQL 插入语句
sql = """INSERT INTO emp(ename,age, sex, income VALUES ('tester', 20, 'M', 2000)"""
try:
cursor.execute(sql) # 执行sql语句
conn.commit() # 提交到数据库执行
except:
conn.rollback() # 如果发生错误则回滚
# 关闭数据库连接
conn.close()
更新操作:用以下语句替换sql变量即可
sql = "UPDATE emp SET age = age + 1 WHERE sex = '%c'" % ('M')
删除操作:用以下语句替换sql变量即可
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
4. 查询操作
Python查询Mysql使用 fetchone() 方法获取单条数据, 使用fetchall() 方法获取多条数据。
- fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
- fetchall(): 接收全部的返回结果行.
- rowcount: 这是一个只读属性,并返回执行execute()方法后影响的行数。
import pymysql
# 打开数据库连接
conn = pymysql.connect(host='119.23.55.123',port=3309,user = "root",passwd = "Luck123!",db = "learn")
# 使用cursor()方法获取操作游标
cursor = conn.cursor()
# SQL 查询语句
sql = "SELECT * FROM emp WHERE sal > {}".format(2000)
try:
cursor.execute(sql) # 执行SQL语句
result = cursor.rowcount
print('一共返回结果:{}行'.format(result))
results = cursor.fetchall() # 获取所有记录列表
print(results) #以元组的方式输出查询结果
for row in results:
fname = row[0]
lname = row[1]
age = row[2]
sex = row[3]
income = row[4]
# 打印结果
print("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % (fname, lname, age, sex, income))
except:
print("Error: unable to fetch data")
# 关闭数据库连接
conn.close()
作者: 千里和他的软件测试
软件测试学习交流: 软件测试交流群 172489141
银行金融业务交流: 软件测试学习交流群 52304542
接口自动化性能交流: 一个正经的测试群 188427938