Bootstrap

typer 模块与装饰器

当使用 Typer 模块时,可以创建一个简单的命令行应用程序示例。以下是一个使用 Typer 创建的示例:

import typer

app = typer.Typer()

@app.command()
def greet(name: str):
    typer.echo(f"Hello, {name}!")

@app.command()
def answer(name: str):
    typer.echo(f"I'am fine, {name}!")

if __name__ == "__main__":
    app()

在这个示例中,我们导入了 typer 模块并创建了一个 Typer 对象 app。然后,我们使用 @app.command() 装饰器将函数 greet 注册为一个命令。

greet 函数接受一个名为 name 的参数,并使用 typer.echo() 打印出一个问候语。

最后,我们使用 if __name__ == "__main__": 来确保只有在直接运行脚本时才执行 app(),而不是在导入为模块时执行。

你可以在命令行中运行这个示例,例如 python script.py greet jhon,它将输出 Hello, jhon!。这是一个简单的 Typer 模块示例,你可以根据需要扩展和定制它。

下面我们来看下修改装饰器会发生什么

import typer

app = typer.Typer()

@app.command("decorator")
def greet(name: str):
    typer.echo(f"Hello, {name}!")

@app.command()
def answer(name: str):
    typer.echo(f"I'am fine, {name}!")

if __name__ == "__main__":
    app()

当我们输入python script.py decorator jhon 时它将输出 Hello, jhon !

接下来我们自定义一个装饰器

from enum import Enum
import typer


app = typer.Typer()


class TriggerOn(Enum):
    PreSubmit = "change-content",
    PostSubmit = "change-commit",

def trigger(name: str, trigger_on: TriggerOn, depot_path: str, param: str = ""):
    def decorator(func):
        print(depot_path)
        return func

    return decorator

@app.command("decorator")
def greet(name: str):
    typer.echo(f"Hello, {name}!")

@app.command()
def answer(name: str):
    typer.echo(f"I'am fine, {name}!")


@trigger("res_check.main", TriggerOn.PreSubmit,
            "//JF_UEProj/main/Common/SpeedGame/Content/...")
@trigger("res_check.RelDM19", TriggerOn.PreSubmit,
            "//JF_UEProj/BR_PRacing_RelDM19/Common/SpeedGame/Content/...")
@trigger("res_check.Dev", TriggerOn.PreSubmit,
            "//JF_UEProj/BR_PRacing_Dev/Common/SpeedGame/Content/...")
@trigger("res_check.RelV1", TriggerOn.PreSubmit,
            "//JF_UEProj/BR_PRacing_RelV1/Common/SpeedGame/Content/...")
@app.command("res_check")
def res_check(change, user, client):
    typer.echo("change:{0},user:{1},client:{2}".format(change, user, client))

if __name__ == "__main__":
    app()

;