转载

【Python】基于pymysql的数据库操作类

一 简介
     Python和MySQL交互的模块有 MySQLdb 和 PyMySQL(pymysql),MySQLdb是基于C 语言编写的，而且Python3 不在支持MySQLdb 。PyMySQL是一个纯Python写的MySQL客户端，它的目标是替代MySQLdb，可以在CPython、PyPy、IronPython和Jython环境下运行,PyMySQL在MIT许可下发布。
    在开发基于Python语言的项目中，为了以后系统能兼容Python3,我们使用了PyMySQL替换了MySQLdb。下面我们来熟悉一下pymysql的使用。
 
二 安装方式
  pymsql的源码 https://github.com/PyMySQL/PyMySQL ，目前还在持续更新。
  1. 安装要求:
  2. Python -- one of the following:
  3.      CPython >= 2.6 or >= 3.3
  4.      PyPy >= 4.0
  5.      IronPython 2.7
  6. MySQL Server -- one of the following:
  7.      MySQL >= 4.1 (tested with only 5.5~)
  8.      MariaDB >= 5.1
  9. 安装
  10.   pip install PyMySQL
  相关文档 , 强烈建议大家仔细阅读一遍。其用法和MySQLdb相差无几，核心用法一致。这样使用pymysql替换mysqldb的成本极小。


三 基于pymysql的数据库交互

  1. #!/usr/bin/env python
  2. # encoding: utf-8
  3. """
  4. author: yangyi@youzan
  5. time: 2015/6/8 上午11:34
  6. func: 基于pymysql的数据库交互类，支持事务提交和回滚，返回结果记录行数，和insert的最新id
  7. """
  8. import pymysql
  9. from warnings import filterwarnings
  10. filterwarnings('ignore', category=pymysql.Warning)
  11. CONNECT_TIMEOUT = 100
  12. IP = 'localhost'
  13. PORT = 3306
  14. USER = 'root'
  15. PASSSWORD = ''

  17. class QueryException(Exception):
  18.     """
  19.     """


  22. class ConnectionException(Exception):
  23.     """
  24.     """

  26. class MySQL_Utils():
  27.     def __init__(
  28.             self, ip=IP, port=PORT, user=USER, password=PASSSWORD,
  29.             connect_timeout=CONNECT_TIMEOUT, remote=False, socket='', dbname='test'):
  30.         self.__conn = None
  31.         self.__cursor = None
  32.         self.lastrowid = None
  33.         self.connect_timeout = connect_timeout
  34.         self.ip = ip
  35.         self.port = port
  36.         self.user = user
  37.         self.password = password
  38.         self.mysocket = socket
  39.         self.remote = remote
  40.         self.db = dbname
  41.         self.rows_affected = 0


  44.     def __init_conn(self):
  45.         try:
  46.             conn = pymysql.connect(
  47.                     host=self.ip,
  48.                     port=int(self.port),
  49.                     user=self.user,
  50.                     db=self.db,
  51.                     connect_timeout=self.connect_timeout,
  52.                     charset='utf8', unix_socket=self.mysocket)
  53.         except pymysql.Error as e:
  54.             raise ConnectionException(e)
  55.         self.__conn = conn


  58.     def __init_cursor(self):
  59.         if self.__conn:
  60.             self.__cursor = self.__conn.cursor(pymysql.cursors.DictCursor)


  63.     def close(self):
  64.         if self.__conn:
  65.             self.__conn.close()
  66.             self.__conn = None

  68.     #专门处理select 语句
  69.     def exec_sql(self, sql, args=None):
  70.         try:
  71.             if self.__conn is None:
  72.                 self.__init_conn()
  73.                 self.__init_cursor()
  74.             self.__conn.autocommit = True
  75.             self.__cursor.execute(sql, args)
  76.             self.rows_affected = self.__cursor.rowcount
  77.             results = self.__cursor.fetchall()
  78.             return results
  79.         except pymysql.Error as e:
  80.             raise pymysql.Error(e)
  81.         finally:
  82.             if self.__conn:
  83.                 self.close()

  85.     # 专门处理dml语句 delete，updete，insert 
  86.     def exec_txsql(self, sql, args=None):
  87.         try:
  88.             if self.__conn is None:
  89.                 self.__init_conn()
  90.                 self.__init_cursor()
  91.             if self.__cursor is None:
  92.                 self.__init_cursor()


  95.             self.rows_affected=self.__cursor.execute(sql, args)
  96.             self.lastrowid = self.__cursor.lastrowid
  97.             return self.rows_affected
  98.         except pymysql.Error as e:
  99.             raise pymysql.Error(e)
  100.         finally:
  101.             if self.__cursor:
  102.                 self.__cursor.close()
  103.                 self.__cursor = None

  105.     # 提交
  106.     def commit(self):
  107.         try:
  108.             if self.__conn:
  109.                 self.__conn.commit()
  110.         except pymysql.Error as e:
  111.             raise pymysql.Error(e)
  112.         finally:
  113.             if self.__conn:
  114.                 self.close()

  116.     #回滚操作
  117.     def rollback(self):
  118.         try:
  119.             if self.__conn:
  120.                 self.__conn.rollback()
  121.         except pymysql.Error as e:
  122.             raise pymysql.Error(e)
  123.         finally:
  124.             if self.__conn:
  125.                 self.close()
  126.     # 适用于需要获取插入记录的主键自增id
  127.     def get_lastrowid(self):
  128.         return self.lastrowid
     #获取dml操作影响的行数
    def get_affectrows(self):
        return self.rows_affected
     #MySQL_Utils初始化的实例销毁之后，自动提交
     def __del__(self):
        self.commit()
四 小结
   前几天刚刚将我们的系统中的MySQLdb 替换为PyMySQL, 还未遇到问题。欢迎大家使用测试上述脚本，有问题欢迎和我讨论。如果本文对您有帮助 ，可以赞助 一瓶饮料。
【Python】基于pymysql的数据库操作类



正文到此结束
所属分类： 编程技术

热门推荐

相关文章

阿里云首购8折
说给你听

本文目录
随机标签
书籍教程
springboot-demo Java入门教程 bootstrap3 CSS Apache基础教程 php ionic 教程 Python mysql教程 eclipse Ubuntu VPS系统配置 AngularJS 教程 MongoDB教程 Struts2教程 Redis教程 Spring教程 Git教程 Jenkins进阶系列 openfire参考指南 Java设计模式 HBase教程 java-demo Maven教程 springcloud-demo hibernate教程 Docker 教程 memcached教程 Quartz指南 Hive教程 ANTLR教程 SpringCloud Ant教程 java实例教程 Hazelcast教程 XStream教程 Elastic-Job-Lite 深入浅出MyBatis ibaties教程 SVN教程 rabittmq教程 Hadoop教程 solr教程 WebService CXF学习 JPA教程 ActiveMQ中文指南 Java内存模型 dubbo教程 python3-demo Linux入门视频教程
近期评论
  1. 1 使用 Docker 打包和运行 Vue 应用
  2. 2 Dependency Check命令行方式扫描jar包的安全漏洞
  3. 3 maven使用Dependency-Check来扫描安全漏洞
  4. 4 使用sonarqube分析代码质量
  5. 5 使用screw来对比数据库表和字段差异
  6. 6 Spring Boot 实现防盗链功能
  7. 7 在Spring Boot中的实现国际化（i18n）
  8. 8 一个用于 SMB/CIFS 网络的 Java 库-JCIFS
  9. 9 Spring Boot 3.x嵌入MongoDB 进行测试
  10. 10 java如何解析和生成yaml文件？
网站信息
  • 文章总数：82,737 篇
  • 文件总数：284,312 个
  • 标签总数：2,416 个
  • 分类总数：85 个
  • 留言数量：2,559 条
  • 在线人数：644 人
  • 运行天数：4,439天
  • 最后更新：2024年12月22日17点
Loading...

其他链接

关于本站

本站定位：个人技术类博客

本站作用：写博客、记日志、闲聊扯淡鼓捣技术。

问题交流

[HBLOG]公众号 HBLOG HBLOG