bitool 0.1.2__py3-none-any.whl

bitool/scripts/bumpversion.py

@@ -0,0 +1,189 @@
+"""版本号自动管理工具.
+
+参考旧版 bumpversion 和 pymake 的设计, 使用 CommandScheduler 模式实现.
+支持语义化版本管理和多文件格式的版本号更新.
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+
+from bitool.cmd import (
+    BumpProjectVersionCommand,
+    CommandScheduler,
+    GitCommitCommand,
+    GitTagCommand,
+)
+from bitool.core import ConfigMixin, logger
+
+
+class BumpversionConfig(ConfigMixin):
+    """Bumpversion 配置类."""
+
+    # 排除的目录
+    EXCLUDED_DIRS: set[str] = {  # noqa: RUF012
+        "__pycache__",
+        ".venv",
+        "venv",
+        "env",
+        ".git",
+        "dist",
+        "build",
+        ".tox",
+        ".pytest_cache",
+    }
+
+
+conf = BumpversionConfig()
+
+
+def create_scheduler_map() -> dict[str, CommandScheduler]:
+    """创建命令调度器映射.
+
+    Returns
+    -------
+    dict[str, CommandScheduler]
+        操作名称到命令调度器的映射字典.
+    """
+
+    return {
+        # === 版本号递增命令 ===
+        # 递增补丁号 (1.0.0 -> 1.0.1)
+        "patch": CommandScheduler(commands=[BumpProjectVersionCommand(part="patch")]),
+        # 递增次版本号 (1.0.0 -> 1.1.0)
+        "minor": CommandScheduler(commands=[BumpProjectVersionCommand(part="minor")]),
+        # 递增主版本号 (1.0.0 -> 2.0.0)
+        "major": CommandScheduler(commands=[BumpProjectVersionCommand(part="major")]),
+        # 递增补丁号并设置预发布标识 (1.0.0 -> 1.0.1-alpha)
+        "patch-alpha": CommandScheduler(
+            commands=[BumpProjectVersionCommand(part="patch", prerelease="alpha")]
+        ),
+    }
+
+
+OPTIONS: list[str] = list(create_scheduler_map().keys())
+
+
+def show_help() -> str:
+    """显示帮助信息.
+
+    Returns
+    -------
+    str
+        帮助信息文本
+    """
+    help_text = """
+╔══════════════════════════════════════════════════════════╗
+║              Bitool 版本号自动管理工具                   ║
+╚══════════════════════════════════════════════════════════╝
+
+📈 版本号递增命令:
+  bumpversion patch        - 递增补丁号 (1.0.0 -> 1.0.1)
+  bumpversion minor        - 递增次版本号 (1.0.0 -> 1.1.0)
+  bumpversion major        - 递增主版本号 (1.0.0 -> 2.0.0)
+  bumpversion patch-alpha  - 递增补丁号并设置预发布标识 (1.0.0 -> 1.0.1-alpha)
+
+💡 支持的文件格式:
+  - pyproject.toml: version = "1.0.0"
+  - __init__.py: __version__ = "1.0.0"
+  - setup.py: version = "1.0.0"
+  - package.json: "version": "1.0.0"
+  - Cargo.toml: version = "1.0.0"
+
+📝 示例:
+  bumpversion patch          # 递增补丁号
+  bumpversion minor          # 递增次版本号  
+  bumpversion patch-alpha    # 递增补丁号并添加 alpha 预发布标识
+"""
+    return help_text
+
+
+def parse_args() -> tuple[argparse.ArgumentParser, argparse.Namespace]:
+    """解析命令行参数.
+
+    Returns
+    -------
+    tuple[argparse.ArgumentParser, argparse.Namespace]
+        (参数解析器, 解析后的参数对象)
+    """
+    parser = argparse.ArgumentParser(
+        description="Bitool 版本号自动管理工具 (支持语义化版本和多文件格式)",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog=show_help(),
+    )
+    parser.add_argument(
+        "action",
+        nargs="?",
+        default="patch",
+        choices=OPTIONS,
+        type=str,
+        help="要执行的操作, 可选值: " + ", ".join(OPTIONS),
+    )
+    parser.add_argument(
+        "-t",
+        "--tag",
+        action="store_true",
+        default=False,
+        help="递增版本号后创建 Git 标签",
+    )
+    parser.add_argument(
+        "-c",
+        "--commit",
+        action="store_true",
+        default=False,
+        help="递增版本号后提交更改到 Git",
+    )
+    return parser, parser.parse_args()
+
+
+def main() -> None:
+    """主函数, 执行版本号管理命令."""
+    parser, args = parse_args()
+
+    # 获取命令调度器
+    scheduler: CommandScheduler | None = create_scheduler_map().get(args.action, None)
+    if not scheduler:
+        logger.error(f"无效的操作: {args.action}")
+        parser.print_help(sys.stderr)
+        sys.exit(1)
+
+    # 执行版本号递增
+    scheduler.run()
+
+    # 如果指定了 tag 或 commit, 需要获取更新后的版本号并执行相应操作
+    if args.tag or args.commit:
+        # 从 BumpProjectVersionCommand 中获取更新后的文件列表
+        for cmd in scheduler.commands:
+            if isinstance(cmd, BumpProjectVersionCommand) and cmd.updated_files:
+                # 从第一个更新的文件中读取新版本号
+                from bitool.cmd.version import BumpFileVersionCommand
+
+                temp_bump_cmd = BumpFileVersionCommand(filepath=cmd.updated_files[0])
+                new_version = temp_bump_cmd._read_version()
+
+                if not new_version:
+                    logger.warning("未能读取到新版本号, 跳过 Git 操作")
+                    break
+
+                logger.info(f"检测到新版本号: {new_version}")
+
+                # 创建 git tag（如果指定）
+                if args.tag:
+                    tag_cmd = GitTagCommand(
+                        tag_name=f"v{new_version}", message=f"Release {new_version}"
+                    )
+                    tag_cmd.validate_and_run()
+
+                # 提交更改（如果指定）
+                if args.commit:
+                    commit_cmd = GitCommitCommand(
+                        message=f"chore: bump version to {new_version}"
+                    )
+                    commit_cmd.validate_and_run()
+
+                break
+
+
+if __name__ == "__main__":
+    main()

bitool/scripts/clearscreen.py

@@ -0,0 +1,37 @@
+"""跨平台控制台清屏工具."""
+
+from __future__ import annotations
+
+import sys
+
+from bitool.cmd import BuiltinConditions, CommandScheduler, RunCommand
+from bitool.consts import Constants
+
+
+def reset_cursor() -> None:
+    """将光标重置到控制台的起始位置."""
+    sys.stdout.write("\033[H")
+    sys.stdout.flush()
+
+
+def create_scheduler() -> CommandScheduler:
+    return CommandScheduler(
+        commands=[
+            RunCommand(
+                cmd=["cls" if Constants.IS_WINDOWS else "clear"],
+                allow_conditions=[BuiltinConditions.IS_WINDOWS],
+            ),
+            RunCommand(cmd=reset_cursor),
+        ]
+    )
+
+
+def main() -> None:
+    """运行命令行界面入口点.
+
+    Returns
+    -------
+        int: 退出码 (0 表示成功,非 0 表示失败)
+    """
+    scheduler = create_scheduler()
+    scheduler.run()

bitool/scripts/envpy.py

@@ -0,0 +1,161 @@
+"""Python环境配置工具模块。
+
+用于设置pip镜像源, 支持清华和阿里云等国内镜像源。
+"""
+
+from __future__ import annotations
+
+import argparse
+from pathlib import Path
+
+from bitool.cmd import CommandScheduler, SetEnvCommand, WriteFileCommand
+from bitool.consts import Constants
+from bitool.core import ConfigMixin
+
+
+class EnvPyConfig(ConfigMixin):
+    """Python环境配置类。"""
+
+    # pip 镜像源配置
+    PIP_DEFAULT_INDEX_URL: str = "https://pypi.org/simple"
+    PIP_INDEX_URLS: dict[str, str] = {  # noqa: RUF012
+        "tsinghua": "https://pypi.tuna.tsinghua.edu.cn/simple",
+        "aliyun": "https://mirrors.aliyun.com/pypi/simple/",
+    }
+    PIP_TRUSTED_HOSTS: dict[str, str] = {  # noqa: RUF012
+        "tsinghua": "pypi.tuna.tsinghua.edu.cn",
+        "aliyun": "mirrors.aliyun.com",
+    }
+
+    # UV相关配置
+    UV_INDEX_URL: str = "https://mirrors.aliyun.com/pypi/simple/"
+    UV_DEFAULT_INDEX: str = "https://mirrors.aliyun.com/pypi/simple/"
+    UV_HTTP_TIMEOUT: int = 600
+    UV_LINK_MODE: str = "copy"
+
+    # Conda相关配置
+    CONDA_DEFAULT_MIRROR = "tsinghua"
+    _CONDA_MIRROR_URLS: dict[str, frozenset[str]] = {  # noqa: RUF012
+        "tsinghua": frozenset(
+            [
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/",
+                "https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/",
+            ]
+        ),
+        "ustc": frozenset(
+            [
+                "https://mirrors.ustc.edu.cn/anaconda/pkgs/main/",
+                "https://mirrors.ustc.edu.cn/anaconda/pkgs/free/",
+                "https://mirrors.ustc.edu.cn/anaconda/pkgs/r/",
+                "https://mirrors.ustc.edu.cn/anaconda/pkgs/msys2/",
+                "https://mirrors.ustc.edu.cn/anaconda/pkgs/pro/",
+                "https://mirrors.ustc.edu.cn/anaconda/pkgs/dev/",
+                "https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/",
+                "https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/",
+                "https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/",
+                "https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/",
+            ]
+        ),
+        "bsfu": frozenset(
+            [
+                "https://mirrors.bsfu.edu.cn/anaconda/pkgs/main/",
+                "https://mirrors.bsfu.edu.cn/anaconda/pkgs/free/",
+                "https://mirrors.bsfu.edu.cn/anaconda/pkgs/r/",
+                "https://mirrors.bsfu.edu.cn/anaconda/pkgs/msys2/",
+                "https://mirrors.bsfu.edu.cn/anaconda/pkgs/pro/",
+                "https://mirrors.bsfu.edu.cn/anaconda/pkgs/dev/",
+                "https://mirrors.bsfu.edu.cn/anaconda/cloud/conda-forge/",
+                "https://mirrors.bsfu.edu.cn/anaconda/cloud/bioconda/",
+                "https://mirrors.bsfu.edu.cn/anaconda/cloud/menpo/",
+                "https://mirrors.bsfu.edu.cn/anaconda/cloud/pytorch/",
+            ]
+        ),
+        "aliyun": frozenset(
+            [
+                "https://mirrors.aliyun.com/anaconda/pkgs/main/",
+                "https://mirrors.aliyun.com/anaconda/pkgs/free/",
+                "https://mirrors.aliyun.com/anaconda/pkgs/r/",
+                "https://mirrors.aliyun.com/anaconda/pkgs/msys2/",
+                "https://mirrors.aliyun.com/anaconda/pkgs/pro/",
+                "https://mirrors.aliyun.com/anaconda/pkgs/dev/",
+                "https://mirrors.aliyun.com/anaconda/cloud/conda-forge/",
+                "https://mirrors.aliyun.com/anaconda/cloud/bioconda/",
+                "https://mirrors.aliyun.com/anaconda/cloud/menpo/",
+                "https://mirrors.aliyun.com/anaconda/cloud/pytorch/",
+            ]
+        ),
+    }
+
+
+conf = EnvPyConfig()
+
+
+def create_scheduler(mirror: str = "tsinghua") -> CommandScheduler:
+    """创建设置pip镜像源的命令调度器。
+
+    Args:
+        mirror: 镜像源名称, 默认为"tsinghua"。
+
+    Returns:
+        命令调度器实例。
+    """
+    return CommandScheduler(
+        commands=[
+            SetEnvCommand(
+                envs=[
+                    (
+                        "PIP_INDEX_URL",
+                        conf.PIP_INDEX_URLS.get(mirror, conf.PIP_DEFAULT_INDEX_URL),
+                    ),
+                    ("UV_INDEX_URL", conf.UV_INDEX_URL),
+                    ("UV_DEFAULT_INDEX", conf.UV_DEFAULT_INDEX),
+                    ("UV_HTTP_TIMEOUT", str(conf.UV_HTTP_TIMEOUT)),
+                    ("UV_LINK_MODE", conf.UV_LINK_MODE),
+                ]
+            ),
+            WriteFileCommand(
+                filepath=Path.home() / ".pip" / "pip.conf"
+                if Constants.IS_UNIX
+                else Path.home() / "pip" / "pip.ini",
+                content=f"[global]\nindex-url = {conf.PIP_INDEX_URLS.get(mirror, conf.PIP_DEFAULT_INDEX_URL)}\
+                    \n[install]\ntrusted-host = {conf.PIP_TRUSTED_HOSTS.get(mirror, '')}",
+            ),
+            WriteFileCommand(
+                filepath=Path.home() / ".condarc",
+                content=f"show_channel_urls: true\nchannels:\n  - {conf._CONDA_MIRROR_URLS.get(mirror, conf._CONDA_MIRROR_URLS['tsinghua'])}\n  - defaults",
+            ),
+        ]
+    )
+
+
+def parse_args() -> argparse.Namespace:
+    """解析命令行参数。
+
+    Returns:
+        解析后的参数命名空间。
+    """
+    parser = argparse.ArgumentParser(description="设置Python包的pip镜像源")
+    parser.add_argument(
+        "mirror",
+        type=str,
+        nargs="?",
+        default="tsinghua",
+        choices=conf.PIP_INDEX_URLS.keys(),
+        help="要使用的pip镜像源",
+    )
+    return parser.parse_args()
+
+
+def main() -> None:
+    """主函数, 解析命令行参数并运行调度器。"""
+    args = parse_args()
+
+    scheduler = create_scheduler(mirror=args.mirror)
+    scheduler.run()

bitool/scripts/envrs.py

@@ -0,0 +1,119 @@
+"""设置 Rust 国内镜像环境变量."""
+
+from __future__ import annotations
+
+import argparse
+from pathlib import Path
+
+from bitool.cmd import (
+    BuiltinConditions,
+    CommandScheduler,
+    RunCommand,
+    SetEnvCommand,
+    WriteFileCommand,
+)
+from bitool.core import ConfigMixin, logger
+
+
+class EnvRsConfig(ConfigMixin):
+    """EnvRs 配置类."""
+
+    DEFAULT_INSTALL_VERSION = "nightly"
+    TOML_CONTENT = """[source.crates-io]
+replace-with = 'aliyun'
+
+[source.aliyun]
+registry = "sparse+https://mirrors.aliyun.com/crates.io-index/"
+    """
+
+    RUSTUP_DIST_SERVER = "https://mirrors.aliyun.com/rustup"
+    RUSTUP_UPDATE_ROOT = "https://mirrors.aliyun.com/rustup/rustup"
+    RUSTUP_DOWNLOAD_URL = "https://static.rust-lang.org/rustup/dist/x86_64-pc-windows-msvc/rustup-init.exe"
+
+
+conf = EnvRsConfig()
+
+
+def create_scheduler(version: str = conf.DEFAULT_INSTALL_VERSION) -> CommandScheduler:
+    """根据版本参数动态创建命令调度器.
+
+    Parameters
+    ----------
+    version : str
+        Rust 版本,默认为 "nightly"
+
+    Returns
+    -------
+    CommandScheduler
+        配置好命令的调度器实例
+    """
+    return CommandScheduler(
+        commands=[
+            SetEnvCommand(
+                envs=[
+                    ("RUSTUP_DIST_SERVER", conf.RUSTUP_DIST_SERVER),
+                    ("RUSTUP_UPDATE_ROOT", conf.RUSTUP_UPDATE_ROOT),
+                ]
+            ),
+            RunCommand(
+                cmd=[
+                    "powershell",
+                    "-Command",
+                    "Invoke-WebRequest",
+                    "-Uri",
+                    "https://static.rust-lang.org/rustup/dist/x86_64-pc-windows-msvc/rustup-init.exe",
+                    "-OutFile",
+                    "rustup-init.exe",
+                ],
+                allow_conditions=[
+                    BuiltinConditions.IS_WINDOWS,
+                    BuiltinConditions.HAS_APP_INSTALLED("rustup"),
+                ],
+                forbid_conditions=[
+                    BuiltinConditions.HAS_ENTRY_IN_DIR("rustup-init.exe", Path.cwd())
+                ],
+                timeout=240,
+            ),
+            RunCommand(
+                cmd=[
+                    "curl",
+                    "--proto",
+                    "=https",
+                    "--tlsv1.2",
+                    "-sSf",
+                    "https://mirrors.aliyun.com/repo/rust/rustup-init.sh",
+                    "|",
+                    "sh",
+                ],
+                forbid_conditions=[BuiltinConditions.IS_WINDOWS],
+            ),
+            WriteFileCommand(
+                filepath=Path.home() / ".rustup" / "config.toml",
+                content=conf.TOML_CONTENT,
+            ),
+            RunCommand(cmd=["rustup", "toolchain", "install", version]),
+        ]
+    )
+
+
+def parse_args() -> argparse.Namespace:
+    import argparse
+
+    parser = argparse.ArgumentParser(description="设置 Rust 国内镜像环境变量")
+    parser.add_argument(
+        "version", type=str, nargs="?", default="nightly", help="Rust 版本"
+    )
+    return parser.parse_args()
+
+
+def main() -> None:
+    """设置 Rust 镜像环境变量并安装指定版本."""
+    args = parse_args()
+    logger.info(f"安装 Rust {args.version} 版本")
+
+    scheduler = create_scheduler(args.version)
+    scheduler.run()
+
+
+if __name__ == "__main__":
+    main()