first commit

code/system/management/commands/generator.py

@@ -0,0 +1,260 @@
+import json
+import logging
+import os
+import shutil
+
+from django.core.management.base import BaseCommand
+
+from fuadmin.settings import BASE_DIR
+
+logger = logging.getLogger(__name__)
+
+
+class Command(BaseCommand):
+    """
+    创建App命令:
+    python manage.py createapp app名
+    python manage.py createapp app01 app02 ...
+    python manage.py createapp 一级文件名/app01 ...    # 支持多级目录建app
+    """
+
+    def add_arguments(self, parser):
+        parser.add_argument('app_info', nargs='*', type=str, )
+
+    def handle(self, *args, **options):
+        app_info = options.get('app_info')
+        for name in app_info:
+            app = json.loads(name)
+            name = app.get('app_name')
+            names = name.split('/')
+            dnames = ".".join(names)
+            app_path = os.path.join(BASE_DIR, "dvadmin", *names)
+            # 判断app是否存在
+            if os.path.exists(app_path):
+                print(f"App {name} 已存在！")
+                # break
+            else:
+                source_path = os.path.join(BASE_DIR, "dvadmin", "utils", "template")
+                target_path = app_path
+                if not os.path.exists(target_path):
+                    # 如果目标路径不存在原文件夹的话就创建
+                    os.makedirs(target_path)
+                if os.path.exists(source_path):
+                    # 如果目标路径存在原文件夹的话就先删除
+                    shutil.rmtree(target_path)
+                shutil.copytree(source_path, target_path)
+                # 注册app到 settings.py 中
+                injection(os.path.join(BASE_DIR, "application", "settings.py"), f"    'dvadmin.{dnames}',\n",
+                          "INSTALLED_APPS",
+                          "]")
+
+                # 注册app到 urls.py 中
+                injection(os.path.join(BASE_DIR, "application", "urls.py"),
+                          f"    path(r'api/{name}/', include('dvadmin.{dnames}.urls')),\n", "urlpatterns = [",
+                          "]")
+
+            # 修改app中的apps 内容
+            app_content = f"""
+from django.apps import AppConfig
+
+
+class {name.capitalize()}Config(AppConfig):
+    name = 'dvadmin.{dnames}'
+    verbose_name = "{names[-1]}App"
+"""
+            with open(os.path.join(app_path, "apps.py"), 'w', encoding='UTF-8') as f:
+                f.write(app_content)
+
+            # 修改Model的内容
+
+            table_name = app.get('table_name')
+            model_name = app.get('model_name')
+
+            # app中所有不重复的字段
+            app_fields = app.get('fields')
+            field_content = ''
+            # app中所有不重复的字段type
+            fields_type = set()
+            # app中所有的字段name
+            fields_name = set()
+            for field in app_fields:
+                fields_type.add(field['type'])
+                fields_name.add(field['name'])
+                if field['type'] == 'CharField':
+                    field_content = f"""{field_content}
+    {field['name']} = {field['type']}(max_length={field['max_length']}, verbose_name='{field['name']}', help_text='{field['description']}')"""
+                elif field['type'] == 'TextField':
+                    field_content = f"""{field_content}
+    {field['name']} = {field['type']}(verbose_name='{field['name']}', help_text='{field['description']}')"""
+            model_content = f"""
+from django.db.models import {','.join(fields_type)}
+from dvadmin.utils.models import CoreModel
+
+
+class {model_name}(CoreModel):
+{field_content}
+
+    class Meta:
+        verbose_name = '{model_name}'
+        verbose_name_plural = verbose_name
+    
+    def __str__(self):
+        return self.{fields_name.pop()} 
+"""
+
+            with open(os.path.join(app_path, "models", f"{table_name}.py"), 'w', encoding='UTF-8') as f:
+                f.write(model_content)
+
+            # 修改 model init文件
+            model_init_content = f"from dvadmin.{name}.models.{table_name} import {model_name}"
+            with open(os.path.join(app_path, "models", "__init__.py"), 'a', encoding='UTF-8') as f:
+                f.write(model_init_content)
+                f.write('\n')
+
+            # 修改filters的内容
+            filter_content = f"""
+import django_filters
+from dvadmin.{name}.models import {model_name}
+
+
+class {model_name}Filter(django_filters.rest_framework.FilterSet):
+
+    # 通过 lookup_expr 可进行模糊查询，其他配置可自行百度
+    # name = django_filters.CharFilter(lookup_expr='icontains')
+
+    class Meta:
+        model = {model_name}
+        fields = '__all__'
+
+"""
+            with open(os.path.join(app_path, "filters", f"{table_name}_filter.py"), 'w', encoding='UTF-8') as f:
+                f.write(filter_content)
+
+            # 修改 filters init文件
+            filter_init_content = f"from .{table_name}_filter import {model_name}Filter"
+            with open(os.path.join(app_path, "filters", "__init__.py"), 'a', encoding='UTF-8') as f:
+                f.write(filter_init_content)
+                f.write('\n')
+
+            # 修改serializers的内容
+            serializer_content = f"""
+from rest_framework import serializers
+from dvadmin.{name}.models import {model_name}
+from dvadmin.utils.serializers import CustomModelSerializer
+
+
+class {model_name}Serializer(CustomModelSerializer):
+
+    class Meta:
+        model = {model_name}
+        fields = '__all__'
+        
+        
+class {model_name}CreateUpdateSerializer(CustomModelSerializer):
+
+    # 此处可写定制的 创建/更新 内容
+    def validate(self, attrs: dict):
+        return super().validate(attrs)
+
+    class Meta:
+        model = {model_name}
+        fields = '__all__'
+
+
+class {model_name}ImportSerializer(CustomModelSerializer):
+
+    class Meta:
+        model = {model_name}
+        fields = '__all__'
+
+
+class {model_name}ExportSerializer(CustomModelSerializer):
+
+    class Meta:
+        model = {model_name}
+        fields = '__all__'
+
+"""
+            with open(os.path.join(app_path, 'serializers', f'{table_name}_serializer.py'), 'w', encoding='UTF-8') as f:
+                f.write(serializer_content)
+
+            # 修改 serializers init文件
+            serializer_init_content = f"from .{table_name}_serializer import {model_name}Serializer, " \
+                                      f"{model_name}CreateUpdateSerializer, " \
+                                      f"{model_name}ImportSerializer, " \
+                                      f"{model_name}ExportSerializer"
+            with open(os.path.join(app_path, "serializers", "__init__.py"), 'a', encoding='UTF-8') as f:
+                f.write(serializer_init_content)
+                f.write('\n')
+            # 修改view内容
+            view_content = f"""
+from dvadmin.utils.filters import DataLevelPermissionsFilter
+from dvadmin.utils.viewset import CustomModelViewSet
+from dvadmin.{name}.models import {model_name}
+from dvadmin.{name}.filters import {model_name}Filter
+from dvadmin.{name}.serializers import {model_name}Serializer, {model_name}CreateUpdateSerializer, {model_name}ImportSerializer, {model_name}ExportSerializer
+                            
+          
+class {model_name}ModelViewSet(CustomModelViewSet):
+
+    queryset = {model_name}.objects.all()
+    serializer_class = {model_name}Serializer  # 序列化器
+    create_serializer_class = {model_name}CreateUpdateSerializer  # 创建/更新时的列化器
+    update_serializer_class = {model_name}CreateUpdateSerializer  # 创建/更新时的列化器
+    filter_class = {model_name}Filter  # 过滤器
+    extra_filter_backends = [DataLevelPermissionsFilter]  # 数据权限类，不需要可注释掉
+    # ordering = ['create_datetime']  # 默认排序
+    # 导出
+    export_field_data = []  # 导出
+    export_serializer_class = {model_name}ExportSerializer  # 导出序列化器
+    # 导入
+    import_field_data = []
+    import_serializer_class = {model_name}ImportSerializer
+
+
+"""
+            with open(os.path.join(app_path, 'views', f'{table_name}_view.py'), 'w', encoding='UTF-8') as f:
+                f.write(view_content)
+
+            # 修改 views init文件
+            view_init_content = f"from .{table_name}_view import {model_name}ModelViewSet"
+            with open(os.path.join(app_path, "views", "__init__.py"), 'a', encoding='UTF-8') as f:
+                f.write(view_init_content)
+                f.write('\n')
+
+            # 修改 url文件
+            url_package = f"from dvadmin.{name}.views import {model_name}ModelViewSet"
+            injection(os.path.join(app_path, "urls.py"),
+                      f"{url_package}\n", "from rest_framework.routers import DefaultRouter",
+                      "router = DefaultRouter()")
+
+            url_path = f"router.register(r'{table_name}', {model_name}ModelViewSet)"
+            injection(os.path.join(app_path, "urls.py"),
+                      f"{url_path}\n", "router = DefaultRouter()",
+                      "urlpatterns = [")
+
+            export_import_url = f"""    # 导出项目
+    path('{table_name}/export/', {model_name}ModelViewSet.as_view({{'get': 'export', }})),
+    # 项目导入模板下载及导入
+    path('{table_name}/importTemplate/', {model_name}ModelViewSet.as_view({{'get': 'importTemplate', 'post': 'importTemplate'}}))"""
+            injection(os.path.join(app_path, "urls.py"),
+                      f"{export_import_url}\n", "urlpatterns = [", "]")
+
+            print(f"创建 {name} App成功")
+
+
+def injection(file_path, insert_content, startswith, endswith):
+    with open(file_path, "r+", encoding="utf-8") as f:
+        data = f.readlines()
+        with open(file_path, 'w', encoding='UTF-8') as f1:
+            is_INSTALLED_APPS = False
+            is_insert = False
+            for content in data:
+                # 判断文件是否 INSTALLED_APPS 开头
+                if not is_insert and content.startswith(startswith):
+                    is_INSTALLED_APPS = True
+                if not is_insert and content.startswith(endswith) and is_INSTALLED_APPS:
+                    # 给前一行插入数据
+                    content = insert_content + content
+                    is_insert = True
+                f1.writelines(content)

code/system/management/commands/init.py

@@ -0,0 +1,39 @@
+import logging
+
+from django.core.management.base import BaseCommand
+
+from fuadmin 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
+        print(f"正在准备初始化数据，{'如有初始化数据，将会不做操作跳过' if not reset else '初始数据将会先删除后新增'}...")
+
+        for app in settings.INSTALLED_APPS:
+
+            try:
+                exec(f"""
+from {app}.initialize import main
+main(reset={reset})
+                """)
+            except ModuleNotFoundError:
+                pass
+        print("初始化数据完成！")

code/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 fuadmin.settings import BASE_DIR
+from system.models import Area
+
+os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'application.settings')
+django.setup()
+
+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, 'utils', '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 hasattr(connection, 'tenant') and connection.tenant.schema_name:
+            from django_tenants.utils import get_tenant_model
+            from django_tenants.utils import tenant_context,schema_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("省份数据初始化数据完成！")
+

code/system/management/commands/vdes_monitor.py

@@ -0,0 +1,42 @@
+import time
+import logging
+from django.core.management.base import BaseCommand
+from system.services.vdes_data_receive import VDESSocketMonitor
+
+logger = logging.getLogger(__name__)
+
+class Command(BaseCommand):
+    help = '控制 VDES 监控服务'
+    
+    def add_arguments(self, parser):
+        parser.add_argument(
+            'action',
+            choices=['start', 'stop', 'restart'],
+            help='操作: start, stop 或 restart'
+        )
+    
+    def handle(self, *args, **options):
+        monitor = VDESSocketMonitor()
+        action = options['action']
+        
+        if action == 'start':
+            monitor.start()
+            self.stdout.write(self.style.SUCCESS('VDES 监控服务已启动'))
+        elif action == 'stop':
+            monitor.stop()
+            self.stdout.write(self.style.SUCCESS('VDES 监控服务已停止'))
+        elif action == 'restart':
+            monitor.stop()
+            time.sleep(1)
+            monitor.start()
+            self.stdout.write(self.style.SUCCESS('VDES 监控服务已重启'))
+
+                # 确保程序不会立即退出
+        if action == 'start' or action == 'restart':
+            try:
+                # 保持命令运行，直到用户中断
+                while True:
+                    time.sleep(1)
+            except KeyboardInterrupt:
+                monitor.stop()
+                self.stdout.write(self.style.SUCCESS('\nVDES 监控服务已停止'))