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

zcx · October 17, 2019 · Last by 测试菜鸟 replied at October 22, 2019 · 2574 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 Author
测试菜鸟 回复

谢谢

昨天看到一个新的连接数据库的
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