修复命令行工具

This commit is contained in:
不胜舟
2023-04-06 20:08:27 +08:00
parent 34f12842b0
commit e18a50ba30
4 changed files with 30 additions and 24 deletions
+6 -2
View File
@@ -1,3 +1,7 @@
from flask.cli import AppGroup
from flask import Flask
admin_cli = AppGroup("admin")
from .admin import admin_cli
def init_script(app: Flask):
app.cli.add_command(admin_cli)
@@ -1,9 +1,16 @@
import os
import click
from flask.cli import AppGroup
import datetime
from . import admin_cli
from applications.common.script.newmodular.new import NewViewModular
from applications.extensions import db
from applications.models import User, Role, Dept, Power
admin_cli = AppGroup("admin")
now_time = datetime.datetime.now()
userdata = [
User(
@@ -687,3 +694,17 @@ def init_db():
add_role_power()
print("角色权限数据存入")
print("数据初始化完成,请使用python app.py命令运行")
@admin_cli.command()
@click.option('--type', prompt="请输入类型", help='新增的类型')
@click.option('--name', prompt="请输入新增的名称", help='新增的名称')
def new(type, name):
if type == 'view':
if name.count('/') > 1:
print("目前只支持二级目录,多级目录需要蓝图嵌套,本命令暂不支持,请手动创建")
quit()
if type == "view" and os.path.exists(f"applications/view/{name}.py"):
print(f'已经存在视图模块{name}.py')
quit()
NewViewModular(name=name).new_view()
-18
View File
@@ -1,18 +0,0 @@
import click
from applications import admin_cli
from applications.common.script.newmodular.new import NewViewModular
@admin_cli.command()
@click.option('--type', prompt="请输入类型", help='新增的类型')
@click.option('--name', prompt="请输入新增的名称", help='新增的名称')
def new(type, name):
if type == 'view':
if name.count('/') > 1:
print("目前只支持二级目录,多级目录需要蓝图嵌套,本命令暂不支持,请手动创建")
quit()
if type == "view" and os.path.exists(f"applications/view/{name}.py"):
print(f'已经存在视图模块{name}.py')
quit()
NewViewModular(name=name).new_view()