加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
main.py 3.04 KB
一键复制 编辑 原始数据 按行查看 历史
# @Version : 1.0
# @Create Time : 2021/10/19 15:47
# @File : main.py
# @IDE : PyCharm
# @Desc : 主程序入口
"""
FastApi 更新文档:https://github.com/tiangolo/fastapi/releases
FastApi Github:https://github.com/tiangolo/fastapi
Typer 官方文档:https://typer.tiangolo.com/
"""
import typer
from kinit_fast_task.utils import log
shell_app = typer.Typer(rich_markup_mode="rich")
@shell_app.command()
def run(reload: bool = typer.Option(default=False, help="是否自动重载")):
"""
项目启动
命令行执行(自动重启):python main.py run --reload
命令行执行(不自动重启):python main.py run
原始命令行执行(uvicorn):uvicorn kinit_fast_task.main:app
# 在 pycharm 中使用自动重载是有 bug 的,很慢,截至 2024-04-30 还未修复,在使用 pycharm 开发时,不推荐使用 reload 功能
:param reload: 是否自动重启
:return:
""" # noqa E501
import uvicorn
from kinit_fast_task.config import settings
import sys
import os
sys.path.append(os.path.abspath(__file__))
uvicorn.run(
app="kinit_fast_task.main:app",
host=str(settings.system.SERVER_HOST),
port=settings.system.SERVER_PORT,
reload=reload,
lifespan="on",
)
@shell_app.command()
def migrate():
"""
将模型迁移到数据库,更新数据库表结构
命令示例:python main.py migrate
:return:
"""
from kinit_fast_task.utils.tools import exec_shell_command
log.info("开始更新数据库表")
exec_shell_command("alembic revision --autogenerate", "生成迁移文件失败")
exec_shell_command("alembic upgrade head", "迁移至数据库失败")
log.info("数据库表迁移完成")
@shell_app.command()
def generate(
model: str = typer.Option(..., "--model", "-m", help="Model 类名, 示例:AuthUserModel"),
app_name: str = typer.Option(..., "--app-name", "-n", help="功能英文名称, 主要用于 Schema, Params, CRUD, URL 命名"),
app_desc: str = typer.Option(..., "--app-desc", "-d", help="功能中文名称, 主要用于描述和注释"),
write_only: bool = typer.Option(False, "--write-only", "-w", help="是否只写入文件"),
overwrite: bool = typer.Option(False, "--overwrite", "-o", help="是否在写入时覆盖文件"),
):
"""
基于 ORM Model 生成 app 代码
命令示例(输出代码模式):python main.py generate -m AuthTestModel -n auth_test -d 测试
命令示例(写入代码不覆盖模式):python main.py generate -m AuthTestModel -n auth_test -d 测试 -w
命令示例(写入代码并覆盖模式):python main.py generate -m AuthTestModel -n auth_test -d 测试 -w -o
""" # noqa E501
from kinit_fast_task.scripts.app_generate.main import AppGenerate
ag = AppGenerate(verbose=False)
ag.model_to_code(
model_class_name=model, app_name=app_name, app_desc=app_desc, write_only=write_only, overwrite=overwrite
)
if __name__ == "__main__":
shell_app()
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化