Remoção das funções xml_download_nfe e download_nfe por serem iguais as xml_consulta_distribuicao_nfe e consulta_distribuicao_nfe;

Alteração na ordenação das funções para seguir o padrão inserido por padrão;
Foi alterado o arquivo de validar esquema para trabalhar com classes para facilitar a utilização e o reaproveitamento de codigo;
Foi alterado a funções de consulta_distribuicao_nfe e autorizar_nfe para validar o esquema do xml antes de enviar e caso tenha algum erro, interromper o envio para evitar bloqueio conforme Nota Técnica 2018.002.

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)

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