def gen_serializer(project_name_settings): init_django_env(project_name_settings) import django model_list = [] model_fk_dict = {} app_model_import_dict = {} sys_label = ['admin', 'auth', 'contenttypes', 'sessions', 'captcha', 'xadmin', 'xadmin_api', 'authtoken', 'social_django'] for one in django.apps.apps.get_models(): columns = [] model_name = one._meta.model.__name__ model_ver_name = one._meta.verbose_name app_label = one._meta.app_label if app_label not in sys_label: fk_field_list = [] try: app_model_import_dict[app_label].append(model_name) except KeyError: app_model_import_dict[app_label] = [model_name] for filed in one.objects.model._meta.fields: name = filed.name ver_name = filed.verbose_name if isinstance(filed, ForeignKey): help_text = filed.help_text.replace(MAIN_DISPLAY + "__", "") fk_field_list.append(name + "$分割$" + help_text) model_fk_dict[model_name] = fk_field_list model_list.append(model_name) # status_text = serializers.CharField(source="status.text", read_only=True) print(model_fk_dict) app_name = "xadmin_api" serializers_txt = f""" from rest_framework import serializers $model_import占位$ """ model_import_rows = [] for app, m_list in app_model_import_dict.items(): txt = f'from {app}.models import {", ".join(m_list)}\n' model_import_rows.append(txt) serializers_txt = serializers_txt.replace("$model_import占位$", "".join(model_import_rows)) print(model_fk_dict) for model, fk_field_l in model_fk_dict.items(): fk_display_p = [] for one_fk in fk_field_l: fk_name, fk_text = one_fk.split("$分割$") fk_one_line = f'{fk_name}_text = serializers.CharField(source="{fk_name}.{fk_text}", read_only=True)\n' fk_display_p.append(fk_one_line) serializers_txt += f""" class {model}Serializer(serializers.ModelSerializer): {" ".join(fk_display_p)} class Meta: model = {model} fields = "__all__" """ # if os.path.exists('../xadmin_api/auto_serializers.py'): print("已存在serializers跳过") else: with open('../xadmin_api/auto_serializers.py', 'w') as fw: fw.write(serializers_txt)
def migration_from_xadmin(project_name_settings): init_django_env(project_name_settings) import xadmin for one, item in xadmin.site._registry.items(): print(one._meta.object_name) print(item.list_display) print(item.list_filter) print(item.search_fields)
def gen_table_display(project_name_settings, list_display, focus_model): init_django_env(project_name_settings) import django for one in django.apps.apps.get_models(): if one._meta.model.__name__ == focus_model: print("{") for filed in one.objects.model._meta.fields: if filed.name in list_display: pass else: print("""%s: { show: false, },""" % (filed.name)) print("}")
def gen_route(project_name_settings): init_django_env(project_name_settings) import django path_list = [] sys_label = ['admin', 'auth', 'contenttypes', 'sessions', 'captcha', 'xadmin', 'xadmin_api', 'authtoken', 'social_django'] for one in django.apps.apps.get_models(): if one._meta.app_label not in sys_label: url = "/xadmin/" + get_lower_case_name(one._meta.model.__name__) one = "{" + f""" name: '{one._meta.verbose_name}', icon: 'smile', path: '{url}', component: './AutoGenPage/{one._meta.model.__name__}List', """ + "}" # print(one) path_list.append(one) with open('../xadmin_api/menu.json', 'w') as fw: fw.write("["+",".join(path_list)+"]")
def gen_url(project_name_settings): init_django_env(project_name_settings) import django model_list = [] model_fk_dict = {} app_model_import_dict = {} sys_label = [ 'admin', 'auth', 'contenttypes', 'sessions', 'captcha', 'xadmin', 'xadmin_api', 'authtoken', 'social_django' ] for one in django.apps.apps.get_models(): columns = [] model_name = one._meta.model.__name__ model_ver_name = one._meta.verbose_name app_label = one._meta.app_label if app_label not in sys_label: model_list.append(model_name) url_txt = f"""from xadmin_api import auto_views from django.urls import re_path, include, path from rest_framework.routers import DefaultRouter router = DefaultRouter(trailing_slash=False) """ for model_name in model_list: url_txt += f""" router.register('{get_lower_case_name(model_name)}/?', auto_views.{model_name}ViewSet) """ url_txt += """ urlpatterns = [ re_path('^', include(router.urls)), ] """ if os.path.exists('../xadmin_api/auto_url.py'): print("已存在urls跳过") else: with open('../xadmin_api/auto_url.py', 'w') as fw: fw.write(url_txt)
def gen_filter(project_name_settings): init_django_env(project_name_settings) import django model_list = [] model_pic_dict = {} model_fk_dict = {} model_date_dict = {} app_model_import_dict = {} sys_label = [ 'admin', 'auth', 'contenttypes', 'sessions', 'captcha', 'xadmin', 'xadmin_api', 'authtoken', 'social_django' ] for one in django.apps.apps.get_models(): columns = [] model_name = one._meta.model.__name__ model_ver_name = one._meta.verbose_name app_label = one._meta.app_label if app_label not in sys_label: try: app_model_import_dict[app_label].append(model_name) except KeyError: app_model_import_dict[app_label] = [model_name] img_field_list = [] fk_field_list = [] date_field_list = [] for filed in one.objects.model._meta.fields: name = filed.name ver_name = filed.verbose_name if isinstance(filed, ImageField): img_field_list.append('"' + name + '"') if isinstance(filed, FileField): img_field_list.append('"' + name + '"') if isinstance(filed, ForeignKey): help_text = filed.help_text.replace( MAIN_DISPLAY + "__", "") fk_field_list.append(name + "$分割$" + help_text) if isinstance(filed, DateTimeField): date_field_list.append(name) model_pic_dict[model_name] = img_field_list model_fk_dict[model_name] = fk_field_list model_date_dict[model_name] = date_field_list model_list.append(model_name) app_name = "xadmin_api" filters_txt = f""" from django_filters import rest_framework as filters from xadmin_api.custom import DateFromToRangeFilter $model_import占位$ """ model_import_rows = [] for app, m_list in app_model_import_dict.items(): txt = f'from {app}.models import {", ".join(m_list)}\n' model_import_rows.append(txt) filters_txt = filters_txt.replace("$model_import占位$", "".join(model_import_rows)) for (model, img_field_l), (model_2, fk_field_l), (model_3, date_field_l) in zip( model_pic_dict.items(), model_fk_dict.items(), model_date_dict.items()): fk_display_p = [] for one_fk in fk_field_l: fk_name, fk_text = one_fk.split("$分割$") fk_one_line = f' {fk_name}_text = filters.CharFilter(field_name="{fk_name}")\n' fk_display_p.append(fk_one_line) date_range_p = [] for one_date in date_field_l: date_one_line = f' {one_date} = DateFromToRangeFilter(field_name="{one_date}")\n' date_range_p.append(date_one_line) filters_txt += f""" class {model}Filter(filters.FilterSet): {"".join(fk_display_p)}{"".join(date_range_p)} class Meta: model = {model} exclude = [{",".join(img_field_l)}] """ if os.path.exists('../xadmin_api/auto_filters.py'): print("已存在filters.py跳过") else: with open('../xadmin_api/auto_filters.py', 'w') as fw: fw.write(filters_txt)
def gen_view(project_name_settings): init_django_env(project_name_settings) import django model_list = [] app_name = "xadmin_api" model_search_dict = {} app_model_import_dict = {} sys_label = [ 'admin', 'auth', 'contenttypes', 'sessions', 'captcha', 'xadmin', 'xadmin_api', 'authtoken', 'social_django' ] for one in django.apps.apps.get_models(): model_name = one._meta.model.__name__ model_ver_name = one._meta.verbose_name app_label = one._meta.app_label if app_label not in sys_label: try: app_model_import_dict[app_label].append(model_name) except KeyError: app_model_import_dict[app_label] = [model_name] search_list = [] model_list.append(model_name) for filed in one.objects.model._meta.fields: name = filed.name ver_name = filed.verbose_name if isinstance(filed, CharField): search_list.append('"' + name + '"') elif isinstance(filed, CharField): search_list.append('"' + name + '"') model_search_dict[model_name] = search_list serializers_list = [one + "Serializer" for one in model_list] filters_list = [one + "Filter" for one in model_list] viewset_txt = f""" from rest_framework import viewsets from xadmin_api.custom import XadminViewSet $model_import占位$ from {app_name}.auto_serializers import {", ".join(serializers_list)} from {app_name}.auto_filters import {", ".join(filters_list)} """ model_import_rows = [] print(app_model_import_dict) for app, m_list in app_model_import_dict.items(): txt = f'from {app}.models import {", ".join(m_list)}\n' model_import_rows.append(txt) viewset_txt = viewset_txt.replace("$model_import占位$", "".join(model_import_rows)) for model_name in model_list: viewset_txt += f""" class {model_name}ViewSet(XadminViewSet): serializer_class = {model_name}Serializer queryset = {model_name}.objects.all() filter_class = {model_name}Filter search_fields = [{",".join(model_search_dict[model_name])}] """ if os.path.exists('../xadmin_api/auto_views.py'): print("已存在views跳过") else: with open('../xadmin_api/auto_views.py', 'w') as fw: fw.write(viewset_txt)
def gen_antd_pages(project_name_settings, focus_model=None, template_type="base"): init_django_env(project_name_settings) import django sys_label = [ 'admin', 'auth', 'contenttypes', 'sessions', 'captcha', 'xadmin', 'xadmin_api', 'authtoken', 'social_django' ] # focus_model = "CourseResource" model_pic_dict = {} model_date_dict = {} user = get_user_model() for one in django.apps.apps.get_models(): columns = [] model_name = one._meta.model.__name__ model_ver_name = one._meta.verbose_name if focus_model and model_name != focus_model: continue if one._meta.app_label not in sys_label: img_field_list = [] date_row_list = [] model_add_item_list = [] model_many_to_many_list = [] model_import_item_list = [] date_field_list = [] fileds_num = len(one.objects.model._meta.fields) for filed in one.objects.model._meta.fields: name = filed.name ver_name = filed.verbose_name if isinstance(filed, ForeignKey): rela_model = filed.related_model._meta.object_name help_text = filed.help_text.replace( MAIN_DISPLAY + "__", "") one_c = """{ title: '%s', dataIndex: '%s', backendType: 'foreignKey', rules: [ { required: true, message: '%s为必填项', }, ], renderFormItem: (item, {value, onChange}) => { const children = $模型名字$ForeignKeyList.map((item) => { return <Option key={item.id} value={item.id}>{item.$外键显示字段$}</Option>; }); return <Select placeholder="请选择$模型显示名字$" onChange={onChange} > {children} </Select>; }, },""" % (ver_name, name, ver_name) one_c = one_c.replace("$模型名字$", name) one_c = one_c.replace("$外键显示字段$", help_text) one_c = one_c.replace("$模型显示名字$", filed.verbose_name) model_add_item = """const [$模型名字$ForeignKeyList, set$模型名字首字母大写$ForeignKeyList] = useState([]); useEffect(() => { query$关联Model$().then(value => { set$模型名字首字母大写$ForeignKeyList(value.data); }); }, []);""".replace("$模型名字$", name).replace("$模型名字首字母大写$", name[0].upper() + name[1:]).replace( "$关联Model$", rela_model) model_add_item_list.append(model_add_item) model_import_item = """import {query$关联Model$} from '@/pages/AutoGenPage//$关联Model$List/service';""".replace( "$关联Model$", rela_model) model_import_item_list.append(model_import_item) elif isinstance(filed, CharField): if filed.choices: filed_choices_list = [] for filed_one in filed.choices: one_line = f'{filed_one[0]}:"{filed_one[1]}"' filed_choices_list.append(one_line) one_c = """{ title: '%s', dataIndex: '%s', rules: [ { required: true, message: '%s为必填项', }, ], valueEnum: { $valueEnum$ }, },""" % (ver_name, name, ver_name) one_c = one_c.replace("$valueEnum$", ",".join(filed_choices_list)) elif name == "password": one_c = """{ title: '%s', dataIndex: '%s', hideInTable: true, hideInForm: true, hideInSearch: true, rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) else: one_c = """{ title: '%s', dataIndex: '%s', rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) elif name == "id": one_c = """{ title: '%s', dataIndex: '%s', hideInForm: true, hideInSearch: true, rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) elif isinstance(filed, DateTimeField): print(filed.default) hideInform = None if filed.default != django.db.models.NOT_PROVIDED or filed.auto_now == True: hideInform = True one_c = """{ title: '%s', dataIndex: '%s', valueType: 'dateTime', $隐藏form$ rules: [ { required: true, message: '%s为必填项', }, ], render: (text, record) => { return dealDateTimeDisplay(text); }, },""" % (ver_name, name, ver_name) if hideInform: one_c = one_c.replace("$隐藏form$", "hideInForm: true,") else: one_c = one_c.replace("$隐藏form$", "") date_field_list.append('"' + name + '"') date_row_list.append( f'record.{name} = moment(record.{name});') elif isinstance(filed, DateField): if filed.auto_now == True: pass one_c = """{ title: '%s', dataIndex: '%s', valueType: 'date', rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) date_field_list.append('"' + name + '"') date_row_list.append( f'record.{name} = moment(record.{name});') elif isinstance(filed, BooleanField): one_c = """{ title: '%s', dataIndex: '%s', rules: [ { required: true, message: '%s为必填项', }, ], render: (text, record) => { return BooleanDisplay(text); }, renderFormItem: (_, {type, defaultRender, ...rest}, form) => { const is_value = form.getFieldValue('%s'); if (type === "form" && !is_value) { form.setFieldsValue({'%s': false}); } return <Switch checked={is_value} onClick={(checked) => { form.setFieldsValue({'%s': checked}); }} />; }, },""" % (ver_name, name, ver_name, name, name, name) elif isinstance(filed, IntegerField) or isinstance( filed, FloatField): if filed.choices: filed_choices_list = [] for filed_one in filed.choices: one_line = f'{filed_one[0]}:"{filed_one[1]}"' filed_choices_list.append(one_line) one_c = """{ title: '%s', dataIndex: '%s', rules: [ { required: true, message: '%s为必填项', }, ], valueEnum: { $valueEnum$ }, },""" % (ver_name, name, ver_name) one_c = one_c.replace("$valueEnum$", ",".join(filed_choices_list)) else: one_c = """{ title: '%s', dataIndex: '%s', valueType: 'digit', rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) elif isinstance(filed, ImageField) or isinstance( filed, SImageField): img_field_list.append('"' + name + '"') help_text = filed.help_text if help_text == MAIN_AVATAR: one_c = """{ title: '%s', dataIndex: '%s', valueType: 'avatar', hideInSearch: true, rules: [ { required: true, message: '%s为必填项', }, ], renderFormItem: (_, {type, defaultRender, ...rest}, form) => { const imageUrl = form.getFieldValue('%s'); return <UploadAvatar img={imageUrl}/> }, },""" % (ver_name, name, ver_name, name) elif help_text == MAIN_PIC: one_c = """{ title: '%s', dataIndex: '%s', render: (text, record) => { return <img src={text} width="80px" alt=""/> }, hideInSearch: true, rules: [ { required: true, message: '%s为必填项', }, ], renderFormItem: (_, {type, defaultRender, ...rest}, form) => { const imageUrl = form.getFieldValue('%s'); return <UploadAvatar img={imageUrl}/> }, },""" % (ver_name, name, ver_name, name) elif isinstance(filed, FileField): img_field_list.append('"' + name + '"') one_c = """{ title: '%s', dataIndex: '%s', hideInSearch: true, rules: [ { required: true, message: '%s为必填项', }, ], render: (text, record) => { return <a download={text.split('/').slice(-1)} href={text}>{text.split('/').slice(-1)}</a>; }, renderFormItem: (_, {type, defaultRender, ...rest}, form) => { const downloadUrl = form.getFieldValue('download'); return fileUpload(downloadUrl); }, },""" % ( ver_name, name, ver_name) elif filed.__class__.__name__ == "TextField": one_c = """{ title: '%s', dataIndex: '%s', valueType: 'textarea', ellipsis: true, rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) elif filed.__class__.__name__ == "UEditorField" or isinstance( filed, richTextField): one_c = """{ title: '%s', dataIndex: '%s', customCol:richCol, ellipsis: true, hideInSearch: true, rules: [ { required: true, message: '%s为必填项', }, ], renderFormItem: (_, {type, defaultRender, ...rest}, form) => { return richForm(form, rest.id); }, },""" % (ver_name, name, ver_name) else: print(filed.__class__.__name__) one_c = """{ title: '%s', dataIndex: '%s', rules: [ { required: true, message: '%s为必填项', }, ], },""" % (ver_name, name, ver_name) model_pic_dict[model_name] = img_field_list model_date_dict[model_name] = date_field_list columns.append(one_c) for filed in one.objects.model._meta.many_to_many: black_many_to_many = [ "groups", "permissions", "user_permissions" ] if one._meta.model == user and filed.name in black_many_to_many: continue else: name = filed.name ver_name = filed.verbose_name rela_model = filed.related_model._meta.object_name help_text = filed.help_text.replace( MAIN_DISPLAY + "__", "") one_c = """{ title: '%s', dataIndex: '%s', rules: [ { required: true, message: '%s为必填项', }, ], renderFormItem: (item, {value, onChange}) => { const children = %sManyToManyList.map(item => { return ( <Option key={item.id} value={item.id}> {item.%s} </Option> ); }); return ( <Select mode="multiple" placeholder="请选择%s" onChange={onChange}> {children} </Select> ); }, render: (text, record) => { const color_arr = [ 'green', 'cyan', 'blue', 'geekblue', 'purple', 'magenta', 'red', 'volcano', 'orange', 'gold', 'lime', ]; const child = []; text.forEach((value, index, arr) => { child.push(<Tag color={color_arr[value $百分$ 10]}>{%sManyToManyMap[value]}</Tag>); }); return <Space>{child}</Space>; }, },""" % (ver_name, name, ver_name, name, help_text, ver_name, name) one_c = one_c.replace("$百分$", "%") model_many_to_many_item = """const [$模型名字$ManyToManyList, set$模型名字首字母大写$ManyToManyList] = useState([]); const [$模型名字$ManyToManyMap, set$模型名字首字母大写$ManyToManyMap] = useState([]); useEffect(() => { query$关联Model$().then(value => { set$模型名字首字母大写$ManyToManyList(value.data); let get$模型名字首字母大写$ManyToManyMap = {}; for (let index in value.data) { let item = value.data[index]; get$模型名字首字母大写$ManyToManyMap[item.id.toString()] = item.$关联字段$; } set$模型名字首字母大写$ManyToManyMap(get$模型名字首字母大写$ManyToManyMap); }); }, []);""".replace("$模型名字$", name).replace("$模型名字首字母大写$", name[0].upper() + name[1:]) \ .replace("$关联Model$", rela_model).replace("$关联字段$", help_text) model_many_to_many_list.append(model_many_to_many_item) model_import_item = """import {query$关联Model$} from '@/pages/AutoGenPage//$关联Model$List/service';""".replace( "$关联Model$", rela_model) model_import_item_list.append(model_import_item) columns.append(one_c) opera = """ { title: '操作', dataIndex: 'option', valueType: 'option', fixed: 'right', width: 100, render: (text, record) => ( <> <EditOutlined title="编辑" className="icon" onClick={async () => { $时间处理占位$ handleUpdateModalVisible(true); setUpdateFormValues(record); }} /> <Divider type="vertical" /> <Popconfirm title="您确定要删除$模型名字$吗?" placement="topRight" onConfirm={() => { handleRemove([record]) actionRef.current.reloadAndRest(); }} okText="确定" cancelText="取消" > <DeleteOutlined title="删除" className="icon" /> </Popconfirm> </> ), },""".replace("$模型名字$", model_ver_name) opera = opera.replace("$时间处理占位$", "".join(date_row_list)) columns.append(opera) dest_path = f'./antd_page_templates/{template_type}' with open(f'{dest_path}/index.jsx') as fr: content = fr.read() if fileds_num > 8: new_content = content.replace( "$两列布局占位$", """ search={{ span: { lg: 12, md: 12, xxl: 12, xl: 12, sm: 12, xs: 24, }, }}""") else: new_content = content.replace("$两列布局占位$", "") new_content = new_content.replace("$占位列数据$", "".join(columns)) new_content = new_content.replace("$占位模型名$", model_name) new_content = new_content.replace("$占位模型显示名$", str(model_ver_name)) new_content = new_content.replace("$外键占位$", "".join(model_add_item_list)) new_content = new_content.replace( "$多对多占位$", "".join(model_many_to_many_list)) new_content = new_content.replace( "$import占位$", "".join(model_import_item_list)) new_content = new_content.replace( "$时间占位$", ",".join(model_date_dict[model_name])) if len(model_pic_dict[model_name]) > 0: with open(f'{dest_path}/service_img.js') as fr: content = fr.read() new_services = content.replace( "$占位path$", get_lower_case_name(model_name)) new_services = new_services.replace("$占位模型名$", model_name) new_services = new_services.replace( "$图片字段列表$", ",".join(model_pic_dict[model_name])) else: with open(f'{dest_path}/service.js') as fr: content = fr.read() new_services = content.replace( "$占位path$", get_lower_case_name(model_name)) new_services = new_services.replace("$占位模型名$", model_name) with open(f'{dest_path}/components/CreateForm.jsx') as fr: create_form = fr.read() create_form = create_form.replace("$占位模型显示名$", str(model_ver_name)) if fileds_num > 8: create_form = create_form.replace("$宽度占位$", 'width={1200}') else: create_form = create_form.replace("$宽度占位$", "width={600}") with open(f'{dest_path}/components/UpdateForm.jsx') as fr: update_form = fr.read() update_form = update_form.replace("$占位模型显示名$", str(model_ver_name)) if fileds_num > 8: update_form = update_form.replace("$宽度占位$", 'width={1200}') else: update_form = update_form.replace("$宽度占位$", "width={600}") target_path = '../xadmin/src/pages/AutoGenPage' cur_path = f'{target_path}/{model_name}List' if not os.path.exists(cur_path): os.mkdir(cur_path) cur_path_co = f'{target_path}/{model_name}List/components' if not os.path.exists(cur_path_co): os.mkdir(cur_path_co) with open(f'{target_path}/{model_name}List/index.jsx', 'w') as fw: fw.write(new_content) with open(f'{target_path}/{model_name}List/service.js', 'w') as fw: fw.write(new_services) with open( f'{target_path}/{model_name}List/components/CreateForm.jsx', 'w') as fw: fw.write(create_form) with open( f'{target_path}/{model_name}List/components/UpdateForm.jsx', 'w') as fw: fw.write(update_form)