[REF] Combined pattern analisar_retorno_raw

Signed-off-by: Luis Felipe Miléo <[email protected]>
erpbrasil · Jul 16, 2024 · e63a588 · e63a588
1 parent 69f4c88
commit e63a588
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/erpbrasil/edoc/resposta.py b/src/erpbrasil/edoc/resposta.py
@@ -6,6 +6,10 @@
 
 from lxml import etree
 
+combined_pattern = "|".join(
+    ["<soap:Body>(.*?)</soap:Body>", "<[a-zA-Z0-9:]Body.?>(.*?)</[a-zA-Z0-9:]*Body>"]
+)
+
 
 class RetornoSoap:
     def __init__(self, webservice, raiz, xml, retorno, resposta):
@@ -18,7 +22,7 @@ def __init__(self, webservice, raiz, xml, retorno, resposta):
 
 def analisar_retorno_raw(operacao, raiz, xml, retorno, classe):
     retorno.raise_for_status()
-    pattern = r"<[a-zA-Z0-9:]Body.?>(.*?)</[a-zA-Z0-9:]*Body>"
+    pattern = re.compile(combined_pattern)
     match = re.search(pattern, retorno.text.replace("\n", ""))
     if match:
         xml_resposta = match.group(1)