bitool 0.1.2__py3-none-any.whl

bitool/scripts/img2pdf.py

@@ -0,0 +1,151 @@
+"""图片转PDF脚本.
+
+将多张图片合并成一个PDF文件, 支持多种图片格式和页面设置.
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+from bitool.cmd import CommandScheduler, ImagesToPdfCommand
+from bitool.core import ConfigMixin, logger
+
+
+class ImageToPdfConfig(ConfigMixin):
+    """图片转PDF配置类."""
+
+    FORMAT: str = "pdf"
+    PAGE_SIZE: str = "auto"
+    VALID_IMAGE_FORMATS: set[str] = {"png", "jpg", "jpeg", "bmp", "gif", "tiff", "tif"}  # noqa: RUF012
+    VALID_OUTPUT_FORMATS: set[str] = {"pdf"}  # noqa: RUF012
+
+
+conf = ImageToPdfConfig()
+
+
+def create_scheduler(
+    image_paths: list[Path], output_path: Path, page_size: str = "auto"
+) -> CommandScheduler:
+    """创建图片转PDF命令调度器.
+
+    Args:
+        image_paths: 输入图片文件路径列表
+        output_path: 输出PDF文件路径
+        page_size: 页面大小设置
+
+    Returns:
+        命令调度器实例
+    """
+    return CommandScheduler(
+        commands=[
+            ImagesToPdfCommand(
+                image_paths=image_paths, output_path=output_path, page_size=page_size
+            )
+        ]
+    )
+
+
+def parse_args() -> argparse.Namespace:
+    """解析命令行参数.
+
+    Returns:
+        解析后的参数命名空间
+    """
+    parser = argparse.ArgumentParser(description="将图片转换为PDF")
+    parser.add_argument(
+        "input_paths",
+        type=str,
+        nargs="*",
+        default=None,
+        help="输入图片文件路径(可选, 默认搜索当前目录)",
+    )
+    parser.add_argument(
+        "output_path", type=str, nargs="?", default=None, help="输出PDF文件路径"
+    )
+    parser.add_argument(
+        "--page-size",
+        type=str,
+        default=conf.PAGE_SIZE,
+        help="页面大小 (auto, A4, letter等)",
+    )
+    return parser.parse_args()
+
+
+def main() -> None:
+    """图片转PDF主函数."""
+    args = parse_args()
+
+    # 验证输入路径
+    if args.input_paths:
+        input_paths = [Path(p) for p in args.input_paths]
+    else:
+        # 自动搜索当前目录下的图片文件
+        current_dir = Path.cwd()
+        input_paths = []
+        for ext in conf.VALID_IMAGE_FORMATS:
+            input_paths.extend(current_dir.glob(f"*.{ext}"))
+            input_paths.extend(current_dir.glob(f"*.{ext.upper()}"))
+
+        if not input_paths:
+            logger.error(f"在当前目录 {current_dir} 中未找到图片文件")
+            sys.exit(1)
+
+        # 去重并排序
+        input_paths = sorted(set(input_paths))
+        logger.info(f"在当前目录 {current_dir} 中找到 {len(input_paths)} 个图片文件")
+        logger.info(f"将处理: {', '.join([str(p.name) for p in input_paths])}")
+
+    for input_path in input_paths:
+        if not input_path.exists():
+            logger.error(f"未找到输入文件: {input_path}")
+            sys.exit(1)
+        # 验证图片格式
+        if input_path.suffix.lower().lstrip(".") not in conf.VALID_IMAGE_FORMATS:
+            logger.error(
+                f"不支持的图片格式: {input_path.suffix}, 请选择 {conf.VALID_IMAGE_FORMATS}"
+            )
+            sys.exit(1)
+
+    # 确定输出路径
+    output_path = Path(args.output_path) if args.output_path else None
+    if output_path is None:
+        # 默认输出到第一个图片所在目录, 命名为 output.pdf
+        output_path = input_paths[0].parent / "output.pdf"
+
+    # 确保输出目录存在
+    if not output_path.parent.exists():
+        output_path.parent.mkdir(parents=True)
+        logger.info(f"创建输出目录: {output_path.parent}")
+
+    # 验证输出格式
+    if output_path.suffix.lower().lstrip(".") not in conf.VALID_OUTPUT_FORMATS:
+        logger.error(
+            f"不支持的输出格式: {output_path.suffix}, 请选择 {conf.VALID_OUTPUT_FORMATS}"
+        )
+        sys.exit(1)
+
+    logger.info(
+        f"开始转换: {len(input_paths)} 张图片 -> {output_path} (页面大小={args.page_size})"
+    )
+
+    try:
+        scheduler = create_scheduler(
+            image_paths=input_paths, output_path=output_path, page_size=args.page_size
+        )
+        success = scheduler.run()
+
+        if success:
+            logger.info("图片转PDF完成")
+            sys.exit(0)
+        else:
+            logger.error("图片转PDF失败")
+            sys.exit(1)
+
+    except KeyboardInterrupt:
+        logger.warning("用户中断操作")
+        sys.exit(130)
+    except Exception as e:  # noqa: BLE001
+        logger.error(f"转换过程发生错误: {e}")
+        sys.exit(1)

bitool/scripts/pdf2img.py

@@ -0,0 +1,139 @@
+"""PDF转图片脚本.
+
+将PDF文件的每一页转换为图片格式, 支持多种输出格式和分辨率设置.
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+from bitool.cmd import CommandScheduler, PdfToImagesCommand
+from bitool.core import ConfigMixin, logger
+
+
+class PdfToImageConfig(ConfigMixin):
+    """PDF转图片配置类."""
+
+    DPI: int = 150
+    DPI_RANGE: tuple[int, int] = (72, 300)
+    FORMAT: str = "png"
+    QUALITY: int = 90
+    QUALITY_RANGE: tuple[int, int] = (1, 100)
+    VALID_FORMATS: set[str] = {"png", "jpg", "jpeg"}  # noqa: RUF012
+
+
+conf = PdfToImageConfig()
+
+
+def create_scheduler(
+    input_path: Path, output_dir: Path, dpi: int, fmt: str = "png"
+) -> CommandScheduler:
+    """创建PDF转图片命令调度器.
+
+    Args:
+        input_path: 输入PDF文件路径
+        output_dir: 输出图片目录
+        dpi: 图片分辨率
+        fmt: 图片格式
+
+    Returns:
+        命令调度器实例
+    """
+    return CommandScheduler(
+        commands=[
+            PdfToImagesCommand(
+                input_path=input_path, output_dir=output_dir, dpi=dpi, fmt=fmt
+            )
+        ]
+    )
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description="将PDF转换为图片")
+    parser.add_argument(
+        "input_path",
+        type=str,
+        nargs="?",
+        default=None,
+        help="输入PDF文件路径(可选, 默认搜索当前目录)",
+    )
+    parser.add_argument(
+        "output_dir", type=str, nargs="?", default=None, help="输出目录"
+    )
+    parser.add_argument("--format", type=str, default=conf.FORMAT, help="输出图片格式")
+    parser.add_argument("--dpi", type=int, default=conf.DPI, help="图片分辨率(DPI)")
+    parser.add_argument("--quality", type=int, default=conf.QUALITY, help="图片质量")
+    return parser.parse_args()
+
+
+def main() -> None:
+    """PDF转图片主函数."""
+    args = parse_args()
+
+    # 验证输入路径
+    input_path = Path(args.input_path) if args.input_path else Path.cwd()
+
+    # 如果输入路径是目录, 搜索该目录下的所有PDF文件
+    if input_path.is_dir():
+        pdf_files = list(input_path.glob("*.pdf")) + list(input_path.glob("*.PDF"))
+        if not pdf_files:
+            logger.error(f"在目录 {input_path} 中未找到PDF文件")
+            sys.exit(1)
+        logger.info(f"在目录 {input_path} 中找到 {len(pdf_files)} 个PDF文件")
+        # 处理第一个找到的PDF文件
+        input_path = pdf_files[0]
+        logger.info(f"将处理: {input_path}")
+
+    if not input_path.exists():
+        logger.error(f"未找到输入路径: {input_path}")
+        sys.exit(1)
+
+    # 确定输出目录
+    output_dir = Path(args.output_dir) if args.output_dir else None
+    if output_dir is None:
+        output_dir = input_path.parent / "converted_images"
+
+    if not output_dir.exists():
+        output_dir.mkdir(parents=True)
+        logger.info(f"创建输出目录: {output_dir}")
+
+    # 验证参数范围
+    if not conf.QUALITY_RANGE[0] <= args.quality <= conf.QUALITY_RANGE[1]:
+        logger.error(
+            f"图片质量必须在 {conf.QUALITY_RANGE[0]} 到 {conf.QUALITY_RANGE[1]} 之间"
+        )
+        sys.exit(1)
+
+    if not conf.DPI_RANGE[0] <= args.dpi <= conf.DPI_RANGE[1]:
+        logger.error(f"DPI必须在 {conf.DPI_RANGE[0]} 到 {conf.DPI_RANGE[1]} 之间")
+        sys.exit(1)
+
+    if args.format not in conf.VALID_FORMATS:
+        logger.error(f"无效的图片格式, 请选择 {conf.VALID_FORMATS}")
+        sys.exit(1)
+
+    logger.info(
+        f"开始转换: {input_path} -> {output_dir} (DPI={args.dpi}, 格式={args.format})"
+    )
+
+    try:
+        scheduler = create_scheduler(
+            input_path=input_path, output_dir=output_dir, dpi=args.dpi, fmt=args.format
+        )
+        success = scheduler.run()
+
+        if success:
+            logger.info("PDF转图片完成")
+            sys.exit(0)
+        else:
+            logger.error("PDF转图片失败")
+            sys.exit(1)
+
+    except KeyboardInterrupt:
+        logger.warning("用户中断操作")
+        sys.exit(130)
+    except Exception as e:  # noqa: BLE001
+        logger.error(f"转换过程发生错误: {e}")
+        sys.exit(1)

bitool/scripts/piptool.py

@@ -0,0 +1,130 @@
+"""pip工具模块。
+
+提供pip包管理操作的封装，支持安装、重新安装和下载功能。
+支持在线和离线两种模式。
+"""
+
+from __future__ import annotations
+
+import argparse
+from typing import Final
+
+from ..cmd import CommandScheduler, RunCommand, RunShellCommand
+from ..core import ConfigMixin
+
+
+class PipToolConfig(ConfigMixin):
+    """pip工具配置类。"""
+
+    TRUSTED_HOST: Final[list[str]] = [
+        "--trusted-host",
+        "mirrors.aliyun.com",
+        "--index-url",
+        "http://mirrors.aliyun.com/pypi/simple/",
+    ]
+    PACKAGE_DIR: Final[str] = "packages"
+    REQUIREMENTS_FILE: Final[str] = "requirements.txt"
+
+
+conf = PipToolConfig()
+
+
+def create_scheduler_map(
+    libnames: list[str] | None = None,
+    *,
+    use_requirements: bool = False,
+    offline: bool = False,
+) -> dict[str, CommandScheduler]:
+    """创建pip操作的命令调度器映射。
+
+    Args:
+        libnames: 要安装的库名称列表。
+        use_requirements: 是否使用requirements文件。
+        offline: 是否使用离线模式。
+
+    Returns:
+        命令调度器的字典映射。
+    """
+    # 修复类型注解冲突：使用新变量名
+    pkg_names: list[str] = [] if libnames is None else libnames
+    options: list[str] = (
+        ["--no-index", "--find-links", "."] if offline else conf.TRUSTED_HOST.copy()
+    )
+
+    if use_requirements:
+        options.extend(["-r", conf.REQUIREMENTS_FILE])
+    else:
+        options.extend(pkg_names)
+
+    return {
+        # 安装
+        "i": CommandScheduler(commands=[RunCommand(cmd=["pip", "install", *options])]),
+        # 重新安装
+        "r": CommandScheduler(
+            commands=[
+                RunCommand(cmd=["pip", "uninstall", "-y", *pkg_names]),
+                RunCommand(cmd=["pip", "install", *options]),
+            ]
+        ),
+        # 下载
+        "d": CommandScheduler(
+            commands=[
+                RunCommand(cmd=["pip", "download", *options, "-d", conf.PACKAGE_DIR])
+            ]
+        ),
+        # 冻结依赖
+        "f": CommandScheduler(
+            commands=[
+                RunShellCommand(
+                    shell_cmd=f"pip freeze --exclude-editable > {conf.REQUIREMENTS_FILE}"
+                )
+            ]
+        ),
+        # 卸载
+        "u": CommandScheduler(
+            commands=[RunCommand(cmd=["pip", "uninstall", "-y", *pkg_names])]
+        ),
+        # 升级pip
+        "up": CommandScheduler(
+            commands=[
+                RunCommand(cmd=["python", "-m", "pip", "install", "--upgrade", "pip"])
+            ]
+        ),
+    }
+
+
+OPTIONS: list[str] = list(create_scheduler_map().keys())
+
+
+def parse_args() -> argparse.Namespace:
+    """解析命令行参数。
+
+    Returns:
+        解析后的参数命名空间。
+    """
+    parser = argparse.ArgumentParser(description="pip安装工具")
+    parser.add_argument("action", choices=OPTIONS, help="要执行的操作")
+    parser.add_argument("libnames", nargs="*", help="要安装的库名称")
+    parser.add_argument("--offline", "-o", action="store_true", help="离线模式")
+    parser.add_argument(
+        "--requirements", "-r", action="store_true", help="使用requirements文件"
+    )
+    return parser.parse_args()
+
+
+def main() -> None:
+    """主函数，执行pip工具命令。"""
+    args = parse_args()
+
+    # 验证参数：使用requirements模式时不需要指定库名
+    if args.action in ["i", "r", "d"] and not args.libnames and not args.requirements:
+        raise ValueError("请指定要安装的库名称，或使用 --requirements 参数")
+
+    scheduler_map: dict[str, CommandScheduler] = create_scheduler_map(
+        args.libnames, offline=args.offline, use_requirements=args.requirements
+    )
+    scheduler: CommandScheduler | None = scheduler_map.get(args.action)
+    if not scheduler:
+        raise ValueError(f"无效的操作: {args.action}")
+
+    scheduler.run()