100 行代码还你一个 Python ORM

jopen 10年前

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

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

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

创建连接 

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

定义模型 

from data_handler import Model, Field class TestModel(Model): db_table = 'test' a = Field()    b = Field()

插入 

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

查询 

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

计数 

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

更新 

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

复杂查询请使用raw sql 

from 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

来自：建造者说