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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
# encoding: utf-8
"""
Step implementations for paragraph-related features
"""
from behave import given, then, when
from docx import Document
from docx.enum.text import WD_ALIGN_PARAGRAPH
from docx.text.parfmt import ParagraphFormat
from helpers import saved_docx_path, test_docx, test_text
# given ===================================================
@given('a document containing three paragraphs')
def given_a_document_containing_three_paragraphs(context):
document = Document()
document.add_paragraph('foo')
document.add_paragraph('bar')
document.add_paragraph('baz')
context.document = document
@given('a paragraph having {align_type} alignment')
def given_a_paragraph_align_type_alignment(context, align_type):
paragraph_idx = {
'inherited': 0,
'left': 1,
'center': 2,
'right': 3,
'justified': 4,
}[align_type]
document = Document(test_docx('par-alignment'))
context.paragraph = document.paragraphs[paragraph_idx]
@given('a paragraph having {style_state} style')
def given_a_paragraph_having_style(context, style_state):
paragraph_idx = {
'no specified': 0,
'a missing': 1,
'Heading 1': 2,
'Body Text': 3,
}[style_state]
document = context.document = Document(test_docx('par-known-styles'))
context.paragraph = document.paragraphs[paragraph_idx]
@given('a paragraph with content and formatting')
def given_a_paragraph_with_content_and_formatting(context):
document = Document(test_docx('par-known-paragraphs'))
context.paragraph = document.paragraphs[0]
# when ====================================================
@when('I add a run to the paragraph')
def when_add_new_run_to_paragraph(context):
context.run = context.p.add_run()
@when('I assign a {style_type} to paragraph.style')
def when_I_assign_a_style_type_to_paragraph_style(context, style_type):
paragraph = context.paragraph
style = context.style = context.document.styles['Heading 1']
style_spec = {
'style object': style,
'style name': 'Heading 1',
}[style_type]
paragraph.style = style_spec
@when('I clear the paragraph content')
def when_I_clear_the_paragraph_content(context):
context.paragraph.clear()
@when('I insert a paragraph above the second paragraph')
def when_I_insert_a_paragraph_above_the_second_paragraph(context):
paragraph = context.document.paragraphs[1]
paragraph.insert_paragraph_before('foobar', 'Heading1')
@when('I set the paragraph text')
def when_I_set_the_paragraph_text(context):
context.paragraph.text = 'bar\tfoo\r'
# then =====================================================
@then('paragraph.paragraph_format is its ParagraphFormat object')
def then_paragraph_paragraph_format_is_its_parfmt_object(context):
paragraph = context.paragraph
paragraph_format = paragraph.paragraph_format
assert isinstance(paragraph_format, ParagraphFormat)
assert paragraph_format.element is paragraph._element
@then('paragraph.style is {value_key}')
def then_paragraph_style_is_value(context, value_key):
styles = context.document.styles
expected_value = {
'Normal': styles['Normal'],
'Heading 1': styles['Heading 1'],
'Body Text': styles['Body Text'],
}[value_key]
paragraph = context.paragraph
assert paragraph.style == expected_value
@then('the document contains four paragraphs')
def then_the_document_contains_four_paragraphs(context):
assert len(context.document.paragraphs) == 4
@then('the document contains the text I added')
def then_document_contains_text_I_added(context):
document = Document(saved_docx_path)
paragraphs = document.paragraphs
p = paragraphs[-1]
r = p.runs[0]
assert r.text == test_text
@then('the paragraph alignment property value is {align_value}')
def then_the_paragraph_alignment_prop_value_is_value(context, align_value):
expected_value = {
'None': None,
'WD_ALIGN_PARAGRAPH.LEFT': WD_ALIGN_PARAGRAPH.LEFT,
'WD_ALIGN_PARAGRAPH.CENTER': WD_ALIGN_PARAGRAPH.CENTER,
'WD_ALIGN_PARAGRAPH.RIGHT': WD_ALIGN_PARAGRAPH.RIGHT,
}[align_value]
assert context.paragraph.alignment == expected_value
@then('the paragraph formatting is preserved')
def then_the_paragraph_formatting_is_preserved(context):
paragraph = context.paragraph
assert paragraph.style.name == 'Heading 1'
@then('the paragraph has no content')
def then_the_paragraph_has_no_content(context):
assert context.paragraph.text == ''
@then('the paragraph has the style I set')
def then_the_paragraph_has_the_style_I_set(context):
paragraph, expected_style = context.paragraph, context.style
assert paragraph.style == expected_style
@then('the paragraph has the text I set')
def then_the_paragraph_has_the_text_I_set(context):
assert context.paragraph.text == 'bar\tfoo\n'
@then('the style of the second paragraph matches the style I set')
def then_the_style_of_the_second_paragraph_matches_the_style_I_set(context):
second_paragraph = context.document.paragraphs[1]
assert second_paragraph.style.name == 'Heading 1'
@then('the text of the second paragraph matches the text I set')
def then_the_text_of_the_second_paragraph_matches_the_text_I_set(context):
second_paragraph = context.document.paragraphs[1]
assert second_paragraph.text == 'foobar'
|
