You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.4 KiB
72 lines
2.4 KiB
# -*- coding: utf-8 -*-
|
|
# © 2016 Danimar Ribeiro, Trustcode
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
|
|
from lxml import etree
|
|
|
|
from lxml import objectify
|
|
from jinja2 import Environment, FileSystemLoader
|
|
from . import filters
|
|
|
|
|
|
def recursively_empty(e):
|
|
if e.text:
|
|
return False
|
|
return all((recursively_empty(c) for c in e.iterchildren()))
|
|
|
|
|
|
def render_xml(path, template_name, remove_empty, **nfe):
|
|
nfe = recursively_normalize(nfe)
|
|
env = Environment(loader=FileSystemLoader(
|
|
path), extensions=["jinja2.ext.with_"])
|
|
env.filters["normalize"] = filters.strip_line_feed
|
|
env.filters["normalize_str"] = filters.normalize_str
|
|
env.filters["format_percent"] = filters.format_percent
|
|
env.filters["format_datetime"] = filters.format_datetime
|
|
env.filters["format_date"] = filters.format_date
|
|
env.filters["comma"] = filters.format_with_comma
|
|
|
|
template = env.get_template(template_name)
|
|
xml = template.render(**nfe).replace("\n", "")
|
|
parser = etree.XMLParser(
|
|
remove_blank_text=True, remove_comments=True, strip_cdata=False
|
|
)
|
|
root = etree.fromstring(xml, parser=parser)
|
|
for element in root.iter("*"): # remove espaços em branco
|
|
if element.text is not None and not element.text.strip():
|
|
element.text = None
|
|
if remove_empty:
|
|
context = etree.iterwalk(root)
|
|
for dummy, elem in context:
|
|
parent = elem.getparent()
|
|
if recursively_empty(elem):
|
|
parent.remove(elem)
|
|
return root
|
|
return root
|
|
|
|
|
|
def sanitize_response(response):
|
|
parser = etree.XMLParser(encoding="utf-8")
|
|
tree = etree.fromstring(response.encode("UTF-8"), parser=parser)
|
|
# Remove namespaces inuteis na resposta
|
|
for elem in tree.getiterator():
|
|
if not hasattr(elem.tag, "find"):
|
|
continue
|
|
i = elem.tag.find("}")
|
|
if i >= 0:
|
|
elem.tag = elem.tag[i + 1:]
|
|
objectify.deannotate(tree, cleanup_namespaces=True)
|
|
return response, objectify.fromstring(etree.tostring(tree))
|
|
|
|
|
|
def recursively_normalize(vals):
|
|
for item in vals:
|
|
if type(vals[item]) is str:
|
|
vals[item] = vals[item].strip()
|
|
vals[item] = filters.normalize_str(vals[item])
|
|
elif type(vals[item]) is dict:
|
|
recursively_normalize(vals[item])
|
|
elif type(vals[item]) is list:
|
|
for a in vals[item]:
|
|
recursively_normalize(a)
|
|
return vals
|