PyPDFForm 3.1.0__tar.gz → 3.1.2__tar.gz

{pypdfform-3.1.0 → pypdfform-3.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyPDFForm
-Version: 3.1.0
+Version: 3.1.2
 Summary: The Python library for PDF forms.
 Author: Jinge Li
 License-Expression: MIT

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm/__init__.py

@@ -20,7 +20,7 @@ The library supports various PDF form features, including:
 PyPDFForm aims to simplify PDF form manipulation, making it accessible to developers of all skill levels.
 """
 
-__version__ = "3.1.0"
+__version__ = "3.1.2"
 
 from .middleware.text import Text  # exposing for setting global font attrs
 from .wrapper import PdfWrapper

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm/constants.py

@@ -105,3 +105,5 @@ IMAGE_FIELD_IDENTIFIER = "event.target.buttonImportIcon();"
 
 COORDINATE_GRID_FONT_SIZE_MARGIN_RATIO = DEFAULT_FONT_SIZE / 100
 UNIQUE_SUFFIX_LENGTH = 20
+
+SLASH = "/"

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm/patterns.py

@@ -8,11 +8,13 @@ properties in the PDF's annotation dictionary. It also provides utility function
 for updating these widgets.
 """
 
+from typing import Union
+
 from pypdf.generic import (ArrayObject, DictionaryObject, NameObject,
                            NumberObject, TextStringObject)
 
-from .constants import (AP, AS, DV, FT, IMAGE_FIELD_IDENTIFIER, JS, TU, A, Btn,
-                        Ch, I, N, Off, Opt, Parent, Sig, T, Tx, V, Yes)
+from .constants import (AP, AS, DV, FT, IMAGE_FIELD_IDENTIFIER, JS, SLASH, TU,
+                        A, Btn, Ch, I, N, Off, Opt, Parent, Sig, T, Tx, V, Yes)
 from .middleware.checkbox import Checkbox
 from .middleware.dropdown import Dropdown
 from .middleware.image import Image
@@ -42,7 +44,7 @@ WIDGET_TYPE_PATTERNS = [
         (
             {FT: Btn},
             {Parent: {FT: Btn}},
-            {AS: (Yes, Off)},
+            {AS: (Yes, Off, SLASH)},
         ),
         Radio,
     ),
@@ -76,7 +78,7 @@ WIDGET_TYPE_PATTERNS = [
     (
         (
             {Parent: {FT: Btn}},
-            {AS: (Yes, Off)},
+            {AS: (Yes, Off, SLASH)},
         ),
         Radio,
     ),
@@ -113,6 +115,23 @@ def update_checkbox_value(annot: DictionaryObject, check: bool = False) -> None:
             break
 
 
+def get_checkbox_value(annot: DictionaryObject) -> Union[bool, None]:
+    """
+    Retrieves the boolean value of a checkbox annotation.
+
+    This function checks the value (V) of the checkbox annotation. If the value
+    is not 'Off', it means the checkbox is checked, and True is returned.
+    Otherwise, if the value is 'Off' or not present, None is returned.
+
+    Args:
+        annot (DictionaryObject): The checkbox annotation dictionary.
+
+    Returns:
+        Union[bool, None]: True if the checkbox is checked, None otherwise.
+    """
+    return True if annot.get(V, Off) != Off else None
+
+
 def update_radio_value(annot: DictionaryObject) -> None:
     """
     Updates the value of a radio button annotation, selecting it.
@@ -133,6 +152,28 @@ def update_radio_value(annot: DictionaryObject) -> None:
             break
 
 
+def get_radio_value(annot: DictionaryObject) -> bool:
+    """
+    Retrieves the boolean value of a radio button annotation.
+
+    This function iterates through the appearance states (AP) of the radio button
+    annotation. If the value (V) of the parent dictionary matches any of these
+    appearance states, it means the radio button is selected, and True is returned.
+    Otherwise, False is returned.
+
+    Args:
+        annot (DictionaryObject): The radio button annotation dictionary.
+
+    Returns:
+        bool: True if the radio button is selected, False otherwise.
+    """
+    for each in annot.get(AP, {}).get(N, []):
+        if annot.get(Parent, {}).get(V) == each:
+            return True
+
+    return False
+
+
 def update_dropdown_value(annot: DictionaryObject, widget: Dropdown) -> None:
     """
     Updates the value of a dropdown annotation, selecting an option from the list.
@@ -157,6 +198,29 @@ def update_dropdown_value(annot: DictionaryObject, widget: Dropdown) -> None:
         annot[NameObject(I)] = ArrayObject([NumberObject(widget.value)])
 
 
+def get_dropdown_value(annot: DictionaryObject, widget: Dropdown) -> None:
+    """
+    Retrieves the selected value of a dropdown annotation and updates the widget.
+
+    This function determines the current value of the dropdown, considering
+    whether it's a child annotation or a top-level one. It then iterates
+    through the widget's choices to find a match and sets the widget's
+    value to the index of the matched choice.
+
+    Args:
+        annot (DictionaryObject): The dropdown annotation dictionary.
+        widget (Dropdown): The Dropdown widget object to update with the retrieved value.
+    """
+    if Parent in annot and T not in annot:
+        to_compare = annot.get(Parent, {}).get(V)
+    else:
+        to_compare = annot.get(V)
+
+    for i, each in enumerate(widget.choices):
+        if each == to_compare:
+            widget.value = i or None  # set None when 0
+
+
 def update_text_value(annot: DictionaryObject, widget: Text) -> None:
     """
     Updates the value of a text annotation, setting the text content.
@@ -176,6 +240,24 @@ def update_text_value(annot: DictionaryObject, widget: Text) -> None:
         annot[NameObject(AP)] = TextStringObject(widget.value)
 
 
+def get_text_value(annot: DictionaryObject, widget: Text) -> None:
+    """
+    Retrieves the text value of a text annotation and updates the widget.
+
+    This function determines the current text value of the annotation, considering
+    whether it's a child annotation or a top-level one, and then sets the
+    widget's value accordingly.
+
+    Args:
+        annot (DictionaryObject): The text annotation dictionary.
+        widget (Text): The Text widget object to update with the retrieved value.
+    """
+    if Parent in annot and T not in annot:
+        widget.value = annot[Parent].get(V)
+    else:
+        widget.value = annot.get(V)
+
+
 def update_annotation_name(annot: DictionaryObject, val: str) -> None:
     """
     Updates the name of an annotation, setting the T (title) entry.

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm/template.py

@@ -16,12 +16,14 @@ from pypdf import PdfReader, PdfWriter
 from pypdf.generic import DictionaryObject
 
 from .constants import WIDGET_TYPES, Annots, MaxLen, Parent, T
+from .middleware.checkbox import Checkbox
 from .middleware.dropdown import Dropdown
 from .middleware.radio import Radio
 from .middleware.text import Text
 from .patterns import (DROPDOWN_CHOICE_PATTERNS, WIDGET_DESCRIPTION_PATTERNS,
                        WIDGET_KEY_PATTERNS, WIDGET_TYPE_PATTERNS,
-                       update_annotation_name)
+                       get_checkbox_value, get_dropdown_value, get_radio_value,
+                       get_text_value, update_annotation_name)
 from .utils import extract_widget_property, find_pattern_match, stream_to_io
 
 
@@ -61,11 +63,16 @@ def build_widgets(
                 if isinstance(_widget, Text):
                     # mostly for schema for now
                     _widget.max_length = get_text_field_max_length(widget)
+                    get_text_value(widget, _widget)
+
+                if type(_widget) is Checkbox:
+                    _widget.value = get_checkbox_value(widget)
 
                 if isinstance(_widget, Dropdown):
                     # actually used for filling value
                     # doesn't trigger hook
                     _widget.__dict__["choices"] = get_dropdown_choices(widget)
+                    get_dropdown_value(widget, _widget)
 
                 if isinstance(_widget, Radio):
                     if key not in results:
@@ -73,6 +80,9 @@ def build_widgets(
 
                     # for schema
                     results[key].number_of_options += 1
+
+                    if get_radio_value(widget):
+                        results[key].value = results[key].number_of_options - 1
                     continue
 
                 results[key] = _widget

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm/utils.py

@@ -23,7 +23,7 @@ from typing import Any, BinaryIO, List, Union
 from pypdf import PdfReader, PdfWriter
 from pypdf.generic import ArrayObject, DictionaryObject, NameObject
 
-from .constants import UNIQUE_SUFFIX_LENGTH, XFA, AcroForm, Annots, Root
+from .constants import SLASH, UNIQUE_SUFFIX_LENGTH, XFA, AcroForm, Annots, Root
 
 
 @lru_cache
@@ -220,6 +220,8 @@ def find_pattern_match(pattern: dict, widget: Union[dict, DictionaryObject]) ->
             else:
                 if isinstance(pattern[key], tuple):
                     result = value in pattern[key]
+                    if not result and SLASH in pattern[key] and value.startswith(SLASH):
+                        result = True
                 else:
                     result = pattern[key] == value
         if result:

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyPDFForm
-Version: 3.1.0
+Version: 3.1.2
 Summary: The Python library for PDF forms.
 Author: Jinge Li
 License-Expression: MIT

{pypdfform-3.1.0 → pypdfform-3.1.2}/PyPDFForm.egg-info/SOURCES.txt

@@ -39,6 +39,7 @@ PyPDFForm/widgets/text.py
 tests/test_adobe_mode.py
 tests/test_create_widget.py
 tests/test_dropdown.py
+tests/test_extract_values.py
 tests/test_fill_max_length_text_field.py
 tests/test_fill_method.py
 tests/test_functional.py

pypdfform-3.1.2/tests/test_extract_values.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+
+import os
+
+from PyPDFForm import PdfWrapper
+
+
+def test_text_check_values(pdf_samples, data_dict):
+    obj = PdfWrapper(os.path.join(pdf_samples, "test_fill.pdf"))
+
+    for k, v in data_dict.items():
+        if v is False:
+            assert obj.widgets[k].value is None
+        else:
+            assert obj.widgets[k].value == v
+
+
+def test_radio_values(pdf_samples):
+    data_dict = {
+        "radio_1": 0,
+        "radio_2": 1,
+        "radio_3": 2,
+    }
+    obj = PdfWrapper(os.path.join(pdf_samples, "test_fill_radiobutton.pdf"))
+
+    for k, v in data_dict.items():
+        assert obj.widgets[k].value == v
+
+
+def test_sejda_values(pdf_samples, sejda_data):
+    obj = PdfWrapper(os.path.join(pdf_samples, "test_fill_sejda.pdf"))
+
+    for k, v in sejda_data.items():
+        if v is False:
+            assert obj.widgets[k].value is None
+        else:
+            assert obj.widgets[k].value == v
+
+
+def test_dropdown_values(pdf_samples):
+    obj = PdfWrapper(os.path.join(pdf_samples, "dropdown", "test_dropdown_two.pdf"))
+
+    assert obj.widgets["dropdown_1"].value == 1
+
+
+def test_dropdown_default_values(pdf_samples):
+    obj = PdfWrapper(os.path.join(pdf_samples, "dropdown", "test_dropdown_one.pdf"))
+
+    assert obj.widgets["dropdown_1"].value is None
+
+
+def test_sejda_dropdown_values(pdf_samples):
+    obj = PdfWrapper(
+        os.path.join(pdf_samples, "dropdown", "test_dropdown_alignment_sejda.pdf")
+    )
+
+    assert obj.widgets["dropdown_left"].value is None
+    assert obj.widgets["dropdown_center"].value == 1
+    assert obj.widgets["dropdown_right"].value == 2
+
+
+def test_addition_operator_3_times_values(template_stream, data_dict):
+    result = PdfWrapper()
+
+    for _ in range(3):
+        result += PdfWrapper(template_stream).fill(data_dict)
+
+    obj = PdfWrapper(result.read())
+
+    for k, v in obj.widgets.items():
+        if k.split("-")[0] in data_dict:
+            assert (v.value or False) == data_dict[k.split("-")[0]]