se simplifica la obtencion de contenido y atributos de multiples elementos xml

FossilOrigin-Name: cf1d924be9a62e231090f6448b10ba23c04bbbacc8d94fc16d5ca4e0735c4bbc
This commit is contained in:
bit4bit
2021-11-06 21:40:36 +00:00
parent 3ec31c36b4
commit d5fd7db23c
3 changed files with 41 additions and 18 deletions

View File

@@ -388,7 +388,7 @@ class FachoXML:
return None
return format_(text)
def get_element_text_or_attribute(self, xpath, default=None, multiple=False):
def get_element_text_or_attribute(self, xpath, default=None, multiple=False, raise_on_fail=False):
parts = xpath.split('/')
is_attribute = parts[-1].startswith('@')
if is_attribute:
@@ -399,9 +399,13 @@ class FachoXML:
if val is None:
return default
return val
except KeyError:
except KeyError as e:
if raise_on_fail:
raise e
return default
except ValueError:
except ValueError as e:
if raise_on_fail:
raise e
return default
else:
try:
@@ -409,9 +413,25 @@ class FachoXML:
if val is None:
return default
return val
except ValueError:
except ValueError as e:
if raise_on_fail:
raise e
return default
def get_elements_text_or_attributes(self, xpaths, raise_on_fail=True):
"""
returna el contenido o attributos de un conjunto de XPATHS
si algun XPATH es una tupla se retorna el primer elemento del mismo.
"""
vals = []
for xpath in xpaths:
if isinstance(xpath, tuple):
val = xpath[0]
else:
val = self.get_element_text_or_attribute(xpath, raise_on_fail=raise_on_fail)
vals.append(val)
return vals
def exist_element(self, xpath):
elem = self.get_element(xpath)

View File

@@ -67,19 +67,21 @@ class InformacionGeneral:
def post_apply(self, fexml, fragment):
# generar cune
campos = [
fexml.get_element_attribute('/fe:NominaIndividual/NumeroSecuenciaXML', 'Numero'),
fexml.get_element_attribute('/fe:NominaIndividual/InformacionGeneral', 'FechaGen'),
fexml.get_element_attribute('/fe:NominaIndividual/InformacionGeneral', 'HoraGen'),
fexml.get_element_text('/fe:NominaIndividual/DevengadosTotal'),
fexml.get_element_text('/fe:NominaIndividual/DeduccionesTotal'),
fexml.get_element_text('/fe:NominaIndividual/ComprobanteTotal'),
fexml.get_element_attribute('/fe:NominaIndividual/Empleador', 'NIT'),
fexml.get_element_attribute('/fe:NominaIndividual/Trabajador', 'NumeroDocumento'),
fexml.get_element_attribute('/fe:NominaIndividual/InformacionGeneral', 'TipoXML'),
self.software_pin,
fexml.get_element_attribute('/fe:NominaIndividual/InformacionGeneral', 'Ambiente')
xpaths = [
'/fe:NominaIndividual/NumeroSecuenciaXML/@Numero',
'/fe:NominaIndividual/InformacionGeneral/@FechaGen',
'/fe:NominaIndividual/InformacionGeneral/@HoraGen',
'/fe:NominaIndividual/DevengadosTotal',
'/fe:NominaIndividual/DeduccionesTotal',
'/fe:NominaIndividual/ComprobanteTotal',
'/fe:NominaIndividual/Empleador/@NIT',
'/fe:NominaIndividual/Trabajador/@NumeroDocumento',
'/fe:NominaIndividual/InformacionGeneral/@TipoXML',
tuple([self.software_pin]),
'/fe:NominaIndividual/InformacionGeneral/@Ambiente'
]
campos = fexml.get_elements_text_or_attributes(xpaths)
cune = "".join(campos)
print(cune)
h = hashlib.sha384()