工作中难免有遇到 MySQL 欧巴的时候,奈何我是个记性不大好的人,临笔忘词,只能祭出谷歌大法,于是想着造个小破轮子,优雅地解决 Python 连接 MySQL 的问题。

这个轮子一定不要像 sqlalchemy 那么复杂,也不需要像 Django 的 ORM 那么丰富,就像一碗白米饭,管你吃饱,还为你解决三大问题:

  1. 轻松连接 MySQL
  2. 数据结构到模型的映射
  3. 满足增改查的基本功能,避免手写SQL

创建连接

pythonfrom data_handler import Database

db_config = {
    \'host\': \'localhost\',
    \'port\': 3306,
    \'user\': \'root\',
    \'password\': \'123456\',
    \'database\': \'test\'
}
Database.connect(**db_config)

定义模型

pythonfrom data_handler import Model, Field

class TestModel(Model):
  db_table = \'test\'
  a = Field()
  b = Field()

插入

pythontest = TestModel()
test.a = 5
test.b = \'john\'
test.save()

查询

pythonfor r in TestModel.where(a=5, b=\'john\').select():
  print r.a
  print r.b

计数

pythonprint TestModel.where(a=5, b=\'john\').count()

更新

pythonTestModel.where(a=5, b=\'john\').update(a=1)

复杂查询请使用raw sql

pythonfrom data_handler import execute_raw_sql

results = execute_raw_sql(\'select b, count(*) from test where b = %s group by b;\', (1,))
for val, cnt in results:
  print val, cnt

代码放GitHub了 –> QuickORM

来自:建造者说