diff --git a/pytrustnfe/nfe/__init__.py b/pytrustnfe/nfe/__init__.py index 8395f24..d1684b6 100644 --- a/pytrustnfe/nfe/__init__.py +++ b/pytrustnfe/nfe/__init__.py @@ -9,6 +9,7 @@ from lxml import etree from .patch import has_patch from .assinatura import Assinatura from pytrustnfe.xml import render_xml, sanitize_response +from pytrustnfe.xml.validate import ValidarXml from pytrustnfe.utils import gerar_chave, ChaveNFe from pytrustnfe.Servidores import localizar_url from pytrustnfe.certificado import extract_cert_and_key_from_pfx, save_cert_key @@ -20,6 +21,9 @@ from requests import Session from zeep import Client from zeep.transports import Transport +# Xml schema validation class +validar_xml = ValidarXml() + def _generate_nfe_id(**kwargs): for item in kwargs['NFes']: @@ -125,7 +129,11 @@ def xml_autorizar_nfe(certificado, **kwargs): def autorizar_nfe(certificado, **kwargs): # Assinar if "xml" not in kwargs: kwargs['xml'] = xml_autorizar_nfe(certificado, **kwargs) - return _send(certificado, 'NfeAutorizacao', **kwargs) + erros_esquemas = validar_xml.valida_nfe(kwargs['xml']) + if erros_esquemas is not False: + return erros_esquemas + else: + return _send(certificado, 'NfeAutorizacao', **kwargs) def xml_retorno_autorizar_nfe(certificado, **kwargs): @@ -214,6 +222,16 @@ def xml_consulta_distribuicao_nfe(certificado, **kwargs): # Assinar return _render(certificado, 'NFeDistribuicaoDFe', False, **kwargs) +def consulta_distribuicao_nfe(certificado, **kwargs): + if "xml" not in kwargs: + kwargs['xml'] = xml_consulta_distribuicao_nfe(certificado, **kwargs) + erros_esquemas = validar_xml.valida_distribuicao(kwargs['xml']) + if erros_esquemas is not False: + return erros_esquemas + else: + return _send_v310(certificado, **kwargs) + + def _send_v310(certificado, **kwargs): xml_send = kwargs["xml"] base_url = localizar_url( @@ -243,19 +261,3 @@ def _send_v310(certificado, **kwargs): 'received_xml': response, 'object': obj.Body.nfeDistDFeInteresseResponse.nfeDistDFeInteresseResult } - - -def consulta_distribuicao_nfe(certificado, **kwargs): - if "xml" not in kwargs: - kwargs['xml'] = xml_consulta_distribuicao_nfe(certificado, **kwargs) - return _send_v310(certificado, **kwargs) - - -def xml_download_nfe(certificado, **kwargs): # Assinar - return _render(certificado, 'NFeDistribuicaoDFe', False, **kwargs) - - -def download_nfe(certificado, **kwargs): - if "xml" not in kwargs: - kwargs['xml'] = xml_download_nfe(certificado, **kwargs) - return _send_v310(certificado, **kwargs) diff --git a/pytrustnfe/xml/validate.py b/pytrustnfe/xml/validate.py index 4564f1a..6ff45b3 100644 --- a/pytrustnfe/xml/validate.py +++ b/pytrustnfe/xml/validate.py @@ -6,23 +6,30 @@ import os from lxml import etree -PATH = os.path.dirname(os.path.abspath(__file__)) - -def valida_nfe(xml_nfe): - xsd = 'schemas/enviNFe_v4.00.xsd' - return valida_esquema(xml_nfe, xsd) - - -def valida_distribuicao(xml): - xsd = 'distDFeInt_v1.01.xsd' - return valida_esquema(xml, xsd) - - -def valida_esquema(xml, xsd_name): - xsd = os.path.join(PATH, xsd_name) - xml_etree = etree.fromstring(xml) - esquema = etree.XMLSchema(etree.parse(xsd)) - esquema.validate(xml_etree) - erros = [x.message for x in esquema.error_log] - return "\n".join(erros) +class ValidarXml: + PATH = os.path.dirname(os.path.abspath(__file__)) + + def valida_nfe(self, xml): + xsd = 'enviNFe_v4.00.xsd' + erros = self.valida_esquema(xml, xsd) + if len(erros) > 0: + return {'ErrosEsquemas': "\n".join(erros)} + else: + return False + + def valida_distribuicao(self, xml): + xsd = 'distDFeInt_v1.01.xsd' + erros = self.valida_esquema(xml, xsd) + if len(erros) >0: + return {'ErrosEsquemas': "\n".join(erros)} + else: + return False + + def valida_esquema(self, xml, xsd_name): + xsd = os.path.join(self.PATH, 'schemas/', xsd_name) + xml_etree = etree.fromstring(xml) + esquema = etree.XMLSchema(etree.parse(xsd)) + esquema.validate(xml_etree) + erros = [x.message for x in esquema.error_log] + return erros