1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
from fpdf.structure_tree import PDFObject, StructureTreeBuilder
def test_pdf_object_serialize():
class Point(PDFObject):
__slots__ = ("_id", "x", "y")
def __init__(self, x=0, y=0):
super().__init__()
self.x = x
self.y = y
class Square(PDFObject):
__slots__ = ("_id", "top_left", "bottom_right")
def __init__(self, top_left, bottom_right):
super().__init__()
self.top_left = top_left
self.bottom_right = bottom_right
point_a = Point()
point_b = Point(x=10, y=10)
square = Square(top_left=point_a, bottom_right=point_b)
point_a.id = 1
point_b.id = 2
square.id = 3
pdf_content = (
point_a.serialize() + "\n" + point_b.serialize() + "\n" + square.serialize()
)
assert (
pdf_content
== """\
1 0 obj
<<
/X 0
/Y 0
>>
endobj
2 0 obj
<<
/X 10
/Y 10
>>
endobj
3 0 obj
<<
/BottomRight 2 0 R
/TopLeft 1 0 R
>>
endobj"""
)
def _serialize(struct_builder, first_object_id=1):
n = first_object_id
for obj in struct_builder:
obj.id = n
n += 1
return "\n".join(obj.serialize() for obj in struct_builder)
def test_empty_structure_tree():
struct_builder = StructureTreeBuilder()
assert (
_serialize(struct_builder)
== """\
1 0 obj
<<
/K [2 0 R]
/ParentTree 3 0 R
/Type /StructTreeRoot
>>
endobj
2 0 obj
<<
/K []
/P 1 0 R
/S /Document
/Type /StructElem
>>
endobj
3 0 obj
<<
/Nums []
>>
endobj"""
)
def test_single_image_structure_tree():
struct_builder = StructureTreeBuilder()
struct_builder.add_marked_content(
1, "/Figure", 0, "Image title", "Image description"
)
assert (
_serialize(struct_builder, first_object_id=3)
== """\
3 0 obj
<<
/K [4 0 R]
/ParentTree 5 0 R
/Type /StructTreeRoot
>>
endobj
4 0 obj
<<
/K [6 0 R]
/P 3 0 R
/S /Document
/Type /StructElem
>>
endobj
5 0 obj
<<
/Nums [0 [6 0 R]]
>>
endobj
6 0 obj
<<
/Alt (Image description)
/K [0]
/P 4 0 R
/S /Figure
/T (Image title)
/Type /StructElem
>>
endobj"""
)
|
