pysfi 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

sfi/pdfsplit/pdfsplit.py

@@ -7,7 +7,6 @@ from pathlib import Path
 import fitz
 
 logging.basicConfig(level=logging.INFO, format="%(message)s")
-cwd = Path.cwd()
 logger = logging.getLogger(__name__)
 
 
@@ -50,7 +49,9 @@ def split_by_number(input_file: Path, output_file: Path, number: int) -> None:
 
         end_page = min(current_page + pages_in_this_part, total_pages)
 
-        part_file = output_file.parent / f"{output_file.stem}_part{i + 1}{output_file.suffix}"
+        part_file = (
+            output_file.parent / f"{output_file.stem}_part{i + 1}{output_file.suffix}"
+        )
         part_doc = fitz.open()
 
         for page_num in range(current_page, end_page):
@@ -58,7 +59,9 @@ def split_by_number(input_file: Path, output_file: Path, number: int) -> None:
 
         part_doc.save(part_file)
         part_doc.close()
-        logger.info(f"Created part {i + 1}: {part_file} (pages {current_page + 1}-{end_page})")
+        logger.info(
+            f"Created part {i + 1}: {part_file} (pages {current_page + 1}-{end_page})"
+        )
 
         current_page = end_page
 
@@ -77,7 +80,10 @@ def split_by_size(input_file: Path, output_file: Path, size: int) -> None:
 
     while start_page < total_pages:
         end_page = min(start_page + size, total_pages)
-        part_file = output_file.parent / f"{output_file.stem}_part{part + 1}{output_file.suffix}"
+        part_file = (
+            output_file.parent
+            / f"{output_file.stem}_part{part + 1}{output_file.suffix}"
+        )
         part_doc = fitz.open()
 
         for page_num in range(start_page, end_page):
@@ -85,7 +91,9 @@ def split_by_size(input_file: Path, output_file: Path, size: int) -> None:
 
         part_doc.save(part_file)
         part_doc.close()
-        logger.info(f"Created part {part + 1}: {part_file} (pages {start_page + 1}-{end_page})")
+        logger.info(
+            f"Created part {part + 1}: {part_file} (pages {start_page + 1}-{end_page})"
+        )
 
         start_page = end_page
         part += 1
@@ -122,18 +130,37 @@ def split_by_range(input_file: Path, output_file: Path, range_str: str) -> None:
 
 
 def main() -> None:
+    """Main entry point for pdfsplit CLI."""
     parser = argparse.ArgumentParser(description="Split PDF files")
     parser.add_argument("input", help="Input PDF file")
-    parser.add_argument("output", nargs="?", help="Output PDF file (optional for -n and -s modes)")
-    parser.add_argument("-o", "--output-dir", default=str(cwd), help="Output directory (default: current directory)")
-    parser.add_argument("-f", "--output-format", help="Output file format pattern, e.g., 'split_{part:02d}.pdf'")
+    parser.add_argument(
+        "output", nargs="?", help="Output PDF file (optional for -n and -s modes)"
+    )
+    parser.add_argument(
+        "-o",
+        "--output-dir",
+        default=".",
+        help="Output directory (default: current directory)",
+    )
+    parser.add_argument(
+        "-f",
+        "--output-format",
+        help="Output file format pattern, e.g., 'split_{part:02d}.pdf'",
+    )
     parser.add_argument("-v", "--verbose", action="store_true", help="Verbose output")
 
     # Split by number, size, or range
     group = parser.add_mutually_exclusive_group(required=True)
     group.add_argument("-n", "--number", type=int, help="Number of splits")
-    group.add_argument("-s", "--size", type=int, default=1, help="Size of each split in pages")
-    group.add_argument("-r", "--range", type=str, help="Range of pages to extract, e.g., '1,2,4-10,15-20,25-'")
+    group.add_argument(
+        "-s", "--size", type=int, default=1, help="Size of each split in pages"
+    )
+    group.add_argument(
+        "-r",
+        "--range",
+        type=str,
+        help="Range of pages to extract, e.g., '1,2,4-10,15-20,25-'",
+    )
 
     args = parser.parse_args()
 
@@ -142,7 +169,9 @@ def main() -> None:
 
     output_dir = Path(args.output_dir)
     if not output_dir.is_dir():
-        logger.error(f"Output directory {args.output_dir} does not exist, please check the path.")
+        logger.error(
+            f"Output directory {args.output_dir} does not exist, please check the path."
+        )
         return
 
     input_file = Path(args.input)
@@ -157,7 +186,11 @@ def main() -> None:
         return
 
     if not args.range:
-        output_file = output_dir / (input_file.stem + "_split.pdf") if not args.output else Path(args.output)
+        output_file = (
+            output_dir / (input_file.stem + "_split.pdf")
+            if not args.output
+            else Path(args.output)
+        )
     else:
         output_file = Path(args.output)
 

sfi/pyarchive/__init__.py

@@ -0,0 +1 @@
+