Python编程：metaclass元类实现简单的ORM
【摘要】 ORM全称“Object Relational Mapping”，即对象-关系映射 把关系数据库的一行映射为一个对象，一个类对应一个表 这样，写代码更简单，不用直接操作SQL语句。
# -*- coding: utf-8 -*-
# @File : 元类实例ORM.py
# @Date : 2018-06-01
# @Author : Peng Shiyu
...
ORM全称“Object Relational Mapping”，即对象-关系映射
把关系数据库的一行映射为一个对象，一个类对应一个表
这样，写代码更简单，不用直接操作SQL语句。
# -*- coding: utf-8 -*-
# @File : 元类实例ORM.py
# @Date : 2018-06-01
# @Author : Peng Shiyu
# 先定义metaclass，就可以创建类，最后创建实例
class Field(object): """保存数据库表的字段名和字段类型""" def __init__(self, name, column_type): self.name = name self.column_type = column_type def __str__(self): return "<%s %s %s>"%(self.__class__.__name__, self.name, self.column_type)
# 定义各种类型
class IntegerField(Field): def __init__(self, name): super(IntegerField, self).__init__(name, "int")
class StringField(Field): def __init__(self, name): super(StringField, self).__init__(name, "varchar(255)")
# 编写元类
"""
排除掉对Model类的修改；
在当前类（比如User）中查找定义的类的所有属性，如果找到一个Field属性，就把它保存到一个__mappings__的dict中，
同时从类属性中删除该Field属性，否则，容易造成运行时错误（实例的属性会遮盖类的同名属性）；
把表名保存到__table__中，这里简化为表名默认为类名。
"""
class ModelMetaclass(type): def __new__(cls, name, bases, attrs): if name == "Model": return type.__new__(cls, name, bases, attrs) print("Fond Model: %s"% name) # 查找Field字段 mappings = {} for k, v in attrs.items(): if isinstance(v, Field): print("Fond Field: %s => %s"% (k, v)) mappings[k] = v # 弹出Field字段 for k in mappings.keys(): attrs.pop(k) attrs["__mappings__"] = mappings # 保存属性和列的映射关系 attrs["__table__"] = name # 假设表名和类名一致 return type.__new__(cls, name, bases, attrs)
# 在Model类中，就可以定义各种操作数据库的方法，比如save()，delete()，find()，update等等
class Model(dict, metaclass=ModelMetaclass): def __init__(self, **kwargs): super(Model, self).__init__(**kwargs) def __getattr__(self, key): return self[key] def __setattr__(self, key, value): self[key] = value def save(self): fields = [] params = [] args = [] for k, v in self.__mappings__.items(): fields.append(v.name) params.append("?") args.append(getattr(self, k, None)) sql = "insert into %s (%s) VALUES (%s)"% (self.__table__, ",".join(fields), ",".join(params)) print("sql: %s"% sql) print("args: %s"% args)
class User(Model): # 定义类的属性到列的映射： id = IntegerField("id") name = StringField("name") email = StringField("email") password = StringField("password")
if __name__ == '__main__': # # 创建一个实例 user = User(id=1, name="Tom", email="12345@qq.com", password="123456") # 保存到数据库 user.save() # 访问元素 print(user.name) print(user["name"]) user.name = "jack" print(user.name)
"""
Fond Model: User
Fond Field: id => <IntegerField id int>
Fond Field: name => <StringField name varchar(255)>
Fond Field: email => <StringField email varchar(255)>
Fond Field: password => <StringField password varchar(255)>
sql: insert into User (id,name,email,password) VALUES (?,?,?,?)
args: [1, 'Tom', '12345@qq.com', '123456']
Tom
Tom
jack
"""
参考：
文章来源: pengshiyu.blog.csdn.net，作者：彭世瑜，版权归原作者所有，如需转载，请联系作者。
原文链接：pengshiyu.blog.csdn.net/article/details/80534768
【版权声明】本文为华为云社区用户转载文章，如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
作者其他文章
评论（0）