PyPDFForm 1.4.12__py3-none-any.whl → 1.4.13__py3-none-any.whl

PyPDFForm/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 """Contains any object users might need."""
 
-__version__ = "1.4.12"
+__version__ = "1.4.13"
 
-from .wrapper import PdfWrapper, PyPDFForm
+from .wrapper import FormWrapper, PdfWrapper, PyPDFForm

PyPDFForm/constants.py

@@ -26,6 +26,7 @@ WIDGET_TYPES = Union[Text, Checkbox, Radio, Dropdown, Signature]
 
 DEPRECATION_NOTICE = "{} will be deprecated soon. Use {} instead."
 
+ANNOTATION_KEY = "/Annots"
 ANNOTATION_FIELD_KEY = "/T"
 ANNOTATION_RECTANGLE_KEY = "/Rect"
 SUBTYPE_KEY = "/Subtype"
@@ -34,6 +35,8 @@ WIDGET_TYPE_KEY = "/FT"
 PARENT_KEY = "/Parent"
 FIELD_FLAG_KEY = "/Ff"
 TEXT_FIELD_IDENTIFIER = "/Tx"
+TEXT_VALUE_IDENTIFIER = "/V"
+TEXT_VALUE_SHOW_UP_IDENTIFIER = "/AP"
 SIGNATURE_FIELD_IDENTIFIER = "/Sig"
 TEXT_FIELD_APPEARANCE_IDENTIFIER = "/DA"
 SELECTABLE_IDENTIFIER = "/Btn"
@@ -43,6 +46,7 @@ CHOICE_FIELD_IDENTIFIER = "/Ch"
 CHOICES_IDENTIFIER = "/Opt"
 BUTTON_IDENTIFIER = "/MK"
 BUTTON_STYLE_IDENTIFIER = "/CA"
+SELECTED_IDENTIFIER = "/AS"
 
 # Field flag bits
 MULTILINE = 1 << 12
@@ -65,4 +69,6 @@ BUTTON_STYLES = {
     "l": "\u25CF",  # circle
 }
 
+CHECKBOX_SELECT = "/Yes"
+
 COORDINATE_GRID_FONT_SIZE_MARGIN_RATIO = DEFAULT_FONT_SIZE / 100

PyPDFForm/filler.py

@@ -1,9 +1,15 @@
 # -*- coding: utf-8 -*-
 """Contains helpers for filling a PDF form."""
 
-from typing import Dict
+from io import BytesIO
+from typing import Dict, cast
 
-from .constants import WIDGET_TYPES
+from pypdf import PdfReader, PdfWriter
+from pypdf.generic import DictionaryObject, NameObject, TextStringObject
+
+from .constants import (ANNOTATION_KEY, CHECKBOX_SELECT, SELECTED_IDENTIFIER,
+                        TEXT_VALUE_IDENTIFIER, TEXT_VALUE_SHOW_UP_IDENTIFIER,
+                        WIDGET_TYPES)
 from .coordinate import (get_draw_checkbox_radio_coordinates,
                          get_draw_sig_coordinates_resolutions,
                          get_draw_text_coordinates,
@@ -11,10 +17,12 @@ from .coordinate import (get_draw_checkbox_radio_coordinates,
 from .font import checkbox_radio_font_size
 from .image import any_image_to_jpg
 from .middleware.checkbox import Checkbox
+from .middleware.dropdown import Dropdown
 from .middleware.radio import Radio
 from .middleware.signature import Signature
+from .middleware.text import Text
 from .template import get_widget_key, get_widgets_by_page
-from .utils import checkbox_radio_to_draw
+from .utils import checkbox_radio_to_draw, stream_to_io
 from .watermark import create_watermarks_and_draw, merge_watermarks_with_pdf
 
 
@@ -117,3 +125,55 @@ def fill(
         result = merge_watermarks_with_pdf(result, image_watermarks)
 
     return result
+
+
+def simple_fill(
+    template: bytes,
+    widgets: Dict[str, WIDGET_TYPES],
+) -> bytes:
+    """Fills a PDF form in place."""
+
+    pdf = PdfReader(stream_to_io(template))
+    out = PdfWriter()
+    out.append(pdf)
+
+    radio_button_tracker = {}
+
+    for page in out.pages:
+        for annot in page.get(ANNOTATION_KEY, []):  # noqa
+            annot = cast(DictionaryObject, annot.get_object())
+            key = get_widget_key(annot.get_object())
+
+            widget = widgets.get(key)
+            if widget is None:
+                continue
+
+            if isinstance(widget, Checkbox) and widget.value is True:
+                annot[NameObject(SELECTED_IDENTIFIER)] = NameObject(CHECKBOX_SELECT)
+            elif isinstance(widget, Radio):
+                if key not in radio_button_tracker:
+                    radio_button_tracker[key] = 0
+                radio_button_tracker[key] += 1
+                if widget.value == radio_button_tracker[key] - 1:
+                    annot[NameObject(SELECTED_IDENTIFIER)] = NameObject(
+                        f"/{widget.value}"
+                    )
+            elif isinstance(widget, Dropdown) and widget.value is not None:
+                annot[NameObject(TEXT_VALUE_IDENTIFIER)] = TextStringObject(
+                    widget.choices[widget.value]
+                )
+                annot[NameObject(TEXT_VALUE_SHOW_UP_IDENTIFIER)] = TextStringObject(
+                    widget.choices[widget.value]
+                )
+            elif isinstance(widget, Text) and widget.value:
+                annot[NameObject(TEXT_VALUE_IDENTIFIER)] = TextStringObject(
+                    widget.value
+                )
+                annot[NameObject(TEXT_VALUE_SHOW_UP_IDENTIFIER)] = TextStringObject(
+                    widget.value
+                )
+
+    with BytesIO() as f:
+        out.write(f)
+        f.seek(0)
+        return f.read()

PyPDFForm/wrapper.py

@@ -11,7 +11,7 @@ from .constants import (DEFAULT_FONT, DEFAULT_FONT_COLOR, DEFAULT_FONT_SIZE,
                         DEPRECATION_NOTICE, VERSION_IDENTIFIER_PREFIX,
                         VERSION_IDENTIFIERS)
 from .coordinate import generate_coordinate_grid
-from .filler import fill
+from .filler import fill, simple_fill
 from .font import register_font
 from .image import any_image_to_jpg, rotate_image
 from .middleware.dropdown import Dropdown
@@ -26,7 +26,40 @@ from .widgets.checkbox import CheckBoxWidget
 from .widgets.text import TextWidget
 
 
-class PdfWrapper:
+class FormWrapper:
+    """A simple base wrapper for just filling a PDF form."""
+
+    def __init__(
+        self,
+        template: Union[bytes, str, BinaryIO] = b"",
+    ) -> None:
+        """Constructs all attributes for the object."""
+
+        self.stream = fp_or_f_obj_or_stream_to_stream(template)
+
+    def read(self) -> bytes:
+        """Reads the file stream of a PDF form."""
+
+        return self.stream
+
+    def fill(
+        self,
+        data: Dict[str, Union[str, bool, int]],
+    ) -> FormWrapper:
+        """Fills a PDF form."""
+
+        widgets = build_widgets(self.stream) if self.stream else {}
+
+        for key, value in data.items():
+            if key in widgets:
+                widgets[key].value = value
+
+        self.stream = simple_fill(self.read(), widgets)
+
+        return self
+
+
+class PdfWrapper(FormWrapper):
     """A class to represent a PDF form."""
 
     def __init__(
@@ -36,7 +69,7 @@ class PdfWrapper:
     ) -> None:
         """Constructs all attributes for the object."""
 
-        self.stream = fp_or_f_obj_or_stream_to_stream(template)
+        super().__init__(template)
         self.widgets = build_widgets(self.stream) if self.stream else {}
 
         self.global_font = kwargs.get("global_font")
@@ -49,11 +82,6 @@ class PdfWrapper:
                 each.font_size = self.global_font_size
                 each.font_color = self.global_font_color
 
-    def read(self) -> bytes:
-        """Reads the file stream of a PDF form."""
-
-        return self.stream
-
     @property
     def elements(self) -> dict:
         """ToDo: deprecate this."""

{PyPDFForm-1.4.12.dist-info → PyPDFForm-1.4.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: PyPDFForm
-Version: 1.4.12
+Version: 1.4.13
 Summary: The Python library for PDF forms.
 Home-page: https://github.com/chinapandaman/PyPDFForm
 Author: Jinge Li

{PyPDFForm-1.4.12.dist-info → PyPDFForm-1.4.13.dist-info}/RECORD

@@ -1,15 +1,15 @@
-PyPDFForm/__init__.py,sha256=_jlDwbZ1Lq0xX2yJKZgvpl_afKE3t9OsTDbc8NHSvlM,136
+PyPDFForm/__init__.py,sha256=KAGYHIP-vZnPpNmy2UPIQSULuxgByHXmJhj303FFmCg,149
 PyPDFForm/adapter.py,sha256=OgAIwcmukpBqHFBLymd3I7wA7wIdipRqgURZA2Y0Hgo,885
-PyPDFForm/constants.py,sha256=OgkM_5-MZzuNG3g5Wy7JEMcBA1Z5IIqPUQS4kAxTp4k,1643
+PyPDFForm/constants.py,sha256=qxC6FN_K5kWD5XmwypeTyGLBG778geIGf2kHnHGcXAQ,1791
 PyPDFForm/coordinate.py,sha256=EZxqeM3mkPSvhrsdOa-R2wPylmIECvUwMIlpigWRSbI,7854
-PyPDFForm/filler.py,sha256=KDANvV4wZ3kjCmRIgANvsG6lR0lL5HzmXhDm8sFLFvI,4436
+PyPDFForm/filler.py,sha256=aVmzJZFW_XwX19KLbRWuDCBZ8ek2s8BapLydf44ZbxY,6707
 PyPDFForm/font.py,sha256=iXsih_EJ9qe_penqakbQWuP44yUdxhWlQzlpZlMl3uw,4347
 PyPDFForm/image.py,sha256=0GV8PFgY5oLJFHmhPD1fG-_5SBEO7jVLLtxCc_Uodfo,1143
 PyPDFForm/patterns.py,sha256=LQRwSk7jImTYZFfN_DC38mCWFrPrXDgkYtu-yQLI7PA,2340
 PyPDFForm/template.py,sha256=bvjrrIz6RZZl2h7MIjpI1g1FGD0mcabCjtseTvVUQTA,13176
 PyPDFForm/utils.py,sha256=_gskLLNmwhfGF16gfMP00twVY6iYADYYAYWHMvKCDrg,4140
 PyPDFForm/watermark.py,sha256=clYdRqCuWEE25IL-BUHnSo4HgLkHPSOl7FUJLhbAfGg,4755
-PyPDFForm/wrapper.py,sha256=Yj56fK744JRtQ8yoJlxyilYlHoszOggZQtkm10N1EN0,9538
+PyPDFForm/wrapper.py,sha256=pXLY84cXuL4X81hvGdmcatp_GV07XXeLfGMdjOT2sx4,10232
 PyPDFForm/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 PyPDFForm/middleware/base.py,sha256=TmoO8XpjFa5NerAumrAGm_kUt2eFP3DyyF-Vx0uXloM,724
 PyPDFForm/middleware/checkbox.py,sha256=Z92awlKjceMd_ryb9Fhrtyd4fLkU84Ii6fSum60rHcA,1305
@@ -21,8 +21,8 @@ PyPDFForm/widgets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 PyPDFForm/widgets/base.py,sha256=Stq-oQt-cfwQ6nilQ4NqpO5udFSKDNb1mAwryrWggZA,2076
 PyPDFForm/widgets/checkbox.py,sha256=issx3onojq9plTHbNQwEnkbnhTo5vJKA5XSAoxRzQCg,264
 PyPDFForm/widgets/text.py,sha256=lUVlfQ-ndemSelsKNzY6nS4Kuf_H4mDPxotFMQFV-so,484
-PyPDFForm-1.4.12.dist-info/LICENSE,sha256=43awmYkI6opyTpg19me731iO1WfXZwViqb67oWtCsFY,1065
-PyPDFForm-1.4.12.dist-info/METADATA,sha256=D2kSi5b0DuWl9rDGvs8PfGNrWnMXuQ_2yXuYGzFWnag,5178
-PyPDFForm-1.4.12.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-PyPDFForm-1.4.12.dist-info/top_level.txt,sha256=GQQKuWqPUjT9YZqwK95NlAQzxjwoQrsxQ8ureM8lWOY,10
-PyPDFForm-1.4.12.dist-info/RECORD,,
+PyPDFForm-1.4.13.dist-info/LICENSE,sha256=43awmYkI6opyTpg19me731iO1WfXZwViqb67oWtCsFY,1065
+PyPDFForm-1.4.13.dist-info/METADATA,sha256=1RfinIDEtIhuU7zhUfbvhAzp479KlQ0MH2GrLy7QxuA,5178
+PyPDFForm-1.4.13.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+PyPDFForm-1.4.13.dist-info/top_level.txt,sha256=GQQKuWqPUjT9YZqwK95NlAQzxjwoQrsxQ8ureM8lWOY,10
+PyPDFForm-1.4.13.dist-info/RECORD,,

{PyPDFForm-1.4.12.dist-info → PyPDFForm-1.4.13.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any