|
12 | 12 |
|
13 | 13 | from docx import Document
|
14 | 14 | from docx.dml.color import ColorFormat
|
15 |
| -from docx.enum.dml import MSO_COLOR_TYPE |
| 15 | +from docx.enum.dml import MSO_COLOR_TYPE, MSO_THEME_COLOR |
16 | 16 | from docx.enum.text import WD_UNDERLINE
|
17 | 17 | from docx.shared import RGBColor
|
18 | 18 |
|
@@ -87,6 +87,13 @@ def when_I_assign_value_to_font_color_rgb(context, value):
|
87 | 87 | font.color.rgb = new_value
|
88 | 88 |
|
89 | 89 |
|
| 90 | +@when('I assign {value} to font.color.theme_color') |
| 91 | +def when_I_assign_value_to_font_color_theme_color(context, value): |
| 92 | + font = context.font |
| 93 | + new_value = None if value == 'None' else getattr(MSO_THEME_COLOR, value) |
| 94 | + font.color.theme_color = new_value |
| 95 | + |
| 96 | + |
90 | 97 | @when('I assign {value} to font.name')
|
91 | 98 | def when_I_assign_value_to_font_name(context, value):
|
92 | 99 | font = context.font
|
@@ -145,6 +152,15 @@ def then_font_color_rgb_is_value(context, value):
|
145 | 152 | assert font.color.rgb == expected_value
|
146 | 153 |
|
147 | 154 |
|
| 155 | +@then('font.color.theme_color is {value}') |
| 156 | +def then_font_color_theme_color_is_value(context, value): |
| 157 | + font = context.font |
| 158 | + expected_value = ( |
| 159 | + None if value == 'None' else getattr(MSO_THEME_COLOR, value) |
| 160 | + ) |
| 161 | + assert font.color.theme_color == expected_value |
| 162 | + |
| 163 | + |
148 | 164 | @then('font.color.type is {value}')
|
149 | 165 | def then_font_color_type_is_value(context, value):
|
150 | 166 | font = context.font
|
|
0 commit comments