import os

import click

from applications.common.script.initdb import init_db
from applications.common.script.newmodular.new import NewViewModular


def init_script(app):
    @app.cli.command()
    def init():
        init_db()

    @app.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()