class ModelMetaclass(type):
def __new__(cls, name, bases, attrs):
mappings = dict()
# 判斷是否需要保存
for k, v in attrs.items():
# 判斷是否是指定StringField或者IntegerField的實例對象
if isinstance(v, tuple):
# print('Found mapping: %s => %s' % (k, v))
mappings[k] = v
# 刪除這些已經在字典中存儲的屬性
for k in mappings.keys():
attrs.pop(k)
attrs['__mappings__'] = mappings
attrs['__table__'] = name
return type.__new__(cls, name, bases, attrs)
class User(metaclass=ModelMetaclass):
uid = ('uid', "int unsigned")
name = ('username', "varchar(40)")
email = ('email', "varchar(80)")
# 當指定元類之後,以上的類屬性將不在類中,而是在__mappings__屬性指定的字典中
# __mappings__ = {
# "uid" : ('uid', "int unsigned"),
# "name": ('username', "varchar(40)"),
# "email": ('email', "varchar(80)"),
# }
# __table__ = "User"
def __init__(self, **kwargs):
for name, value in kwargs.items():
setattr(self, name, value)
def save(self):
fields = []
args = []
for k, v in self.__mappings__.items():
fields.append(v[0])
args.append(getattr(self, k, None))
# sql = "insert into %s (%s) value (%s)" % (self.__table__, ','.join(fields), ','.join([str(i) for i in args]))
args_temp = list()
for temp in args:
# 判斷是否是數字類型
if isinstance(temp, int):
args_temp.append(str(temp))
elif isinstance(temp, str):
args_temp.append("""'%s'""" % temp)
sql = 'insert into %s (%s) value (%s)' % (self.__table__, ','.join(fields), ','.join(args_temp))
print(sql)
u = User(uid=123, name="test", email="[email protected]")
u.save()
元類實例,操作數據庫
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.