Python学习 之 pymysql

安装

pip3 install pymysql

---

使用示例

• 代码

  import pymysql
  
  
  class DB_handler:
      def __init__(self, db_ip, db_port, db_user, db_pwd, db_name):
          self.ip = db_ip
          self.port = db_port
          self.user = db_user
          self.pwd = db_pwd
          self.dbname = db_name
          self.connect()
  
      # 连接数据库
      def connect(self):
          self.db = pymysql.connect(user=self.user, password=self.pwd, host=self.ip, port=self.port, db=self.dbname)
          self.cur = self.db.cursor(cursor=pymysql.cursors.DictCursor)
  
      # 关闭连接
      def close(self):
          self.cur.close()
          self.db.close()
  
      # 查询
      def query(self, sql):
          try:
              self.cur.execute(sql)
              return self.cur.fetchall()
          except Exception as e:
              raise e
  
      # 更新
      def update(self, sql):
          # 增加/删除/修改数据
          # sql_del = 'delete from user where id = %d'  # 删除
          # sql_add = 'insert into user(id, username, password) values(4, "username", "password")'  # 增加
          # sql_update = 'update user set username = %s where id = %d'  # 修改
          try:
              self.cur.execute(sql)
              self.db.commit()
          except:
              self.db.rollback()
  
  
  if __name__ == '__main__':
      # 实例化操作数据库的对象
      db_obj = DB_handler("192.168.248.79", 3366, "root", "xxx", "t")
  
      # 查询
      r = db_obj.query("select * from stu;")
      print(r)
  
      # 更新
      r = db_obj.update("update stu set name = 'Fam' where id = 1;")
      print(r)
  
      # 查询, 验证
      r = db_obj.query("select * from stu;")
      print(r)
  
      db_obj.close()
  

• 效果
  

---

游标

创建

# 默认, 数据类型为 ((),()..)
cursor = conn.cursor()

# [{},{}..], list 中元素为 dict
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

# [(),()..], list 中元素为 tuple, SS(流式游标), 处理大量数据
# 流式游标, 不会将数据存在内存
# 但是未完成前, 该 conn 不能去处理其他 sql, 包括生成另一个 cur 也不行
# 超过 60s 会自动断开, 也可以修改 SET NET_WRITE_TIMEOUT 来增加超时间隔
cursor = conn.cursor(cursor=pymysql.cursors.SSCursor)

# [{},{}..], list 中元素为 dict
cursor = conn.cursor(cursor=pymysql.cursors.SSDictCursor)

---

取值

fetchone()
# 返回单个元祖, 即一条记录(row), 如果没有结果返回 None
# 重复多次可逐条取出

fetchall()
# 返回((), (), ()..), 即多条记录(rows), 如果没有结果返回 ()
# 通过索引取出

fetchmany(int)
# 返回 fetchall() 的前 int 条数的记录, int 不填默认为一条
# 起始查询为空, 则返回(), 最后的条目查询为空, 则只返回前边查询到的

# fetchmany 的示例, 一共只有 3 条记录
# 取 1 条
fetchmany(1) 
((1, 'Tom', datetime.datetime(2021, 6, 1, 13, 4, 34)),)

# 取 3 条, 此时第三条数据为空, 但不会返回((xxx,),(xxx,),()), 而是返回((xxx,),(xxx,))
fetchmany(3)
((2, 'Tim', datetime.datetime(2021, 6, 1, 17, 17, 53)), (3, 'Sam', datetime.datetime(2021, 6, 1, 17, 18)))

# 再取 2 条, 此时两条数据均为空, 但不会返回((),()), 而是返回()
fetchmany(2)
()

---

流式游标的使用

import pymysql

conn = pymysql.connect(user="root", password="xxx", host="192.168.248.79", port=3366, db="t")

# 创建流式游标
cur = conn.cursor(cursor=pymysql.cursors.SSDictCursor)

# 创建结果集
ret = []

# 执行 sql
cur.execute("select * from stu;")

# 流式游标只有和 fetchone()配合才有意义
r = cur.fetchone()

# 如果结果读取完毕, 此时 r 是 None, 就会退出循环
while r:
    ret.append(r)
    r = cur.fetchone()

# 输出结果
print(ret)