我的 MySQL 客户端:支持 pymysql / mysqlclient 双模块、自动重连、独创事务回滚装饰器

torndb曾经是我非常喜欢使用的一个MySQL客户端,虽然不足300行代码,却支持pymysql/mysqlclient双模块,且能在连接失效时自动重连。torndb的函数封装也很有特色,相同的功能提供了不同的名字。遗憾的是,进入Py3之后,torndb没有相应的升级,只好放弃了。

近日,重读了torndb的源码,结合我对数据库应用的理解,整理出了一个新的MySQL客户端,暂且命名为youthdb,经过测试,基本实现了我对MySQL客户端所有要求。youthdb的主要特点如下:

  • 支持 pymysql / mysqlclient 双模块
  • 自动重连,继承自torndb
  • 独创事务回滚装饰器

代码总共150行,只有一个Connection类定义,实例化方式如下:

db = Connection(host=‘locahost’, port=3306, db=‘test’, user=‘your_account’, passwd=‘your_password’)

对使用者而言,Connection类只有两个方法:query()和transaction()。query()用于执行sql语句。该方法有select、insert、update、delete等4个别名。query会根据sql语句的不同,自动返回多行查询结果、影响到的行数或最后操作的id等。query的arg参数为None或者元组,则表示单次操作;若arg参数为列表,则表示批量操作。具体用法如下:

db.select(‘select * from member where age = %s and rank = %s’, (18, ‘1d’))
db.insert(‘insert into log (op, oid) values (%s, %s)’, [(‘r’, 3), (‘w’, 5)])

有个问题需要特别说明,就是很多同学在写sql语句的时候,喜欢用格式化字符串的方法,将参数直接拼合到查询语句中——这样做,虽然不会导致直接的错误,却是

猜你喜欢

转载自blog.csdn.net/xufive/article/details/103909533