NPO-Accounting/conservancy_beancount Changeset - 391fde5447de

Changeset - 391fde5447de

Parent rev.

Child rev.

[Not reviewed]

0 2 0

Brett Smith - 3 years ago 2021-01-09 15:09:08
brettcsmith@brettcsmith.org

fields: TextField.set_value() accepts numeric types.

2 files changed with 28 insertions and 1 deletions:

conservancy_beancount/pdfforms/fields.py

tests/test_pdfforms_fields.py

0 comments (0 inline, 0 general)

conservancy_beancount/pdfforms/fields.py

➞

Show inline comments

@@ ... / @@ -5,12 +5,16 @@ @@
 # Full copyright and licensing details can be found at toplevel file
 # LICENSE.txt in the repository.
 import enum
 import functools
 from decimal import Decimal
 import babel.numbers  # type:ignore[import]
 from pdfminer.pdftypes import resolve1  # type:ignore[import]
 from pdfminer import psparser  # type:ignore[import]
 from . import utils as pdfutils
 from .errors import PDFKeyError, PDFSpecError
 from typing import (
@@ ... / @@ -19,12 +23,13 @@ from typing import ( @@
     Optional,
     Mapping,
     MutableMapping,
     Sequence,
     Tuple,
     Union,
     cast,
+)
 FieldSource = MutableMapping[str, Any]
 class FieldFlags(enum.IntFlag):
     # Flags for all fields
@@ ... / @@ -235,11 +240,13 @@ class TextField(FormField): @@
         else:
             raise PDFSpecError("text field value is not bytes")
     def value(self) -> Optional[str]:
         return self._decode(super().value())
     def set_value(self, value: Optional[str]) -> None:
     def set_value(self, value: Union[str, Decimal, float, int, None]) -> None:
         if isinstance(value, (Decimal, float, int)):
             value = cast(str, babel.numbers.format_decimal(value))
         super().set_value(None if value is None else pdfutils.encode_text(value))
     def fill_value(self) -> Optional[str]:
         return self._decode(super().fill_value())

tests/test_pdfforms_fields.py

➞

Show inline comments

@@ ... / @@ -7,12 +7,14 @@ @@
 import codecs
 import itertools
 import pytest
 from decimal import Decimal
 from pdfminer.psparser import PSLiteral
 from conservancy_beancount.pdfforms import fields as fieldsmod
 def field_source(
         name=None,
@@ ... / @@ -246,12 +248,30 @@ def test_text_set_value(text, bprefix): @@
     field = fieldsmod.TextField(source)
     field.set_value(text)
     assert field.value() == text
     actual = fieldsmod.FormField.value(field)
     assert actual == bprefix + text.encode('utf-16be' if bprefix else 'ascii')
 @pytest.mark.parametrize('expected', [
     '0',
     '32',
     '32.45',
     '32,768',
     '32,768.95',
 ])
 def test_text_set_value_numeric(expected):
     num_s = expected.replace(',', '')
     field = fieldsmod.TextField({})
     num_types = [Decimal, float]
     if '.' not in expected:
         num_types.append(int)
     for num_type in num_types:
         field.set_value(num_type(num_s))
         assert field.value() == expected
         field.set_value(None)
 def test_text_set_value_none():
     source = field_source('t', b'set None test', 'Tx')
     field = fieldsmod.TextField(source)
     field.set_value(None)
     assert fieldsmod.FormField.value(field) is None

0 comments (0 inline, 0 general)