新功能: 添加后端代码

backend/dvadmin/system/management/commands/generate_init_json.py

@@ -0,0 +1,100 @@
+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()))

backend/dvadmin/system/management/commands/init.py

@@ -0,0 +1,53 @@
+import logging
+
+from django.core.management.base import BaseCommand
+
+from application import settings
+
+logger = logging.getLogger(__name__)
+
+
+class Command(BaseCommand):
+    """
+    项目初始化命令: python manage.py init
+    """
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "init_name",
+            nargs="*",
+            type=str,
+        )
+        parser.add_argument("-y", nargs="*")
+        parser.add_argument("-Y", nargs="*")
+        parser.add_argument("-n", nargs="*")
+        parser.add_argument("-N", nargs="*")
+
+    def handle(self, *args, **options):
+        reset = False
+        if isinstance(options.get("y"), list) or isinstance(options.get("Y"), list):
+            reset = True
+        if isinstance(options.get("n"), list) or isinstance(options.get("N"), list):
+            reset = False
+
+        for app in settings.INSTALLED_APPS:
+
+            try:
+                exec(
+                    f"""
+from {app}.fixtures.initialize import Initialize
+Initialize(reset={reset},app="{app}").run()
+                """
+                )
+            except ModuleNotFoundError:
+                # 兼容之前版本初始化
+                try:
+                    exec(
+                        f"""
+from {app}.initialize import main
+main(reset={reset})
+                """
+                    )
+                except ModuleNotFoundError:
+                    pass
+        print("初始化数据完成！")

backend/dvadmin/system/management/commands/init_area.py

@@ -0,0 +1,83 @@
+# 城市联动
+"""
+到乡级 使用方法
+1. https://www.npmjs.com/package/china-division 下载数据，把对应的json放入对应目录
+2. 修改此文件中对应json名
+3. 右击执行此py文件进行初始化
+"""
+import json
+import os
+
+import django
+import pypinyin
+from django.core.management import BaseCommand
+from django.db import connection
+
+from application import dispatch
+
+os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'application.settings')
+django.setup()
+from application.settings import BASE_DIR
+from dvadmin.system.models import Area
+
+area_code_list = []
+
+
+def area_list(code_list, pcode=None, depth=1):
+    """
+    递归获取所有列表
+    """
+    for code_dict in code_list:
+        code = code_dict.get('code', None)
+        name = code_dict.get('name', None)
+        children = code_dict.get('children', None)
+        pinyin = ''.join([''.join(i) for i in pypinyin.pinyin(name, style=pypinyin.NORMAL)])
+        area_code_list.append(
+            {
+                "name": name,
+                "code": code,
+                "level": depth,
+                "pinyin": pinyin,
+                "initials": pinyin[0].upper() if pinyin else "#",
+                "pcode_id": pcode,
+            }
+        )
+        if children:
+            area_list(code_list=children, pcode=code, depth=depth + 1)
+
+
+def main():
+    with open(os.path.join(BASE_DIR, 'dvadmin', 'system', 'util', 'pca-code.json'), 'r', encoding="utf-8") as load_f:
+        code_list = json.load(load_f)
+    area_list(code_list)
+    if Area.objects.count() == 0:
+        Area.objects.bulk_create([Area(**ele) for ele in area_code_list])
+    else:
+        for ele in area_code_list:
+            code = ele.pop("code")
+            Area.objects.update_or_create(code=code, defaults=ele)
+
+
+class Command(BaseCommand):
+    """
+    项目初始化命令: python manage.py init
+    """
+
+    def add_arguments(self, parser):
+        pass
+
+    def handle(self, *args, **options):
+
+        print(f"正在准备初始化省份数据...")
+
+        if dispatch.is_tenants_mode():
+            from django_tenants.utils import get_tenant_model
+            from django_tenants.utils import tenant_context
+            for tenant in get_tenant_model().objects.exclude(schema_name='public'):
+                with tenant_context(tenant):
+                    print(f"租户[{connection.tenant.schema_name}]初始化数据开始...")
+                    main()
+                    print(f"租户[{connection.tenant.schema_name}]初始化数据完成！")
+        else:
+            main()
+        print("省份数据初始化数据完成！")