品易云推流 关闭
文章详情页
文章 > Python基础教程 > python简单实现了一个ORM

python简单实现了一个ORM

头像

小妮浅浅

2021-05-10 10:00:475208浏览 · 0收藏 · 0评论

(推荐操作系统:windows7系统、Python 3.9.1、DELL G3电脑。)

1、定义Field类,它负责保存数据库表的字段名和字段的类型:

class Field(object):
    def __init__(self, name, column_type):
        self.name = name
        self.column_type = column_type

2、在Field的基础上,进一步定义各种具体类型的Field,比如StringField,IntegerField等等:

class StringField(Field):
    def __init__(self, name, max_length):
        super(StringField, self).__init__(name, 'varchar(%d)' % max_length)
 
class IntegerField(Field):
    def __init__(self, name):
        super(IntegerField, self).__init__(name, 'int')

3、下一步,就是编写最复杂的ModelMetaclass了:

class ModelMetaclass(type):
    def __new__(cls, name, bases, attrs):
        if name == 'Model':
            return type.__new__(cls, name, bases, attrs)
        print('Found model: %s' % name)
        mappings = dict()
        for k, v in attrs.items():
            if isinstance(v, Field):
                mappings[k] = v
        for k in mappings.keys():
            attrs.pop(k)
        attrs['__mappings__'] = mappings
        attrs['__table__'] = name
        return type.__new__(cls, name, bases, attrs)

4、创建User类,并调用save方法,保存:

class User(Model):
    id = IntegerField('id')
    name = StringField('username', max_length=48)
    email = StringField('email', max_length=128)
    password = StringField('password', max_length=24)
 
# 创建一个实例:
u = User(id=12345, name='xiaoming', email='xiaoming@xiaomi.com', password='test')
# 保存到数据库:
u.save()

以上就是python简单实现了一个ORM,希望能对大家有所帮助。更多Python学习指路:python基础教程

关注

关注公众号,随时随地在线学习

本教程部分素材来源于网络,版权问题联系站长!

底部广告图