使用元类实现Django的ORM

夜幕下的无眠

一、ORM基本介绍

1. ORM 是 python编程语言后端web框架 Django的核心思想，“Object Relational Mapping”，即对象-关系映射，简称ORM。

复制代码

二、实现目标

1. 创建一个实例对象，用创建它的类名当做数据表名，用创建它的类属性对应数据表的字段，当对这个实例对象操作时，能够对应MySQL语句

复制代码

如图：

三、代码实现

1. class ModelMetaClass(type):
3.     def __new__(cls, name, bases, attrs):
4.         mappings = dict()
5.         for k, v in attrs.items():
6.             # 只处理类中属性值为元组的键值对
7.             if isinstance(v, tuple):
8.                 mappings[k] = v
10.         # 处理完成后删除属性
11.         for k in mappings:
12.             attrs.pop(k)
14.         attrs["__mappings__"] = mappings
15.         attrs["__table__"] = name
16.         # 这里不能用type(name, bases, attrs) 直接返回创建好的类, 不会触发type的__init__方法
17.         return type.__new__(cls, name, bases, attrs)
20. class Model(metaclass=ModelMetaClass):
21.     # uid = ("uid", "int unsigned")
22.     # name = ("username", "varchar(30)")
23.     # email = ("email", "varchar(30)")
24.     # password = ("password", "varchar(30)")
26.     def __init__(self, **kwargs):
27.         # 将关键字参数添加到实例属性
28.         for k, v in kwargs.items():
29.             setattr(self, k, v)
31.     def save(self):
32.         table_name = self.__table__
33.         field = list()
34.         args = list()
35.         for k, v in self.__mappings__.items():
36.             field.append(v[0])
37.             arg = getattr(self, k, None)
38.             # ["12345","""'Michael'""","""'test@orm.org'""","""'my-pwd'"""]
39.             if isinstance(arg, str):
40.                 args.append("""'{}'""".format(arg))
41.             else:
42.                 args.append(str(arg))
44.         sql = "insert into {}({}) values ({})".format(table_name, ",".join(field), ",".join(args))
45.         print(f"SQL语句: {sql}")
48. class User(Model):
49.     uid = ("uid", "int unsigned")
50.     name = ("username", "varchar(30)")
51.     email = ("email", "varchar(30)")
52.     password = ("password", "varchar(30)")
55. u = User(uid=12345, name="Michael", email="test@orm.org", password="my-pwd")
56. u.save()

复制代码

代码运行结果:

来源:https://www.cnblogs.com/cs-songbai/p/18087184
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！