194 lines
7.3 KiB
Python
194 lines
7.3 KiB
Python
import copy
|
||
import uuid
|
||
from django.apps import apps
|
||
from django.core.validators import ValidationError
|
||
from django.db import models
|
||
from django.db.models import ManyToManyRel
|
||
from dependency.base.utils import get_models_field_detail, get_app_info, judge_the_same, \
|
||
judge_diff_be_two_models_and_change, table_exists, build_serializers, install_model
|
||
from django.core.exceptions import FieldDoesNotExist
|
||
|
||
|
||
def get_model(model_name, queryset_dict=None):
|
||
if queryset_dict is None:
|
||
model_infos = Model.objects.get(name=model_name)
|
||
else:
|
||
model_infos = Model.objects.filter(**queryset_dict)
|
||
get_serializers = build_serializers(Model.__name__, Model, {}, False)
|
||
model_data = get_serializers(model_infos).data
|
||
app_config = apps.get_app_config(model_data["app"]["name"])
|
||
return app_config.get_model(model_name)
|
||
|
||
|
||
def get_model_status(model_name, queryset_dict=None):
|
||
if queryset_dict is None:
|
||
model_infos = Model.objects.get(name=model_name)
|
||
else:
|
||
model_infos = Model.objects.filter(**queryset_dict)
|
||
return model_infos
|
||
|
||
|
||
def get_model_fields(model_name):
|
||
model_id = Model.objects.get(name=model_name).id
|
||
get_serializers = build_serializers(ModelField.__name__, ModelField, {}, False)
|
||
return get_serializers(ModelField.objects.filter(model_id=model_id), many=True).data
|
||
|
||
|
||
def reverse_add_sub_data(add_data, sub_model_instance=None):
|
||
return_datas = []
|
||
for items in add_data:
|
||
need_del_key = []
|
||
need_add_sub_list = {}
|
||
for key, value in items.items():
|
||
if isinstance(value, list) and len(value) > 0:
|
||
sub_model = sub_model_instance._meta.get_field(key).remote_field.model
|
||
res = reverse_add_sub_data(value, sub_model_instance=sub_model)
|
||
need_add_sub_list.update({key: res})
|
||
need_del_key.append(key)
|
||
elif isinstance(value, list) and len(value) == 0:
|
||
need_del_key.append(key)
|
||
# 数据库添加数据
|
||
for keys in need_del_key:
|
||
del items[keys]
|
||
items["uuid"] = str(uuid.uuid4())
|
||
if 'id' in items:
|
||
sub_item_ins = sub_model_instance.objects.get(id=items["id"])
|
||
else:
|
||
sub_item_ins = sub_model_instance.objects.create(**items)
|
||
return_datas.append(sub_item_ins)
|
||
for add_sub_item_k, add_sub_item_v in need_add_sub_list.items():
|
||
for add_sub_item in add_sub_item_v:
|
||
many_to_many_manager = getattr(sub_item_ins, add_sub_item_k)
|
||
many_to_many_manager.add(add_sub_item)
|
||
return return_datas
|
||
|
||
|
||
def def_new_model(name, fields=None, app_label='', module='', options=None):
|
||
class Meta:
|
||
# Using type('Meta', ...) gives a dictproxy error during model creation
|
||
pass
|
||
|
||
if app_label:
|
||
# app_label必须用Meta内部类来设定
|
||
setattr(Meta, 'app_label', app_label)
|
||
|
||
# 若提供了options参数，就要用更新Meta类
|
||
if options is not None and not isinstance(options, bool):
|
||
for key, value in options.iteritems():
|
||
setattr(Meta, key, value)
|
||
|
||
# 创建一个字典来模拟类的声明，module和当前所在的module对应
|
||
attrs = {'__module__': module, 'Meta': Meta}
|
||
|
||
# 创建这个类，这会触发ModelBase来处理
|
||
model = type(name, (models.Model,), attrs)
|
||
|
||
# 加入所有提供的字段
|
||
if fields:
|
||
for item, value in fields.items():
|
||
if isinstance(value, models.CharField):
|
||
value.max_length = int(value.max_length)
|
||
value.contribute_to_class(model, item)
|
||
return model
|
||
|
||
|
||
def create_model(name, fields=None, app_label='', module='', options=None):
|
||
"""
|
||
创建指定model
|
||
"""
|
||
# 查询当前app内的模型是否注册
|
||
all_models = get_app_info(app_label)
|
||
app_config = apps.get_app_config(app_label)
|
||
for model in all_models:
|
||
if model.__name__ == name:
|
||
exist_list = get_models_field_detail(model)
|
||
new_list = get_models_field_detail({}, fields=fields)
|
||
add_item = []
|
||
for item in model._meta.get_fields():
|
||
if isinstance(item, ManyToManyRel):
|
||
add_item = get_models_field_detail({}, fields={item.name: item})
|
||
new_list.extend(add_item)
|
||
for field in model._meta.get_fields():
|
||
if isinstance(field, models.ManyToManyField):
|
||
# 获取关联模型的序列化器
|
||
related_model = field.remote_field.through
|
||
if related_model.__name__ in app_config.models:
|
||
del app_config.models[related_model.__name__]
|
||
if judge_the_same(exist_list, new_list):
|
||
## 清理模型缓存
|
||
apps.clear_cache()
|
||
|
||
# 从 sys.modules 中移除旧模型
|
||
model_key = f"{name.lower()}"
|
||
|
||
if model_key in app_config.models.keys():
|
||
del app_config.models[model_key]
|
||
|
||
# 重新定义模型
|
||
new_model = def_new_model(name, fields, app_label, module, options)
|
||
judge_diff_be_two_models_and_change(model, new_model)
|
||
return new_model
|
||
else:
|
||
return model
|
||
model = def_new_model(name, fields, app_label, module, options)
|
||
if not table_exists((app_label + "_" + name)):
|
||
print("模型初始化：", app_label + "_" + name)
|
||
install_model(model)
|
||
return model
|
||
|
||
|
||
class App(models.Model):
|
||
name = models.CharField(max_length=255)
|
||
name_cn = models.CharField(max_length=255)
|
||
module = models.CharField(max_length=255)
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
|
||
class Model(models.Model):
|
||
app = models.ForeignKey(App, related_name='models', on_delete=models.CASCADE)
|
||
name = models.CharField(max_length=255)
|
||
name_cn = models.CharField(max_length=255)
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
def get_django_model(self):
|
||
"Returns a functional Django model based on current data"
|
||
# Get all associated fields into a list ready for dict()
|
||
fields = [(f.name, f.get_django_field()) for f in self.model_fields.all()]
|
||
# Use the create_model function defined above
|
||
return create_model(self.name, dict(fields), self.app.name, self.app.module)
|
||
|
||
class Meta:
|
||
unique_together = (('app', 'name'),)
|
||
|
||
|
||
class ModelField(models.Model):
|
||
model = models.ForeignKey(Model, related_name='model_fields', on_delete=models.CASCADE)
|
||
name = models.CharField(max_length=255)
|
||
name_cn = models.CharField(max_length=255)
|
||
type = models.CharField(max_length=255)
|
||
is_show = models.BooleanField(default=True)
|
||
|
||
def get_django_field(self):
|
||
"Returns the correct field type, instantiated with applicable settings"
|
||
# Get all associated settings into a list ready for dict()
|
||
settings = [(s.name, s.value) for s in self.model_settings.all()]
|
||
# Instantiate the field with the settings as **kwargs
|
||
return getattr(models, self.type)(**dict(settings))
|
||
|
||
class Meta:
|
||
unique_together = (('model', 'name'),)
|
||
|
||
|
||
class ModelSetting(models.Model):
|
||
field = models.ForeignKey(ModelField, related_name='model_settings', on_delete=models.CASCADE)
|
||
name = models.CharField(max_length=255)
|
||
name_cn = models.CharField(max_length=255)
|
||
value = models.CharField(max_length=255)
|
||
|
||
class Meta:
|
||
unique_together = (('field', 'name'),)
|