101 lines
4.0 KiB
Python
101 lines
4.0 KiB
Python
|
import json
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
import django
|
|||
|
|
from django.db.models import QuerySet
|
|||
|
|
|
|||
|
|
from dvadmin.system.views.system_config import SystemConfigInitSerializer
|
|||
|
|
|
|||
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'application.settings')
|
|||
|
|
django.setup()
|
|||
|
|
from django.core.management.base import BaseCommand
|
|||
|
|
|
|||
|
|
from application.settings import BASE_DIR
|
|||
|
|
from dvadmin.system.models import Menu, Users, Dept, Role, ApiWhiteList, Dictionary, SystemConfig
|
|||
|
|
from dvadmin.system.views.api_white_list import ApiWhiteListInitSerializer
|
|||
|
|
from dvadmin.system.views.dept import DeptInitSerializer
|
|||
|
|
from dvadmin.system.views.dictionary import DictionaryInitSerializer
|
|||
|
|
from dvadmin.system.views.menu import MenuInitSerializer
|
|||
|
|
from dvadmin.system.views.role import RoleInitSerializer
|
|||
|
|
from dvadmin.system.views.user import UsersInitSerializer
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Command(BaseCommand):
|
|||
|
|
"""
|
|||
|
|
生产初始化菜单: python3 manage.py generate_init_json 生成初始化的model名
|
|||
|
|
例如：
|
|||
|
|
全部生成：python3 manage.py generate_init_json
|
|||
|
|
只生成某个model的： python3 manage.py generate_init_json users
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def serializer_data(self, serializer, query_set: QuerySet):
|
|||
|
|
serializer = serializer(query_set, many=True)
|
|||
|
|
data = json.loads(json.dumps(serializer.data, ensure_ascii=False))
|
|||
|
|
with open(os.path.join(BASE_DIR, f'init_{query_set.model._meta.model_name}.json'), 'w') as f:
|
|||
|
|
json.dump(data, f, indent=4, ensure_ascii=False)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
def add_arguments(self, parser):
|
|||
|
|
parser.add_argument("generate_name", nargs="*", type=str, help="初始化生成的表名")
|
|||
|
|
|
|||
|
|
def generate_users(self):
|
|||
|
|
self.serializer_data(UsersInitSerializer, Users.objects.all())
|
|||
|
|
|
|||
|
|
def generate_role(self):
|
|||
|
|
self.serializer_data(RoleInitSerializer, Role.objects.all())
|
|||
|
|
|
|||
|
|
def generate_dept(self):
|
|||
|
|
self.serializer_data(DeptInitSerializer, Dept.objects.filter(parent_id__isnull=True))
|
|||
|
|
|
|||
|
|
def generate_menu(self):
|
|||
|
|
self.serializer_data(MenuInitSerializer, Menu.objects.filter(parent_id__isnull=True))
|
|||
|
|
|
|||
|
|
def generate_api_white_list(self):
|
|||
|
|
self.serializer_data(ApiWhiteListInitSerializer, ApiWhiteList.objects.all())
|
|||
|
|
|
|||
|
|
def generate_dictionary(self):
|
|||
|
|
self.serializer_data(DictionaryInitSerializer, Dictionary.objects.filter(parent_id__isnull=True))
|
|||
|
|
|
|||
|
|
def generate_system_config(self):
|
|||
|
|
self.serializer_data(SystemConfigInitSerializer, SystemConfig.objects.filter(parent_id__isnull=True))
|
|||
|
|
|
|||
|
|
def handle(self, *args, **options):
|
|||
|
|
generate_name = options.get('generate_name')
|
|||
|
|
generate_name_dict = {
|
|||
|
|
"users": self.generate_users,
|
|||
|
|
"role": self.generate_role,
|
|||
|
|
"dept": self.generate_dept,
|
|||
|
|
"menu": self.generate_menu,
|
|||
|
|
"api_white_list": self.generate_api_white_list,
|
|||
|
|
"dictionary": self.generate_dictionary,
|
|||
|
|
"system_config": self.generate_system_config,
|
|||
|
|
}
|
|||
|
|
if not generate_name:
|
|||
|
|
for ele in generate_name_dict.keys():
|
|||
|
|
generate_name_dict[ele]()
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
for generate_name in generate_name:
|
|||
|
|
if generate_name not in generate_name_dict:
|
|||
|
|
print(f"该初始化方法尚未配置\n{generate_name_dict}")
|
|||
|
|
raise Exception(f"该初始化方法尚未配置,已配置项:{list(generate_name_dict.keys())}")
|
|||
|
|
generate_name_dict[generate_name]()
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
# with open(os.path.join(BASE_DIR, 'temp_init_menu.json')) as f:
|
|||
|
|
# for menu_data in json.load(f):
|
|||
|
|
# menu_data['creator'] = 1
|
|||
|
|
# menu_data['modifier'] = 1
|
|||
|
|
# menu_data['dept_belong_id'] = 1
|
|||
|
|
# request.user = Users.objects.order_by('create_datetime').first()
|
|||
|
|
# serializer = MenuInitSerializer(data=menu_data, request=request)
|
|||
|
|
# serializer.is_valid(raise_exception=True)
|
|||
|
|
# serializer.save()
|
|||
|
|
a = Users.objects.filter()
|
|||
|
|
print(type(Users.objects.filter()))
|