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)