Igor20264
35 lines
861 B
Python
35 lines
861 B
Python
from docx.oxml import OxmlElement
|
|
from docx.oxml.ns import qn
|
|
from lxml.etree import _Element
|
|
|
|
|
|
def create_element(name: str, *args: dict[str, str] | list[_Element] | str)\
|
|
-> _Element:
|
|
"""Creates an OxmlElement
|
|
|
|
Variable arguments:
|
|
* dict -- element's attributes
|
|
* list -- element's children
|
|
* string -- element's text
|
|
"""
|
|
attrs = {}
|
|
children = []
|
|
text = None
|
|
|
|
for arg in args:
|
|
if isinstance(arg, dict):
|
|
attrs.update(arg)
|
|
elif isinstance(arg, list):
|
|
children.extend(arg)
|
|
elif isinstance(arg, str):
|
|
text = arg
|
|
|
|
element = OxmlElement(name, {
|
|
(qn(name) if ":" in name else name): value for name, value in attrs.items()
|
|
})
|
|
for child in children:
|
|
element.append(child)
|
|
if text:
|
|
element.text = text
|
|
return element
|