ENH: Allow int for indirect_reference in PdfWriter.get_object

py-pdf · Dec 11, 2022 · 6d96232 · 6d96232
1 parent 437aec9
commit 6d96232
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/PyPDF2/_writer.py b/PyPDF2/_writer.py
@@ -204,7 +204,7 @@ def _add_object(self, obj: PdfObject) -> IndirectObject:
 
     def get_object(
         self,
-        indirect_reference: Optional[IndirectObject] = None,
+        indirect_reference: Union[None, int, IndirectObject] = None,
         ido: Optional[IndirectObject] = None,
     ) -> PdfObject:
         if ido is not None:  # deprecated
@@ -221,6 +221,8 @@ def get_object(
         assert (
             indirect_reference is not None
         )  # the None value is only there to keep the deprecated name
+        if isinstance(indirect_reference, int):
+            return self._objects[indirect_reference - 1]
         if indirect_reference.pdf != self:
             raise ValueError("pdf must be self")
         return self._objects[indirect_reference.idnum - 1]  # type: ignore