Skip to content

Commit e5ec61c

Browse files
author
black-isort-bot
committed
linting: black/isort
1 parent 00cf349 commit e5ec61c

File tree

6 files changed

+22
-26
lines changed

6 files changed

+22
-26
lines changed

PyPDFForm/core/template.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,7 @@ class Template:
1616
"""Contains methods for interacting with a pdfrw parsed PDF form."""
1717

1818
@staticmethod
19-
def remove_all_elements(
20-
pdf: bytes
21-
) -> bytes:
19+
def remove_all_elements(pdf: bytes) -> bytes:
2220
"""Removes all elements from a pdfrw parsed PDF form."""
2321

2422
pdf = pdfrw.PdfReader(fdata=pdf)

PyPDFForm/middleware/wrapper.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,14 @@ def __add__(self, other: "PyPDFForm") -> "PyPDFForm":
8787
TemplateMiddleware().validate_stream(self.stream)
8888
TemplateMiddleware().validate_stream(other.stream)
8989

90-
pdf_one = TemplateCore().assign_uuid(self.stream) if not self.sejda else self.stream
91-
pdf_two = TemplateCore().assign_uuid(other.stream) if not other.sejda else other.stream
90+
pdf_one = (
91+
TemplateCore().assign_uuid(self.stream) if not self.sejda else self.stream
92+
)
93+
pdf_two = (
94+
TemplateCore().assign_uuid(other.stream)
95+
if not other.sejda
96+
else other.stream
97+
)
9298

9399
new_obj = self.__class__()
94100
new_obj.stream = UtilsCore().merge_two_pdfs(pdf_one, pdf_two)

tests/conftest.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,7 @@ def pdf_samples():
1212

1313
@pytest.fixture
1414
def sejda_template(pdf_samples):
15-
with open(
16-
os.path.join(pdf_samples, "sample_template_sejda.pdf"), "rb+"
17-
) as f:
15+
with open(os.path.join(pdf_samples, "sample_template_sejda.pdf"), "rb+") as f:
1816
return f.read()
1917

2018

tests/unit/test_filler.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,9 @@ def test_fill_sejda(sejda_template, sejda_data):
101101
assert result_stream != template_stream
102102

103103
for element in TemplateCore().iterate_elements(result_stream):
104-
assert element[TemplateConstants().parent_key][TemplateConstants().field_editable_key] == pdfrw.PdfObject(1)
104+
assert element[TemplateConstants().parent_key][
105+
TemplateConstants().field_editable_key
106+
] == pdfrw.PdfObject(1)
105107

106108

107109
def test_fill_with_radiobutton(template_with_radiobutton_stream, data_dict):

tests/unit/test_template.py

+6-16
Original file line numberDiff line numberDiff line change
@@ -79,13 +79,8 @@ def test_iterate_elements_and_get_element_key(
7979
assert data_dict[k]
8080

8181

82-
def test_iterate_elements_and_get_element_key_sejda(
83-
sejda_template, sejda_data
84-
):
85-
data_dict = {
86-
key: False
87-
for key in sejda_data.keys()
88-
}
82+
def test_iterate_elements_and_get_element_key_sejda(sejda_template, sejda_data):
83+
data_dict = {key: False for key in sejda_data.keys()}
8984
for each in TemplateCore().iterate_elements(sejda_template, sejda=True):
9085
data_dict[TemplateCore().get_element_key(each, sejda=True)] = True
9186

@@ -273,14 +268,9 @@ def test_build_elements(template_with_radiobutton_stream, data_dict):
273268

274269

275270
def test_build_elements_sejda(sejda_template, sejda_data):
276-
data_dict = {
277-
key: False
278-
for key in sejda_data.keys()
279-
}
271+
data_dict = {key: False for key in sejda_data.keys()}
280272

281-
for k, v in (
282-
TemplateMiddleware().build_elements(sejda_template, sejda=True).items()
283-
):
273+
for k, v in TemplateMiddleware().build_elements(sejda_template, sejda=True).items():
284274
if k in data_dict and k == v.name:
285275
data_dict[k] = True
286276

@@ -305,8 +295,8 @@ def test_get_draw_checkbox_radio_coordinates(sejda_template):
305295
for element in TemplateCore().iterate_elements(sejda_template):
306296
assert TemplateCore().get_draw_checkbox_radio_coordinates(element) == (
307297
(
308-
float(element[TemplateCoreConstants().annotation_rectangle_key][0])
309-
+ float(element[TemplateCoreConstants().annotation_rectangle_key][2])
298+
float(element[TemplateCoreConstants().annotation_rectangle_key][0])
299+
+ float(element[TemplateCoreConstants().annotation_rectangle_key][2])
310300
)
311301
/ 2
312302
- 5,

tests/unit/test_utils.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,9 @@ def checkbox_radio_to_draw():
7878
assert new_checkbox_to_draw.font_color == TextConstants().global_font_color
7979
assert new_checkbox_to_draw.text_x_offset == TextConstants().global_text_x_offset
8080
assert new_checkbox_to_draw.text_y_offset == TextConstants().global_text_y_offset
81-
assert new_checkbox_to_draw.text_wrap_length == TextConstants().global_text_wrap_length
81+
assert (
82+
new_checkbox_to_draw.text_wrap_length == TextConstants().global_text_wrap_length
83+
)
8284

8385
assert new_radio_to_draw.value == u"\u25CF"
8486
assert new_radio_to_draw.type == ElementType.text

0 commit comments

Comments
 (0)