pdate pdfly/extract_annotated_pages.py

Seems that the `| None` syntax is not valid with Python 3.8

Co-authored-by: Lucas Cimon <[email protected]>

Author: wolfram77

pdfly/cli.py

@@ -286,7 +286,7 @@ def update_offsets(
     ],
     file_out: Annotated[
         Path, typer.Option("-o", "--output")  # noqa
-    ] = None,  # type: ignore
+    ] = None,  # type: ignore[assignment]
     encoding: str = typer.Option(
         "ISO-8859-1",
         help="Encoding used to read and write the files, e.g. UTF-8.",

pdfly/extract_annotated_pages.py

@@ -6,26 +6,31 @@
 """
 
 from pathlib import Path
+from typing import Optional
+
 from pypdf import PdfReader, PdfWriter
+from pypdf.annotations import AnnotationDictionary
+from pypdf.generic import ArrayObject  # noqa: TCH002
 
 
 # Check if an annotation is manipulable.
-def is_manipulable(annot) -> bool:
+def is_manipulable(annot: AnnotationDictionary) -> bool:
     return annot.get("/Subtype") not in ["/Link"]
 
 
 # Main function.
-def main(input_pdf: Path, output_pdf: Path) -> None:
+def main(input_pdf: Path, output_pdf: Optional[Path]) -> None:
     if not output_pdf:
-        output_pdf = input_pdf.with_stem(input_pdf.stem + "_annotated")
+        output_pdf = input_pdf.with_name(input_pdf.stem + "_annotated.pdf")
     input = PdfReader(input_pdf)
     output = PdfWriter()
     output_pages = 0
     # Copy only the pages with annotations
     for page in input.pages:
-        if not "/Annots" in page:
+        if "/Annots" not in page:
             continue
-        if not any(is_manipulable(annot) for annot in page["/Annots"]):
+        page_annots: ArrayObject = page["/Annots"]  # type: ignore[assignment]
+        if not any(is_manipulable(annot) for annot in page_annots):
             continue
         output.add_page(page)
         output_pages += 1