Python PyMySQL-python 中数据库的简单调用

zcx · October 17, 2019 · Last by 测试菜鸟 replied at October 22, 2019 · 628 hits

MySQLdb在python3无法使用,Python2中是使用mysqldb。而在python3中可以使用库pymysql。
1.先导入库名
可以在pycharm中下载相应的包

import pymysql

2.连接数据库

db=pymysql.connect(host="localhost",port=3306,database='test_db',user='root',password='123456')

说明:

  • host是数据库的地址,因为我的数据库就在本机,所以就是localhost
  • port是在docker建立mysql的时候设置的端口(注意端口号非字符串,不用引号)
  • database要使用到的数据库名
  • user和password是连接mysql的账号、密码

3.获取游标

#通过连接对象获取游标
cursor = db.cursor()

5.执行sql语句

(1)查询语句

#使用execute()方法执行 SQL 查询
cursor.execute("select count(*) from TB_Student;")

(2)修改语句
此处以更新操作为例

sql="update TB_Student set seq=%s where id=%s"%(j,i)
cursor.execute(sql)

说明:可以一直使用一开始声明的游标对象,去执行数据库操作

6.获取sql语句执行后的结果

result_count = cursor.fetchone()
  • fetchone()获取sql语句执行后的单条数据
  • fetchall()获取sql语句执行后的所有数据 注意:这两个函数返回结果是元组类型

7.将修改操作提交到数据库
说明:

  • 步骤5并不会将操作提交到数据库,若想实际修改数据库的数据,需要执行commit()函数
#连接完数据库并不会自动提交,所以需要手动 commit 你的改动
db.commit()

下面举一个实例(面试题)
存在一张表如下:

在上述名单中,ID是连续编号的数字。现在需要按照班级给他们设一个新的序号(Seq),每个班级从1开始连续编号,输出如下表所示的结果,请写成相应的sql语句或者代码。

代码如下:

#!/usr/bin/python3
import pymysql
#连接数据库
#cursorclass = pymysql.cursors.DictCursor变成字典作为元素的列表,例如[{'count(*)': 13}]
db = pymysql.connect(host="localhost",port=3306,database='test_db',user='root',password='123456',
cursorclass = pymysql.cursors.DictCursor)

#通过连接对象获取游标
cursor = db.cursor()
#查询表TB_Student行数(在元组中)
#使用execute()方法执行 SQL 查询
cursor.execute("select count(*) from TB_Student;")
#获取表里所有内容
result_count = cursor.fetchall()
#输出结果为[{'count(*)': 13}],'count(*)'是select语句里需要查询的字段
print(result_count)
#result_count[0][0]才是真正的表TB_Student行数
print(result_count[0]['count(*)'])
#为表TB_Student增加字段seq
#cursor.execute("alter table TB_Student add seq int;")
#i是列数
#j是插入的数
j=0
n=5
for i in range(1,result_count[0]['count(*)']+1):
#seq的规律是(1234512345612345..)
#当到第6、12、19列需要从新从1开始(j赋值为1)
if j < n:
j = j+1
else:
n = n+1
j=1
print("i:"+str(i)+" j:"+str(j))
sql1="update TB_Student set seq=%s where id=%s"%(j,i)
cursor.execute(sql1)
cursor.execute("select * from TB_Student;")
r1=cursor.fetchall()
print(r1)
#连接完数据库并不会自动提交,所以需要手动 commit 你的改动
db.commit()
db.close()

测试菜鸟一枚,多谢各位指教

共收到 6 条回复 时间 点赞

优秀哦~

第11列是6,不是1开始

在init连接的时候加个这个参数更方便操作 cursorclass = pymysql.cursors.DictCursor
即可转换为字典

pymysql的游标最好设置返回为键值对形式,建议将各种查询方法进行封装

zcx #5 · October 21, 2019 作者
测试菜鸟 回复

谢谢

昨天看到一个新的连接数据库的
db = records.Database('mysql+pymysql://user:pwd@ip:port/db_name')
# 查询
rows = db.query('SELECT * FROM xxx;')
print(rows.all(as_dict=True))

requests作者封装的,查询同样一条sql,未封装的大概是0.1s,这个封装过后的records大概是0.4s左右

需要 Sign In 后方可回复, 如果你还没有账号请点击这里 Sign Up