100行代码还你一个Python ORM

工作中难免有遇到MySQL欧巴的时候,奈何我是个记性不大好的人,临笔忘词,只能祭出谷歌大法,于是想着造个小破轮子,优雅地解决Python连接MySQL的问题。
这个轮子一定不要像sqlalchemy那么复杂,也不需要像Django的ORM那么丰富,就像一碗白米饭,管你吃饱,还为你解决三大问题:

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

创建连接

1
2
3
4
5
6
7
8
9
10
from data_handler import Database

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

定义模型

1
2
3
4
5
6
from data_handler import Model, Field

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

插入

1
2
3
4
test = TestModel()
test.a = 5
test.b = 'john'
test.save()

查询

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

计数

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

更新

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

复杂查询请使用raw sql

1
2
3
4
5
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

喜欢的客官别忘了给小2赏个星星哦:)

2shou wechat
更多精彩内容,欢迎关注微信公众号「极客思享」