功能变化: 初始化系统管理代码优化

treebeard
李强 2022-05-18 15:16:19 +08:00
parent 04d18f617a
commit bd20a54618
2 changed files with 6 additions and 6 deletions

View File

@ -2,16 +2,13 @@
import os import os
import django import django
from rest_framework import request
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "application.settings") os.environ.setdefault("DJANGO_SETTINGS_MODULE", "application.settings")
django.setup() django.setup()
from dvadmin.system.views.user import UsersInitSerializer from dvadmin.system.views.user import UsersInitSerializer
from application.settings import BASE_DIR
from dvadmin.system.views.menu import MenuInitSerializer from dvadmin.system.views.menu import MenuInitSerializer
from dvadmin.utils.core_initialize import CoreInitialize from dvadmin.utils.core_initialize import CoreInitialize
from dvadmin.system.models import Users
from dvadmin.system.views.role import RoleInitSerializer from dvadmin.system.views.role import RoleInitSerializer
from dvadmin.system.views.api_white_list import ApiWhiteListInitSerializer from dvadmin.system.views.api_white_list import ApiWhiteListInitSerializer
from dvadmin.system.views.dept import DeptInitSerializer from dvadmin.system.views.dept import DeptInitSerializer

View File

@ -2,10 +2,10 @@
import json import json
import os import os
from django.apps import apps
from rest_framework import request from rest_framework import request
from application import settings from application import settings
from application.settings import BASE_DIR
from dvadmin.system.models import Users from dvadmin.system.models import Users
@ -30,8 +30,11 @@ class CoreInitialize:
def init_base(self, Serializer, unique_fields=None): def init_base(self, Serializer, unique_fields=None):
model = Serializer.Meta.model model = Serializer.Meta.model
with open(os.path.join(BASE_DIR, *self.app.split('.'), 'fixtures', path_file = os.path.join(apps.get_app_config(self.app.split('.')[-1]).path, 'fixtures',
f'init_{Serializer.Meta.model._meta.model_name}.json')) as f: f'init_{Serializer.Meta.model._meta.model_name}.json')
if not os.path.isfile(path_file):
return
with open(path_file) as f:
for data in json.load(f): for data in json.load(f):
filter_data = {} filter_data = {}
# 配置过滤条件,如果有唯一标识字段则使用唯一标识字段,否则使用全部字段 # 配置过滤条件,如果有唯一标识字段则使用唯一标识字段,否则使用全部字段