有了db模块,操作数据库直接写SQL就很方便。但是,我们还缺少ORM。如果有了ORM,就可以用类似这样的语句获取User对象: user = User.get('123') 而不是写SQL然后再转换成User对象: u = db.select_one('select * from users where id=?', '123') user = User(**u)
所以我们开始编写ORM模块: 设计ORM接口和设计db模块类似,设计ORM也是从上层调用者角度来设计。
我们先考虑如何定义一个User对象,然后把数据库表 from transwarp.orm import Model, StringField, IntegerField class User(Model): __table__ = 'users' id = IntegerField(primary_key=True) name = StringField()
注意到定义在 # 创建实例: user = User(id=123, name='Michael') # 存入数据库: user.insert() 实现ORM模块有了定义,我们就可以开始实现ORM模块。
首先要定义的是所有ORM映射的基类 class Model(dict): __metaclass__ = ModelMetaclass def __init__(self, **kw): super(Model, self).__init__(**kw) def __getattr__(self, key): try: return self[key] except KeyError: raise AttributeError(r"'Dict' object has no attribute '%s'" % key) def __setattr__(self, key, value): self[key] = value
>>> user['id'] 123 >>> user.id 123
class ModelMetaclass(type): def __new__(cls, name, bases, attrs): mapping = ... # 读取cls的Field字段 primary_key = ... # 查找primary_key字段 __table__ = cls.__talbe__ # 读取cls的__table__字段 # 给cls增加一些字段: attrs['__mapping__'] = mapping attrs['__primary_key__'] = __primary_key__ attrs['__table__'] = __table__ return type.__new__(cls, name, bases, attrs)
这样,任何继承自
然后,我们往 class Model(dict): ... @classmethod def get(cls, pk): d = db.select_one('select * from %s where %s=?' % (cls.__table__, cls.__primary_key__.name), pk) return cls(**d) if d else None
user = User.get('123')
往 class Model(dict): ... def insert(self): params = {} for k, v in self.__mappings__.iteritems(): params[v.name] = getattr(self, k) db.insert(self.__table__, **params) return self
这样,就可以把一个 user = User(id=123, name='Michael') user.insert() 最后一步是完善ORM,对于查找,我们可以实现以下方法:
对于count,可以实现:
以及 最后看看我们实现的ORM模块一共多少行代码?加上注释和doctest才仅仅300多行。用Python写一个ORM是不是很容易呢? 转载请保留固定链接: https://linuxeye.com/program/2544.html |