如何正确的使用pymysql模块

本篇文章给大家分享的是有关如何正确的使用pymysql模块，小编觉得挺实用的，因此分享给大家学习，希望大家阅读完这篇文章后可以有所收获，话不多说，跟着小编一起来看看吧。

pymysql模块的使用

查询一条数据fetchone()

from pymysql import * conn = connect(   host='127.0.0.1',   port=3306, user='root',   password='123456',   database='itcast',   charset='utf8') # 创建游标 c = conn.cursor() # 执行sql语句 c.execute("select * from student") # 查询一行数据 result = c.fetchone() print(result) # 关闭游标 c.close() # 关闭数据库连接 conn.close() """ (1, '张三', 18, b'\x01') """

查询多条数据fetchall()

from pymysql import * conn = connect(   host='127.0.0.1',   port=3306, user='root',   password='123456',   database='itcast',   charset='utf8') # 创建游标 c = conn.cursor() # 执行sql语句 c.execute("select * from student") # 查询多行数据 result = c.fetchall() for item in result:   print(item) # 关闭游标 c.close() # 关闭数据库连接 conn.close() """ (1, '张三', 18, b'\x01') (2, '李四', 19, b'\x00') (3, '王五', 20, b'\x01') """

更改游标的默认设置，返回值为字典

from pymysql import * conn = connect(   host='127.0.0.1',   port=3306, user='root',   password='123456',   database='itcast',   charset='utf8') # 创建游标，操作设置为字典类型 c = conn.cursor(cursors.DictCursor) # 执行sql语句 c.execute("select * from student") # 查询多行数据 result = c.fetchall() for item in result:   print(item) # 关闭游标 c.close() # 关闭数据库连接 conn.close() """ {'id': 1, 'name': '张三', 'age': 18, 'sex': b'\x01'} {'id': 2, 'name': '李四', 'age': 19, 'sex': b'\x00'} {'id': 3, 'name': '王五', 'age': 20, 'sex': b'\x01'} """

返回一条数据时也是一样的。返回字典或者时元组看个人需要。

2|2使用数据操作语句

执行增加、删除、更新语句的操作其实是一样的。只写一个作为示范。

from pymysql import * conn = connect(   host='127.0.0.1',   port=3306, user='root',   password='123456',   database='itcast',   charset='utf8') # 创建游标 c = conn.cursor() # 执行sql语句 c.execute("insert into student(name,age,sex) values (%s,%s,%s)",("小二",28,1)) # 提交事务 conn.commit() # 关闭游标 c.close() # 关闭数据库连接 conn.close()

和查询语句不同的是必须使用commit()提交事务，否则操作就是无效的。

3|0编写数据库连接类

普通版

MysqlHelper.py

from pymysql import connect,cursors class MysqlHelper:   def __init__(self,          host="127.0.0.1",          user="root",          password="123456",          database="itcast",          charset='utf8',          port=3306):     self.host = host     self.port = port     self.user = user     self.password = password     self.database = database     self.charset = charset     self._conn = None     self._cursor = None   def _open(self):     # print("连接已打开")     self._conn = connect(host=self.host,                port=self.port,                user=self.user,                password=self.password,                database=self.database,                charset=self.charset)     self._cursor = self._conn.cursor(cursors.DictCursor)   def _close(self):     # print("连接已关闭")     self._cursor.close()     self._conn.close()   def one(self, sql, params=None):     result: tuple = None     try:       self._open()       self._cursor.execute(sql, params)       result = self._cursor.fetchone()     except Exception as e:       print(e)     finally:       self._close()     return result   def all(self, sql, params=None):     result: tuple = None     try:       self._open()       self._cursor.execute(sql, params)       result = self._cursor.fetchall()     except Exception as e:       print(e)     finally:       self._close()     return result   def exe(self, sql, params=None):     try:       self._open()       self._cursor.execute(sql, params)       self._conn.commit()     except Exception as e:       print(e)     finally:       self._close()

该类封装了fetchone、fetchall、execute，省去了数据库连接的打开和关闭和游标的打开和关闭。

下面的代码是调用该类的小示例：

from MysqlHelper import * mysqlhelper = MysqlHelper() ret = mysqlhelper.all("select * from student") for item in ret:   print(item) """ {'id': 1, 'name': '张三', 'age': 18, 'sex': b'\x01'} {'id': 2, 'name': '李四', 'age': 19, 'sex': b'\x00'} {'id': 3, 'name': '王五', 'age': 20, 'sex': b'\x01'} {'id': 5, 'name': '小二', 'age': 28, 'sex': b'\x01'} {'id': 6, 'name': '娃哈哈', 'age': 28, 'sex': b'\x01'} {'id': 7, 'name': '娃哈哈', 'age': 28, 'sex': b'\x01'} """

上下文管理器版

mysql_with.py

from pymysql import connect, cursors class DB:   def __init__(self,          host='localhost',          port=3306,          db='itcast',          user='root',          passwd='123456',          charset='utf8'):     # 建立连接     self.conn = connect(       host=host,       port=port,       db=db,       user=user,       passwd=passwd,       charset=charset)     # 创建游标，操作设置为字典类型     self.cur = self.conn.cursor(cursor=cursors.DictCursor)   def __enter__(self):     # 返回游标     return self.cur   def __exit__(self, exc_type, exc_val, exc_tb):     # 提交数据库并执行     self.conn.commit()     # 关闭游标     self.cur.close()     # 关闭数据库连接     self.conn.close()

如何使用：

from mysql_with import DB with DB() as db:   db.execute("select * from student")   ret = db.fetchone()   print(ret) """ {'id': 1, 'name': '张三', 'age': 18, 'sex': b'\x01'} """

以上就是如何正确的使用pymysql模块，小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注亿速云行业资讯频道。