#!/bin/python #coding: utf-8 # +------------------------------------------------------------------- # | django-vue-lyadmin 专业版 # +------------------------------------------------------------------- # | Author: lybbn # +------------------------------------------------------------------- # | QQ: 1042594286 # +------------------------------------------------------------------- # | Date: 2023-11-07 # +------------------------------------------------------------------- # | version: v1.5 # +------------------------------------------------------------------- # ------------------------------ # 表单构建 # ------------------------------ import os import sys import json from django.conf import settings from django.core import management from django.utils import autoreload from mysystem.models import Menu,MenuButton from utils.lyformbuilder.template.lymodels import lyGenerateModels from utils.lyformbuilder.template.lyviewset import lyGenerateViewSet from utils.lyformbuilder.template.lyurls import lyGenerateUrls from utils.lyformbuilder.template.element.indexvue_lytable import lyGenerateIndexVue class GenerateCode: def __init__(self): pass def generate(self,objects,mount=False,menu_instance=None,is_delete_file=False): """ is_delete_file:是否同时删除文件和相关菜单权限等 """ data = [] object = {}#处理后的需要的后台字段 object['formConfig'] = objects['formConfig'] object['widgetList'] = self.filterFormField(objects['widgetList']) if len(object['widgetList']) <1: raise ValueError("暂无可生成的CRUD配置") models = lyGenerateModels(object) data.append({ 'code':models, 'file':'models.py', }) viewset = lyGenerateViewSet(object) data.append({ 'code': viewset, 'file': 'views.py' }) url_route_name = object['formConfig']['modelClassName'] base_url_path = "/api/lyformbuilder/"+url_route_name+"/" api_list = [base_url_path,base_url_path,base_url_path+"{id}/",base_url_path+"{id}/",base_url_path+"{id}/",base_url_path+"export/"] urls = lyGenerateUrls(object) data.append({ 'code': urls, 'file': 'urls.py' }) VueIndexName = 'lyFormBuilder%s'%object['formConfig']['modelClassName'] indexvue = lyGenerateIndexVue(json.dumps(objects),object,VueIndexName) data.append({ 'code': indexvue, 'file': '%s.vue'%VueIndexName }) if not mount:#预览代码 return data if not settings.DEBUG: raise ValueError("当前非debug状态，不支持生成") # modles检测是否存在新生成的目标文件 models_path = self.generate_backend_filepath("models") models_name = "models_%s.py" % objects['class_name'] models_name_old = "models_%s.py" % objects['file_name_old'] self.check_exists_filename(models_path,models_name,objects) # views检测是否存在新生成的目标文件 viewset_path = self.generate_backend_filepath("views") viewset_name = "views_%s.py" % objects['class_name'] viewset_name_old = "views_%s.py" % objects['file_name_old'] self.check_exists_filename(viewset_path,viewset_name, objects) # urls检测是否存在新生成的目标文件 urls_path = self.generate_backend_filepath("urls") urls_name = "urls_%s.py" % objects['class_name'] urls_name_old = "urls_%s.py" % objects['file_name_old'] self.check_exists_filename(urls_path,urls_name, objects) # 前端vue检测是否存在新生成的目标文件 indexvue_path = self.generate_frontend_filepath() indexvue_name = '%s.vue' % VueIndexName indexvue_name_old = "lyFormBuilder%s.vue" % objects['file_name_old'] self.check_exists_filename(indexvue_path,indexvue_name, objects) if is_delete_file: self.delete_menu_button(menu_instance) self.delete_file(urls_path,urls_name) self.delete_file(viewset_path,viewset_name) self.delete_file(models_path,models_name) self.delete_file(indexvue_path,indexvue_name) self.delete_migrations(objects) return True #modles创建文件 self.check_update_filename(models_path,models_name,models_name_old,objects) self.generate_file(models_path,models_name,models) #views创建文件 self.check_update_filename(viewset_path,viewset_name, viewset_name_old, objects) self.generate_file(viewset_path,viewset_name,viewset) #urls创建文件 self.check_update_filename(urls_path,urls_name, urls_name_old, objects) self.generate_file(urls_path,urls_name,urls) #前端vue创建文件 self.check_update_filename(indexvue_path,indexvue_name, indexvue_name_old, objects) self.generate_file(indexvue_path,indexvue_name,indexvue) #配置路由菜单(外层已配置) #配置按钮权限 if menu_instance: self.create_update_menu_button(menu_instance,api_list) # #同步数据库 # self.sync_db() return True #过滤form字段 def filterFormField(self,obj,data_list=None): if data_list is None: data_list = [] for o in obj: if o.get('type',None) == 'grid' and o.get('category',None) == 'container': for c in o['cols']: if c.get('formItemFlag',False) and c.get('options').get('isFormField'): data_list.append(c) continue else: self.filterFormField([c],data_list) elif o.get('type',None) == 'grid-col' and o.get('category',None) == 'container': for c in o['widgetList']: if c.get('formItemFlag',False) and c.get('options').get('isFormField'): data_list.append(c) continue else: self.filterFormField([c],data_list) elif o.get('type',None) == 'table' and o.get('category',None) == 'container': for c in o['rows']: if len(c)>0: for cc in c['cols']: if len(cc)>0: for w in cc['widgetList']: if w.get('formItemFlag',False) and w.get('options').get('isFormField'): data_list.append(w) continue else: self.filterFormField([w],data_list) elif o.get('type',None) == 'tab' and o.get('category',None) == 'container': for c in o['tabs']: if len(c)>0: for w in c['widgetList']: if w.get('formItemFlag',False) and w.get('options').get('isFormField'): data_list.append(w) continue else: self.filterFormField([w],data_list) elif o.get('type',None) == 'card' and o.get('category',None) == 'container': for w in o['widgetList']: if w.get('formItemFlag',False) and w.get('options').get('isFormField'): data_list.append(w) continue else: self.filterFormField([w],data_list) else: if o.get('formItemFlag',False) and o.get('options').get('isFormField'): data_list.append(o) return data_list def generate_backend_filepath(self,moduleName): autoCodePath = os.path.join(settings.BASE_DIR, 'apps', 'lyFormBuilder', moduleName) return autoCodePath def generate_frontend_filepath(self): autoCodePath = os.path.join(os.path.abspath(os.path.join(settings.BASE_DIR,"..")),'frontend','src','views','lyFormBuilderManage') return autoCodePath def check_exists_filename(self,path,filename,object): """ filename 新文件名 filename_old 原文件名 """ if not object['file_name_old']: raise ValueError("field file_name_old error") if not object['class_name'] == object['file_name_old']: if os.path.exists(os.path.join(path, filename)): raise ValueError("检测生成文件错误，已存在%s文件" % filename) def check_update_filename(self,path,filename,filename_old,object): """ filename 新文件名 filename_old 原文件名 """ if not object['file_name_old']: raise ValueError("field file_name_old error") if not object['class_name'] == object['file_name_old']: if os.path.exists(os.path.join(path, filename)): raise ValueError("生成文件错误，已存在%s文件" % filename) if os.path.exists(os.path.join(path, filename_old)): os.rename(os.path.join(path, filename_old),os.path.join(path, filename)) def generate_file(self,path,filename,content): # 创建app文件夹 if not os.path.exists(path): # 如果目标路径不存在就创建 os.makedirs(path) with open(os.path.join(path, filename), 'w', encoding='UTF-8') as f:#w覆盖,a追加 f.write(content) def create_update_menu_button(self,menu,api_list): button_name_list = ["查询", "新增", "编辑", "删除", "单例","导出"] # (0, "GET"),(1, "POST"),(2, "PUT"),(3, "DELETE") crud_button = ['Search','Create','Update','Delete','Retrieve',"Export"] button_name_num_list = [0, 1, 2, 3, 0] menubutton_queryset = MenuButton.objects.filter(menu=menu,name__in=button_name_list) if menubutton_queryset.count() == 0: intance_list = [] for index,value in enumerate(button_name_list): obj = MenuButton(menu=menu, name=value, value=crud_button[index],method=button_name_num_list[index],api=api_list[index]) intance_list.append(obj) MenuButton.objects.bulk_create(intance_list) else: for m in menubutton_queryset: for i,c in enumerate(crud_button): if m.value == c: m.api = api_list[i] m.save() #删除菜单、按钮、权限 def delete_menu_button(self,menu): print('删除文件>>> 删除菜单、按钮、权限') if menu: Menu.objects.filter(id=menu.id).delete() def delete_file(self,path,filename): try: print('删除文件>>> ', filename) os.remove(os.path.join(path, filename)) except: pass def get_file_list(self,directory): return [f for f in os.listdir(directory) if os.path.isfile(os.path.join(directory, f))] #删除迁移相关 def delete_migrations(self,object): class_name = object['class_name'] db_table = object['formConfig']['modelDbTable'] lower_class_name = class_name.lower() contains_str = "_"+lower_class_name from django.db import connection with connection.cursor() as cursor: print('删除表和迁移数据>>> ', db_table) cursor.execute("DELETE FROM django_migrations WHERE app='lyFormBuilder' AND name LIKE '%s'"%("%"+contains_str+"%")) cursor.execute("DROP TABLE IF EXISTS %s"%db_table) migrationsPath = self.generate_backend_filepath("migrations") files = self.get_file_list(migrationsPath) for file_k in files: if file_k != '__init__.py' and contains_str in file_k: dst_file = os.path.join(migrationsPath, file_k) try: print('删除文件>>> ', dst_file) os.remove(dst_file) except: pass #重载 def reload_server(self): autoreload.restart_with_reloader() def sync_db(self): if settings.DEBUG: os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'application.settings') try: from django.core.management import execute_from_command_line except ImportError as exc: raise ImportError( "Couldn't import Django. Are you sure it's installed and " "available on your PYTHONPATH environment variable? Did you " "forget to activate a virtual environment?" ) from exc execute_from_command_line(["manage.py","makemigrations","--noinput",'lyFormBuilder']) execute_from_command_line(["manage.py","migrate","--noinput","lyFormBuilder"]) def syncdb(self): try: management.call_command("makemigrations","--noinput",'lyFormBuilder') management.call_command("migrate","--noinput","lyFormBuilder") return True except Exception as e: print("表单构建【同步数据库】出现异常，异常内容：%s"%e) return False