refactor: 统一使用px.task/px.cmd替代旧版TaskSpec创建任务

本次提交将项目内所有使用px.TaskSpec创建任务的代码，替换为新的px.task和px.cmd快捷API，简化了任务定义写法，同时更新了版本号到0.3.5。重构过程中保持了原有功能逻辑不变，仅调整了代码书写格式，提升了代码可读性和编写效率。
2026-07-04 15:22:27 +08:00
parent a0b1814024
commit c93f45dcb8
16 changed files with 313 additions and 426 deletions
@@ -261,20 +261,16 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "fmt":
-        graph = px.Graph.from_specs([px.TaskSpec("ruff_format", cmd=["ruff", "format", args.target], verbose=True)])
+        graph = px.Graph.from_specs([px.cmd(["ruff", "format", args.target], verbose=True)])
    elif args.command == "lint":
-        cmd = ["ruff", "check", args.target]
+        check_cmd = ["ruff", "check", args.target]
        if args.fix:
-            cmd.extend(["--fix", "--unsafe-fixes"])
-        graph = px.Graph.from_specs([px.TaskSpec("ruff_check", cmd=cmd, verbose=True)])
+            check_cmd.extend(["--fix", "--unsafe-fixes"])
+        graph = px.Graph.from_specs([px.cmd(check_cmd, verbose=True)])
    elif args.command == "doc":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("auto_docstring", fn=auto_add_docstrings, args=(Path(args.root_dir),), verbose=True)
-        ])
+        graph = px.Graph.from_specs([px.task(auto_add_docstrings, args=(Path(args.root_dir),), verbose=True)])
    elif args.command == "sync":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("sync_config", fn=sync_pyproject_config, args=(Path(args.root_dir),), verbose=True)
-        ])
+        graph = px.Graph.from_specs([px.task(sync_pyproject_config, args=(Path(args.root_dir),), verbose=True)])
    else:
        parser.print_help()
        return
@@ -212,14 +212,16 @@ def main() -> None:

    # 更新所有文件的版本号（使用顺序执行避免竞争条件）
    # 使用相对于 cwd 的路径作为任务名，确保唯一性
-    graph = px.Graph.from_specs([
-        px.TaskSpec(
-            f"bump_{file.relative_to(Path.cwd())}".replace("\\", "_").replace("/", "_").replace(".", "_"),
-            fn=bump_file_version,
-            args=(file, part),
-        )
-        for file in all_files
-    ])
+    graph = px.Graph.from_specs(
+        [
+            px.task(
+                bump_file_version,
+                args=(file, part),
+                name=f"bump_{file.relative_to(Path.cwd())}".replace("\\", "_").replace("/", "_").replace(".", "_"),
+            )
+            for file in all_files
+        ]
+    )
    report = px.run(graph, strategy="sequential")

    # 收集新版本号（取第一个成功的结果）
@@ -238,23 +240,13 @@ def main() -> None:

    # 提交修改并创建标签
    tasks = [
-        px.TaskSpec("git_add", cmd=["git", "add", "."]),
-        px.TaskSpec(
-            "git_commit",
-            cmd=["git", "commit", "-m", f"bump version to {new_version}"],
-            depends_on=("git_add",),
-        ),
+        px.cmd(["git", "add", "."]),
+        px.cmd(["git", "commit", "-m", f"bump version to {new_version}"], depends_on=("git_add",)),
    ]

    if not args.no_tag:
        tag_name = f"v{new_version}"
-        tasks.append(
-            px.TaskSpec(
-                "git_tag",
-                cmd=["git", "tag", "-a", tag_name, "-m", f"Release {tag_name}"],
-                depends_on=("git_commit",),
-            )
-        )
+        tasks.append(px.cmd(["git", "tag", "-a", tag_name, "-m", f"Release {tag_name}"], depends_on=("git_commit",)))

    graph = px.Graph.from_specs(tasks)
    px.run(graph, strategy="sequential")
@@ -209,107 +209,112 @@ def main() -> None:
    RUST_SCCACHE_DIR.mkdir(parents=True, exist_ok=True)

    # 使用 conditions 自动控制任务执行
-    graph = px.Graph.from_specs([
-        # 系统镜像配置（仅 Linux 且未配置国内镜像）
-        px.TaskSpec(
-            "download_mirror",
-            cmd=DOWNLOAD_MIRROR_SCRIPT,
-            conditions=(
-                BuiltinConditions.IS_LINUX(),
-                BuiltinConditions.NOT(
-                    BuiltinConditions.OR(
-                        *[
-                            BuiltinConditions.FILE_CONTENT_EXISTS(f, m)
-                            for f in [
-                                "/etc/apt/sources.list",
-                                "/etc/apt/sources.list.d/ubuntu.sources",
-                            ]
-                            for m in get_args(PyMirrorType)
-                        ],
-                    )
+    graph = px.Graph.from_specs(
+        [
+            # 系统镜像配置（仅 Linux 且未配置国内镜像）
+            px.TaskSpec(
+                "download_mirror",
+                cmd=DOWNLOAD_MIRROR_SCRIPT,
+                conditions=(
+                    BuiltinConditions.IS_LINUX(),
+                    BuiltinConditions.NOT(
+                        BuiltinConditions.OR(
+                            *[
+                                BuiltinConditions.FILE_CONTENT_EXISTS(f, m)
+                                for f in [
+                                    "/etc/apt/sources.list",
+                                    "/etc/apt/sources.list.d/ubuntu.sources",
+                                ]
+                                for m in get_args(PyMirrorType)
+                            ],
+                        )
+                    ),
                ),
+                verbose=True,
            ),
-            verbose=True,
-        ),
-        px.TaskSpec(
-            "install_mirror",
-            cmd=INSTALL_MIRROR_SCRIPT,
-            depends_on=("download_mirror",),
-            verbose=True,
-        ),
-        # 安装 Qt 依赖（仅 Linux）
-        px.TaskSpec(
-            "install_qt_libs",
-            cmd=["sudo", "apt", "install", "-y", *QT_LIBS],
-            conditions=(BuiltinConditions.IS_LINUX(),),
-            depends_on=("install_mirror",),
-            allow_upstream_skip=True,
-            verbose=True,
-        ),
-        # 安装中文字体（仅 Linux）
-        px.TaskSpec(
-            "install_fonts",
-            cmd=["sudo", "apt", "install", "-y", *CHINESE_FONTS],
-            conditions=(BuiltinConditions.IS_LINUX(),),
-            depends_on=("install_mirror",),
-            allow_upstream_skip=True,
-            verbose=True,
-        ),
-        # 安装 Docker
-        px.TaskSpec(
-            "install_docker",
-            cmd=["sudo", "apt", "install", "-y", "docker-compose-v2"],
-            conditions=(BuiltinConditions.IS_LINUX(),),
-            depends_on=("install_mirror",),
-            allow_upstream_skip=True,
-            verbose=True,
-        ),
-        px.TaskSpec(
-            "add_docker_group",
-            cmd=["sudo", "usermod", "-aG", "docker", getpass.getuser()],
-            conditions=(BuiltinConditions.IS_LINUX(),),
-            depends_on=("install_docker",),
-            allow_upstream_skip=True,
-            verbose=True,
-        ),
-        px.TaskSpec(
-            "refresh_docker_group",
-            cmd=["newgrp", "docker"],
-            conditions=(BuiltinConditions.IS_LINUX(),),
-            depends_on=("add_docker_group",),
-            allow_upstream_skip=True,
-            verbose=True,
-        ),
-        # 设置 Python 环境变量
-        *setenv_group({
-            "PIP_INDEX_URL": PIP_INDEX_URLS[python_mirror],
-            "PIP_TRUSTED_HOSTS": PIP_TRUSTED_HOSTS[python_mirror],
-            "UV_INDEX_URL": UV_INDEX_URLS[python_mirror],
-            "UV_PYTHON_INSTALL_MIRROR": UV_PYTHON_INSTALL_MIRROR,
-            "UV_HTTP_TIMEOUT": "600",
-            "UV_LINK_MODE": "copy",
-        }),
-        # 写入 Python 配置（仅当未配置）
-        write_file(
-            str(PIP_CONFIG_PATH),
-            f"[global]\nindex-url = {PIP_INDEX_URLS[python_mirror]}\ntrusted-host = {PIP_TRUSTED_HOSTS[python_mirror]}",
-        ),
-        # 写入 Conda 配置（仅当未配置）
-        write_file(
-            str(CONDA_CONFIG_PATH),
-            "show_channel_urls: true\nchannels:\n  - " + "\n  - ".join(conda_mirror_urls) + "\n  - defaults",
-        ),
-        # 设置 Rust 镜像源
-        *setenv_group({
-            "RUSTUP_DIST_SERVER": RUSTUP_MIRRORS[rust_mirror]["RUSTUP_DIST_SERVER"],
-            "RUSTUP_UPDATE_ROOT": RUSTUP_MIRRORS[rust_mirror]["RUSTUP_UPDATE_ROOT"],
-            "RUST_SCCACHE_DIR": str(RUST_SCCACHE_DIR),
-            "RUST_SCCACHE_CACHE_SIZE": RUST_SCCACHE_CACHE_SIZE,
-        }),
-        # 写入 Rust 配置（仅当未配置）
-        write_file(
-            str(RUST_CONFIG_PATH),
-            f"""
+            px.TaskSpec(
+                "install_mirror",
+                cmd=INSTALL_MIRROR_SCRIPT,
+                depends_on=("download_mirror",),
+                verbose=True,
+            ),
+            # 安装 Qt 依赖（仅 Linux）
+            px.TaskSpec(
+                "install_qt_libs",
+                cmd=["sudo", "apt", "install", "-y", *QT_LIBS],
+                conditions=(BuiltinConditions.IS_LINUX(),),
+                depends_on=("install_mirror",),
+                allow_upstream_skip=True,
+                verbose=True,
+            ),
+            # 安装中文字体（仅 Linux）
+            px.TaskSpec(
+                "install_fonts",
+                cmd=["sudo", "apt", "install", "-y", *CHINESE_FONTS],
+                conditions=(BuiltinConditions.IS_LINUX(),),
+                depends_on=("install_mirror",),
+                allow_upstream_skip=True,
+                verbose=True,
+            ),
+            # 安装 Docker
+            px.TaskSpec(
+                "install_docker",
+                cmd=["sudo", "apt", "install", "-y", "docker-compose-v2"],
+                conditions=(BuiltinConditions.IS_LINUX(),),
+                depends_on=("install_mirror",),
+                allow_upstream_skip=True,
+                verbose=True,
+            ),
+            px.TaskSpec(
+                "add_docker_group",
+                cmd=["sudo", "usermod", "-aG", "docker", getpass.getuser()],
+                conditions=(BuiltinConditions.IS_LINUX(),),
+                depends_on=("install_docker",),
+                allow_upstream_skip=True,
+                verbose=True,
+            ),
+            px.TaskSpec(
+                "refresh_docker_group",
+                cmd=["newgrp", "docker"],
+                conditions=(BuiltinConditions.IS_LINUX(),),
+                depends_on=("add_docker_group",),
+                allow_upstream_skip=True,
+                verbose=True,
+            ),
+            # 设置 Python 环境变量
+            *setenv_group(
+                {
+                    "PIP_INDEX_URL": PIP_INDEX_URLS[python_mirror],
+                    "PIP_TRUSTED_HOSTS": PIP_TRUSTED_HOSTS[python_mirror],
+                    "UV_INDEX_URL": UV_INDEX_URLS[python_mirror],
+                    "UV_PYTHON_INSTALL_MIRROR": UV_PYTHON_INSTALL_MIRROR,
+                    "UV_HTTP_TIMEOUT": "600",
+                    "UV_LINK_MODE": "copy",
+                }
+            ),
+            # 写入 Python 配置（仅当未配置）
+            write_file(
+                str(PIP_CONFIG_PATH),
+                f"[global]\nindex-url = {PIP_INDEX_URLS[python_mirror]}\ntrusted-host = {PIP_TRUSTED_HOSTS[python_mirror]}",
+            ),
+            # 写入 Conda 配置（仅当未配置）
+            write_file(
+                str(CONDA_CONFIG_PATH),
+                "show_channel_urls: true\nchannels:\n  - " + "\n  - ".join(conda_mirror_urls) + "\n  - defaults",
+            ),
+            # 设置 Rust 镜像源
+            *setenv_group(
+                {
+                    "RUSTUP_DIST_SERVER": RUSTUP_MIRRORS[rust_mirror]["RUSTUP_DIST_SERVER"],
+                    "RUSTUP_UPDATE_ROOT": RUSTUP_MIRRORS[rust_mirror]["RUSTUP_UPDATE_ROOT"],
+                    "RUST_SCCACHE_DIR": str(RUST_SCCACHE_DIR),
+                    "RUST_SCCACHE_CACHE_SIZE": RUST_SCCACHE_CACHE_SIZE,
+                }
+            ),
+            # 写入 Rust 配置（仅当未配置）
+            write_file(
+                str(RUST_CONFIG_PATH),
+                f"""
 [source.crates-io]
 replace-with = '{rust_mirror}'

@@ -319,39 +324,43 @@ registry = "sparse+{RUSTUP_MIRRORS[rust_mirror]["TOML_REGISTRY"]}"
 [registries.{rust_mirror}]
 index = "sparse+{RUSTUP_MIRRORS[rust_mirror]["TOML_REGISTRY"]}"
 """,
-        ),
-        # 下载 Rustup 安装脚本
-        px.TaskSpec(
-            "download_rustup",
-            cmd=["curl", "-fsSL", RUSTUP_DOWNLOAD_URL_LINUX, "-o", "rustup-init.sh"],
-            conditions=(BuiltinConditions.IS_LINUX(), BuiltinConditions.NOT(BuiltinConditions.HAS_INSTALLED("rustup"))),
-            verbose=True,
-        ),
-        px.TaskSpec(
-            "download_rustup_win",
-            cmd=[
-                "powershell",
-                "-Command",
-                "Invoke-WebRequest",
-                "-Uri",
-                RUSTUP_DOWNLOAD_URL_WINDOWS,
-                "-OutFile",
-                "rustup-init.exe",
-            ],
-            conditions=(
-                BuiltinConditions.IS_WINDOWS(),
-                BuiltinConditions.NOT(BuiltinConditions.HAS_INSTALLED("rustup")),
            ),
-            verbose=True,
-        ),
-        # 安装 Rust 工具链
-        px.TaskSpec(
-            "install_rust",
-            cmd=["rustup", "toolchain", "install", rust_version],
-            conditions=(BuiltinConditions.HAS_INSTALLED("rustup"),),
-            depends_on=("setenv_rustup_dist_server",),
-            allow_upstream_skip=True,
-            verbose=True,
-        ),
-    ])
+            # 下载 Rustup 安装脚本
+            px.TaskSpec(
+                "download_rustup",
+                cmd=["curl", "-fsSL", RUSTUP_DOWNLOAD_URL_LINUX, "-o", "rustup-init.sh"],
+                conditions=(
+                    BuiltinConditions.IS_LINUX(),
+                    BuiltinConditions.NOT(BuiltinConditions.HAS_INSTALLED("rustup")),
+                ),
+                verbose=True,
+            ),
+            px.TaskSpec(
+                "download_rustup_win",
+                cmd=[
+                    "powershell",
+                    "-Command",
+                    "Invoke-WebRequest",
+                    "-Uri",
+                    RUSTUP_DOWNLOAD_URL_WINDOWS,
+                    "-OutFile",
+                    "rustup-init.exe",
+                ],
+                conditions=(
+                    BuiltinConditions.IS_WINDOWS(),
+                    BuiltinConditions.NOT(BuiltinConditions.HAS_INSTALLED("rustup")),
+                ),
+                verbose=True,
+            ),
+            # 安装 Rust 工具链
+            px.TaskSpec(
+                "install_rust",
+                cmd=["rustup", "toolchain", "install", rust_version],
+                conditions=(BuiltinConditions.HAS_INSTALLED("rustup"),),
+                depends_on=("setenv_rustup_dist_server",),
+                allow_upstream_skip=True,
+                verbose=True,
+            ),
+        ]
+    )
    px.run(graph, strategy="thread", verbose=True)
@@ -567,13 +567,15 @@ class EmlManagerHandler(BaseHTTPRequestHandler):

        emails = self.db.search_emails(keyword, field, limit, offset)
        total_count = self.db.get_email_count()
-        self._send_json_response({
-            "emails": emails,
-            "count": len(emails),
-            "total": total_count,
-            "limit": limit,
-            "offset": offset,
-        })
+        self._send_json_response(
+            {
+                "emails": emails,
+                "count": len(emails),
+                "total": total_count,
+                "limit": limit,
+                "offset": offset,
+            }
+        )

    def _api_get_email(self, query_params: dict[str, list[str]]) -> None:
        """API: 获取单个邮件详情."""
@@ -113,23 +113,13 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "add":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "process_files_date",
-                fn=process_files_date,
-                args=([Path(f) for f in args.files],),
-                kwargs={"clear": False},
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(process_files_date, args=([Path(f) for f in args.files],), kwargs={"clear": False})]
+        )
    elif args.command == "clear":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "process_files_date",
-                fn=process_files_date,
-                args=([Path(f) for f in args.files],),
-                kwargs={"clear": True},
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(process_files_date, args=([Path(f) for f in args.files],), kwargs={"clear": True})]
+        )
    else:
        parser.print_help()
        return
@@ -126,13 +126,7 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "set":
-        graph = px.Graph.from_specs(
-            [
-                px.TaskSpec(
-                    "process_files_level", fn=process_files_level, args=([Path(f) for f in args.files], args.level)
-                )
-            ]
-        )
+        graph = px.Graph.from_specs([px.task(process_files_level, args=([Path(f) for f in args.files], args.level))])
    else:
        parser.print_help()
        return
@@ -33,16 +33,11 @@ def init_sub_dirs() -> None:
    sub_dirs = [subdir for subdir in Path.cwd().iterdir() if subdir.is_dir()]
    for subdir in sub_dirs:
        px.run(
-            px.Graph.from_specs([
-                px.TaskSpec(
-                    "init",
-                    cmd=["git", "init"],
-                    conditions=(lambda _: not_has_git_repo(),),
-                    cwd=subdir,
-                ),
-                px.TaskSpec("add", cmd=["git", "add", "."], depends_on=("init",)),
-                px.TaskSpec("commit", cmd=["git", "commit", "-m", "init commit"], depends_on=("add",)),
-            ]),
+            px.Graph().chain(
+                px.cmd(["git", "init"], conditions=(lambda _: not_has_git_repo(),), cwd=subdir),
+                px.cmd(["git", "add", "."]),
+                px.cmd(["git", "commit", "-m", "init commit"]),
+            ),
        )


@@ -52,9 +47,9 @@ def isub() -> None:
    init_sub_dirs()


-push: px.TaskSpec = px.TaskSpec("push", cmd=["git", "push"])
-pull: px.TaskSpec = px.TaskSpec("pull", cmd=["git", "pull"])
-kill_tgit: px.TaskSpec = px.TaskSpec("task_kill", cmd=["taskkill", "/f", "/t", "/im", "tgitcache.exe"])
+push: px.TaskSpec = px.cmd(["git", "push"], name="push")
+pull: px.TaskSpec = px.cmd(["git", "pull"], name="pull")
+kill_tgit: px.TaskSpec = px.cmd(["taskkill", "/f", "/t", "/im", "tgitcache.exe"], name="task_kill")


 def not_has_git_repo() -> bool:
@@ -74,26 +69,21 @@ def main() -> None:
        description="Gittool - Git 执行工具.",
        aliases={
            # 添加并提交
-            "a": px.Graph.from_specs([
-                px.TaskSpec("add", cmd=["git", "add", "."], conditions=(lambda _: has_files(),)),
-                px.TaskSpec("commit", cmd=["git", "commit", "-m", "chore: update"], depends_on=("add",)),
-            ]),
+            "a": px.Graph().chain(
+                px.cmd(["git", "add", "."], conditions=(lambda _: has_files(),)),
+                px.cmd(["git", "commit", "-m", "chore: update"]),
+            ),
            # 清理（chain: clean → status）
            "c": px.Graph().chain(
-                px.TaskSpec("clean", cmd=["git", "clean", "-xfd", *EXCLUDE_CMDS]),
-                px.TaskSpec("status", cmd=["git", "status", "--porcelain"]),
+                px.cmd(["git", "clean", "-xfd", *EXCLUDE_CMDS]),
+                px.cmd(["git", "status", "--porcelain"]),
            ),
            # 初始化、添加并提交
-            "i": px.Graph.from_specs([
-                px.TaskSpec("init", cmd=["git", "init"], conditions=(lambda _: not_has_git_repo(),)),
-                px.TaskSpec("add", cmd=["git", "add", "."], depends_on=("init",), conditions=(lambda _: has_files(),)),
-                px.TaskSpec(
-                    "commit",
-                    cmd=["git", "commit", "-m", "init commit"],
-                    depends_on=("add",),
-                    conditions=(lambda _: has_files(),),
-                ),
-            ]),
+            "i": px.Graph().chain(
+                px.cmd(["git", "init"], conditions=(lambda _: not_has_git_repo(),)),
+                px.cmd(["git", "add", "."], conditions=(lambda _: has_files(),)),
+                px.cmd(["git", "commit", "-m", "init commit"], conditions=(lambda _: has_files(),)),
+            ),
            # 初始化子目录
            "isub": isub,
            # 推送
@@ -22,20 +22,22 @@ def main():
    download_dir: Path = Path(args.dir) if args.dir else Path.home() / ".models" / args.name.split("/")[-1]
    download_dir.mkdir(parents=True, exist_ok=True)

-    graph = px.Graph.from_specs([
-        px.TaskSpec(
-            name="download",
-            cmd=[
-                "uvx",
-                "modelscope",
-                "download",
-                f"--{args.type}",
-                args.name,
-                "--local_dir",
-                str(download_dir),
-            ],
-            verbose=True,
-        ),
-    ])
+    graph = px.Graph.from_specs(
+        [
+            px.TaskSpec(
+                name="download",
+                cmd=[
+                    "uvx",
+                    "modelscope",
+                    "download",
+                    f"--{args.type}",
+                    args.name,
+                    "--local_dir",
+                    str(download_dir),
+                ],
+                verbose=True,
+            ),
+        ]
+    )

    px.run(graph, strategy="thread", verbose=True)
@@ -24,40 +24,42 @@ def main():
    if not model_dir.exists():
        parser.error(f"Model directory {model_dir} does not exist.")

-    graph = px.Graph.from_specs([
-        px.TaskSpec(
-            name="download",
-            cmd=[
-                "uv",
-                "install",
-                "sglang[all]",
-            ],
-            conditions=(BuiltinConditions.NOT(BuiltinConditions.HAS_INSTALLED("sglang")),),
-            verbose=True,
-        ),
-        px.TaskSpec(
-            name="run",
-            cmd=[
-                "python" if Constants.IS_WINDOWS else "python3",
-                "-m",
-                "sglang.launch_server",
-                "--model-path",
-                str(model_dir),
-                "--host",
-                str(args.host),
-                "--port",
-                "8000",
-                "--mem-fraction-static",
-                str(args.mem),
-                "--context-length",
-                "32768",
-                "--tool-call-parser",
-                "qwen",
-                "--log-level",
-                str(args.log_level),
-            ],
-            verbose=True,
-        ),
-    ])
+    graph = px.Graph.from_specs(
+        [
+            px.TaskSpec(
+                name="download",
+                cmd=[
+                    "uv",
+                    "install",
+                    "sglang[all]",
+                ],
+                conditions=(BuiltinConditions.NOT(BuiltinConditions.HAS_INSTALLED("sglang")),),
+                verbose=True,
+            ),
+            px.TaskSpec(
+                name="run",
+                cmd=[
+                    "python" if Constants.IS_WINDOWS else "python3",
+                    "-m",
+                    "sglang.launch_server",
+                    "--model-path",
+                    str(model_dir),
+                    "--host",
+                    str(args.host),
+                    "--port",
+                    "8000",
+                    "--mem-fraction-static",
+                    str(args.mem),
+                    "--context-length",
+                    "32768",
+                    "--tool-call-parser",
+                    "qwen",
+                    "--log-level",
+                    str(args.log_level),
+                ],
+                verbose=True,
+            ),
+        ]
+    )

    px.run(graph, strategy="sequential", verbose=True)
@@ -158,15 +158,11 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "run":
-        graph = px.Graph.from_specs(
-            [px.TaskSpec("run_ls_dyna", fn=run_ls_dyna, args=(args.input_file,), kwargs={"ncpu": args.ncpu})]
-        )
+        graph = px.Graph.from_specs([px.task(run_ls_dyna, args=(args.input_file,), kwargs={"ncpu": args.ncpu})])
    elif args.command == "mpi":
-        graph = px.Graph.from_specs(
-            [px.TaskSpec("run_ls_dyna_mpi", fn=run_ls_dyna_mpi, args=(args.input_file,), kwargs={"ncpu": args.ncpu})]
-        )
+        graph = px.Graph.from_specs([px.task(run_ls_dyna_mpi, args=(args.input_file,), kwargs={"ncpu": args.ncpu})])
    elif args.command == "status":
-        graph = px.Graph.from_specs([px.TaskSpec("check_ls_dyna_status", fn=check_ls_dyna_status)])
+        graph = px.Graph.from_specs([px.task(check_ls_dyna_status)])
    else:
        parser.print_help()
        return
@@ -291,57 +291,17 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "src":
-        graph = px.Graph.from_specs(
-            [
-                px.TaskSpec(
-                    "pack_source",
-                    fn=pack_source,
-                    args=(Path(args.project_dir), Path(args.output_dir)),
-                )
-            ]
-        )
+        graph = px.Graph.from_specs([px.task(pack_source, args=(Path(args.project_dir), Path(args.output_dir)))])
    elif args.command == "deps":
-        graph = px.Graph.from_specs(
-            [
-                px.TaskSpec(
-                    "pack_deps",
-                    fn=pack_dependencies,
-                    args=(Path(args.lib_dir), args.dependencies),
-                )
-            ]
-        )
+        graph = px.Graph.from_specs([px.task(pack_dependencies, args=(Path(args.lib_dir), args.dependencies))])
    elif args.command == "wheel":
-        graph = px.Graph.from_specs(
-            [
-                px.TaskSpec(
-                    "pack_wheel",
-                    fn=pack_wheel,
-                    args=(Path(args.project_dir), Path(args.output_dir)),
-                )
-            ]
-        )
+        graph = px.Graph.from_specs([px.task(pack_wheel, args=(Path(args.project_dir), Path(args.output_dir)))])
    elif args.command == "embed":
-        graph = px.Graph.from_specs(
-            [
-                px.TaskSpec(
-                    "install_embed",
-                    fn=install_embed_python,
-                    args=(args.version, Path(args.output_dir)),
-                )
-            ]
-        )
+        graph = px.Graph.from_specs([px.task(install_embed_python, args=(args.version, Path(args.output_dir)))])
    elif args.command == "zip":
-        graph = px.Graph.from_specs(
-            [
-                px.TaskSpec(
-                    "create_zip",
-                    fn=create_zip_package,
-                    args=(Path(args.source_dir), Path(args.output_file)),
-                )
-            ]
-        )
+        graph = px.Graph.from_specs([px.task(create_zip_package, args=(Path(args.source_dir), Path(args.output_file)))])
    elif args.command == "clean":
-        graph = px.Graph.from_specs([px.TaskSpec("clean_build", fn=clean_build_dir, args=(Path(DEFAULT_BUILD_DIR),))])
+        graph = px.Graph.from_specs([px.task(clean_build_dir, args=(Path(DEFAULT_BUILD_DIR),))])
    else:
        parser.print_help()
        return
@@ -443,79 +443,49 @@ def main() -> None:  # noqa: PLR0912
    args = parser.parse_args()

    if args.command == "m":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_merge", fn=pdf_merge, args=([Path(p) for p in args.inputs], Path(args.output)))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_merge, args=([Path(p) for p in args.inputs], Path(args.output)))])
    elif args.command == "s":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_split", fn=pdf_split, args=(Path(args.input), Path(args.output_dir)))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_split, args=(Path(args.input), Path(args.output_dir)))])
    elif args.command == "c":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_compress", fn=pdf_compress, args=(Path(args.input), Path(args.output)))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_compress, args=(Path(args.input), Path(args.output)))])
    elif args.command == "e":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_encrypt", fn=pdf_encrypt, args=(Path(args.input), Path(args.output), args.password))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_encrypt, args=(Path(args.input), Path(args.output), args.password))])
    elif args.command == "d":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_decrypt", fn=pdf_decrypt, args=(Path(args.input), Path(args.output), args.password))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_decrypt, args=(Path(args.input), Path(args.output), args.password))])
    elif args.command == "xt":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_extract_text", fn=pdf_extract_text, args=(Path(args.input), Path(args.output)))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_extract_text, args=(Path(args.input), Path(args.output)))])
    elif args.command == "xi":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_extract_images", fn=pdf_extract_images, args=(Path(args.input), Path(args.output_dir)))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_extract_images, args=(Path(args.input), Path(args.output_dir)))])
    elif args.command == "w":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "pdf_watermark",
-                fn=pdf_add_watermark,
-                args=(Path(args.input), Path(args.output)),
-                kwargs={"text": args.text},
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(pdf_add_watermark, args=(Path(args.input), Path(args.output)), kwargs={"text": args.text})]
+        )
    elif args.command == "r":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "pdf_rotate",
-                fn=pdf_rotate,
-                args=(Path(args.input), Path(args.output)),
-                kwargs={"rotation": args.rotation},
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(pdf_rotate, args=(Path(args.input), Path(args.output)), kwargs={"rotation": args.rotation})]
+        )
    elif args.command == "crop":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "pdf_crop",
-                fn=pdf_crop,
-                args=(Path(args.input), Path(args.output)),
-                kwargs={"margins": (args.left, args.top, args.right, args.bottom)},
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [
+                px.task(
+                    pdf_crop,
+                    args=(Path(args.input), Path(args.output)),
+                    kwargs={"margins": (args.left, args.top, args.right, args.bottom)},
+                )
+            ]
+        )
    elif args.command == "i":
-        graph = px.Graph.from_specs([px.TaskSpec("pdf_info", fn=pdf_info, args=(Path(args.input),))])
+        graph = px.Graph.from_specs([px.task(pdf_info, args=(Path(args.input),))])
    elif args.command == "ocr":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_ocr", fn=pdf_ocr, args=(Path(args.input), Path(args.output)), kwargs={"lang": args.lang})
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(pdf_ocr, args=(Path(args.input), Path(args.output)), kwargs={"lang": args.lang})]
+        )
    elif args.command == "img":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "pdf_to_images",
-                fn=pdf_to_images,
-                args=(Path(args.input), Path(args.output_dir)),
-                kwargs={"dpi": args.dpi},
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(pdf_to_images, args=(Path(args.input), Path(args.output_dir)), kwargs={"dpi": args.dpi})]
+        )
    elif args.command == "repair":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pdf_repair", fn=pdf_repair, args=(Path(args.input), Path(args.output)))
-        ])
+        graph = px.Graph.from_specs([px.task(pdf_repair, args=(Path(args.input), Path(args.output)))])
    else:
        parser.print_help()
        return
@@ -21,10 +21,12 @@ PACKAGE_DIR = "packages"
 REQUIREMENTS_FILE = "requirements.txt"

 # 受保护的包名集合
-_PROTECTED_PACKAGES: frozenset[str] = frozenset({
-    "pyflowx",
-    "bitool",
-})
+_PROTECTED_PACKAGES: frozenset[str] = frozenset(
+    {
+        "pyflowx",
+        "bitool",
+    }
+)


 # ============================================================================
@@ -157,37 +159,23 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "i":
-        graph = px.Graph.from_specs([px.TaskSpec("pip_install", cmd=["pip", "install", *args.packages], verbose=True)])
+        graph = px.Graph.from_specs([px.cmd(["pip", "install", *args.packages], verbose=True)])
    elif args.command == "u":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pip_uninstall", fn=pip_uninstall, args=(args.packages,), verbose=True)
-        ])
+        graph = px.Graph.from_specs([px.task(pip_uninstall, args=(args.packages,), verbose=True)])
    elif args.command == "r":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "pip_reinstall",
-                fn=pip_reinstall,
-                args=(args.packages,),
-                kwargs={"offline": args.offline},
-                verbose=True,
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(pip_reinstall, args=(args.packages,), kwargs={"offline": args.offline}, verbose=True)]
+        )
    elif args.command == "d":
-        graph = px.Graph.from_specs([
-            px.TaskSpec(
-                "pip_download",
-                fn=pip_download,
-                args=(args.packages,),
-                kwargs={"offline": args.offline},
-                verbose=True,
-            )
-        ])
+        graph = px.Graph.from_specs(
+            [px.task(pip_download, args=(args.packages,), kwargs={"offline": args.offline}, verbose=True)]
+        )
    elif args.command == "up":
-        graph = px.Graph.from_specs([
-            px.TaskSpec("pip_upgrade", cmd=["python", "-m", "pip", "install", "--upgrade", "pip"], verbose=True)
-        ])
+        graph = px.Graph.from_specs(
+            [px.cmd(["python", "-m", "pip", "install", "--upgrade", "pip"], name="pip_upgrade", verbose=True)]
+        )
    elif args.command == "f":
-        graph = px.Graph.from_specs([px.TaskSpec("pip_freeze", fn=pip_freeze, verbose=True)])
+        graph = px.Graph.from_specs([px.task(pip_freeze, verbose=True)])
    else:
        parser.print_help()
        return
@@ -149,13 +149,9 @@ def main() -> None:
    args = parser.parse_args()

    if args.command == "full":
-        graph = px.Graph.from_specs(
-            [px.TaskSpec("screenshot_full", fn=take_screenshot_full, kwargs={"filename": args.filename})]
-        )
+        graph = px.Graph.from_specs([px.task(take_screenshot_full, kwargs={"filename": args.filename})])
    elif args.command == "area":
-        graph = px.Graph.from_specs(
-            [px.TaskSpec("screenshot_area", fn=take_screenshot_area, kwargs={"filename": args.filename})]
-        )
+        graph = px.Graph.from_specs([px.task(take_screenshot_area, kwargs={"filename": args.filename})])
    else:
        parser.print_help()
        return
@@ -110,11 +110,11 @@ def main() -> None:
    args = parser.parse_args()

    graph = px.Graph.from_specs([
-        px.TaskSpec(
-            "ssh_deploy",
-            fn=ssh_copy_id,
+        px.task(
+            ssh_copy_id,
            args=(args.hostname, args.username, args.password),
            kwargs={"port": args.port, "keypath": args.keypath, "timeout": args.timeout},
+            name="ssh_deploy",
        )
    ])
    px.run(graph, strategy="thread")
@@ -2736,7 +2736,7 @@ wheels = [

 [[package]]
 name = "pyflowx"
-version = "0.3.2"
+version = "0.3.5"
 source = { editable = "." }
 dependencies = [
    { name = "graphlib-backport", marker = "python_full_version < '3.9'" },