Remoção da classe de validação do esquema e deixando somente como função

pytrustnfe/nfe/__init__.py

@@ -9,7 +9,6 @@ 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
@@ -21,9 +20,6 @@ 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']:
@@ -129,10 +125,6 @@ def xml_autorizar_nfe(certificado, **kwargs):
 def autorizar_nfe(certificado, **kwargs):  # Assinar
     if "xml" not in kwargs:
         kwargs['xml'] = xml_autorizar_nfe(certificado, **kwargs)
-    erros_esquemas = validar_xml.valida_nfe(kwargs['xml'])
-    if erros_esquemas is not False:
-        return erros_esquemas
-    else:
     return _send(certificado, 'NfeAutorizacao', **kwargs)
 
 
@@ -225,10 +217,6 @@ def xml_consulta_distribuicao_nfe(certificado, **kwargs):  # Assinar
 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)
 
 

pytrustnfe/xml/validate.py

@@ -6,28 +6,29 @@ import os
 
 from lxml import etree
 
-
-class ValidarXml:
 PATH = os.path.dirname(os.path.abspath(__file__))
 
-    def valida_nfe(self, xml):
+
+def valida_nfe(xml):
     xsd = 'enviNFe_v4.00.xsd'
-        erros = self.valida_esquema(xml, xsd)
+    erros = valida_esquema(xml, xsd)
     if len(erros) > 0:
         return {'ErrosEsquemas': "\n".join(erros)}
     else:
         return False
 
-    def valida_distribuicao(self, xml):
+
+def valida_distribuicao(xml):
     xsd = 'distDFeInt_v1.01.xsd'
-        erros = self.valida_esquema(xml, xsd)
+    erros = 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)
+
+def valida_esquema(xml, xsd_name):
+    xsd = os.path.join(PATH, 'schemas/', xsd_name)
     xml_etree = etree.fromstring(xml)
     esquema = etree.XMLSchema(etree.parse(xsd))
     esquema.validate(xml_etree)