From 0402510f7bd6704f19aaf19300143e3a5b3d86a2 Mon Sep 17 00:00:00 2001 From: Martin Thoma Date: Sun, 12 Jun 2022 09:52:24 +0200 Subject: [PATCH] STY: Use more tuples and list/dict comprehensions --- PyPDF2/_page.py | 8 ++++---- PyPDF2/_reader.py | 9 ++++----- PyPDF2/_writer.py | 12 ++++++------ PyPDF2/constants.py | 4 ++-- PyPDF2/papersizes.py | 4 ++-- 5 files changed, 18 insertions(+), 19 deletions(-) diff --git a/PyPDF2/_page.py b/PyPDF2/_page.py index ae44506e6..cb4648cb1 100644 --- a/PyPDF2/_page.py +++ b/PyPDF2/_page.py @@ -548,13 +548,13 @@ def _merge_page( def _expand_mediabox( self, page2: "PageObject", ctm: Optional[CompressedTransformationMatrix] ) -> None: - corners1 = [ + corners1 = ( self.mediabox.left.as_numeric(), self.mediabox.bottom.as_numeric(), self.mediabox.right.as_numeric(), self.mediabox.top.as_numeric(), - ] - corners2 = [ + ) + corners2 = ( page2.mediabox.left.as_numeric(), page2.mediabox.bottom.as_numeric(), page2.mediabox.left.as_numeric(), @@ -563,7 +563,7 @@ def _expand_mediabox( page2.mediabox.top.as_numeric(), page2.mediabox.right.as_numeric(), page2.mediabox.bottom.as_numeric(), - ] + ) if ctm is not None: ctm = tuple(float(x) for x in ctm) # type: ignore[assignment] new_x = [ diff --git a/PyPDF2/_reader.py b/PyPDF2/_reader.py index 9babe7195..3f6292832 100644 --- a/PyPDF2/_reader.py +++ b/PyPDF2/_reader.py @@ -525,7 +525,7 @@ def _check_kids( self.get_fields(kid.get_object(), retval, fileobj) def _write_field(self, fileobj: Any, field: Any, field_attributes: Any) -> None: - order = ["/TM", "/T", "/FT", PA.PARENT, "/TU", "/Ff", "/V", "/DV"] + order = ("/TM", "/T", "/FT", PA.PARENT, "/TU", "/Ff", "/V", "/DV") for attr in order: attr_name = field_attributes[attr] try: @@ -701,10 +701,9 @@ def _get_page_number_by_indirect( ) -> int: """Generate _page_id2num""" if self._page_id2num is None: - id2num = {} - for i, x in enumerate(self.pages): - id2num[x.indirect_ref.idnum] = i # type: ignore - self._page_id2num = id2num + self._page_id2num = { + x.indirect_ref.idnum: i for i, x in enumerate(self.pages) + } if indirect_ref is None or isinstance(indirect_ref, NullObject): return -1 diff --git a/PyPDF2/_writer.py b/PyPDF2/_writer.py index a070b3387..c0528eeaa 100644 --- a/PyPDF2/_writer.py +++ b/PyPDF2/_writer.py @@ -1209,7 +1209,7 @@ def remove_images(self, ignore_byte_string_object: bool = False) -> None: """ pg_dict = cast(DictionaryObject, self.get_object(self._pages)) pages = cast(ArrayObject, pg_dict[PA.KIDS]) - jump_operators = [ + jump_operators = ( b_("cm"), b_("w"), b_("J"), @@ -1235,7 +1235,7 @@ def remove_images(self, ignore_byte_string_object: bool = False) -> None: b_("B"), b_("Do"), b_("sh"), - ] + ) for j in range(len(pages)): page = pages[j] page_ref = cast(DictionaryObject, self.get_object(page)) @@ -1539,7 +1539,7 @@ def addLink( # pragma: no cover deprecate_with_replacement("addLink", "add_link") return self.add_link(pagenum, pagedest, rect, border, fit, *args) - _valid_layouts = [ + _valid_layouts = ( "/NoLayout", "/SinglePage", "/OneColumn", @@ -1547,7 +1547,7 @@ def addLink( # pragma: no cover "/TwoColumnRight", "/TwoPageLeft", "/TwoPageRight", - ] + ) def _get_page_layout(self) -> Optional[LayoutType]: try: @@ -1656,14 +1656,14 @@ def pageLayout(self, layout: LayoutType) -> None: # pragma: no cover deprecate_with_replacement("pageLayout", "page_layout") self.page_layout = layout - _valid_modes = [ + _valid_modes = ( "/UseNone", "/UseOutlines", "/UseThumbs", "/FullScreen", "/UseOC", "/UseAttachments", - ] + ) def _get_page_mode(self) -> Optional[PagemodeType]: try: diff --git a/PyPDF2/constants.py b/PyPDF2/constants.py index a62aea1a4..4a6be917e 100644 --- a/PyPDF2/constants.py +++ b/PyPDF2/constants.py @@ -285,7 +285,7 @@ class CatalogDictionary: NEEDS_RENDERING = "/NeedsRendering" # boolean, optional -PDF_KEYS = [ +PDF_KEYS = ( PagesAttributes, PageAttributes, Ressources, @@ -302,4 +302,4 @@ class CatalogDictionary: Core, TrailerKeys, CatalogAttributes, -] +) diff --git a/PyPDF2/papersizes.py b/PyPDF2/papersizes.py index 5ba49d698..51aa2de59 100644 --- a/PyPDF2/papersizes.py +++ b/PyPDF2/papersizes.py @@ -35,7 +35,7 @@ class PaperSize: C4 = Dimensions(649, 918) -_din_a = [ +_din_a = ( PaperSize.A0, PaperSize.A1, PaperSize.A2, @@ -45,4 +45,4 @@ class PaperSize: PaperSize.A6, PaperSize.A7, PaperSize.A8, -] +)