From c7beb73a6bf52a690eb0ec06cd52d2ac6ef26aad Mon Sep 17 00:00:00 2001
From: Gabriela Cavalcante da Silva <gabicavalcantesilva@gmail.com>
Date: Wed, 29 Jan 2020 10:44:37 -0300
Subject: [PATCH] add natal nfse

---
 pytrustnfe/nfse/natal/__init__.py                  |  77 ++++++++++++++++
 pytrustnfe/nfse/natal/templates/Rps.xml            | 101 +++++++++++++++++++++
 pytrustnfe/nfse/natal/templates/cancelarNfse.xml   |  15 +++
 .../nfse/natal/templates/consultarLoteRps.xml      |   7 ++
 .../nfse/natal/templates/recepcionarLoteRps.xml    |  13 +++
 5 files changed, 213 insertions(+)
 create mode 100644 pytrustnfe/nfse/natal/__init__.py
 create mode 100644 pytrustnfe/nfse/natal/templates/Rps.xml
 create mode 100644 pytrustnfe/nfse/natal/templates/cancelarNfse.xml
 create mode 100644 pytrustnfe/nfse/natal/templates/consultarLoteRps.xml
 create mode 100644 pytrustnfe/nfse/natal/templates/recepcionarLoteRps.xml

diff --git a/pytrustnfe/nfse/natal/__init__.py b/pytrustnfe/nfse/natal/__init__.py
new file mode 100644
index 0000000..a96948d
--- /dev/null
+++ b/pytrustnfe/nfse/natal/__init__.py
@@ -0,0 +1,77 @@
+# © 2019 Danimar Ribeiro, Trustcode
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+
+import os
+from requests import Session
+from zeep import Client
+from zeep.transports import Transport
+from requests.packages.urllib3 import disable_warnings
+
+from pytrustnfe.xml import render_xml, sanitize_response
+from pytrustnfe.certificado import extract_cert_and_key_from_pfx, save_cert_key
+from pytrustnfe.nfe.assinatura import Assinatura
+
+
+def _render(certificado, method, **kwargs):
+    path = os.path.join(os.path.dirname(__file__), "templates")
+    xml_send = render_xml(path, "%s.xml" % method, True, **kwargs)
+
+    reference = ""
+    signer = Assinatura(certificado.pfx, certificado.password)
+    xml_send = signer.assina_xml(xml_send, reference)
+    return xml_send
+
+
+def _send(certificado, method, **kwargs):
+    base_url = ""
+    if kwargs["ambiente"] == "producao":
+        base_url = "https://aparecida.siltecnologia.com.br/tbw/services/Abrasf10?wsdl"
+    else:
+        base_url = "https://aparecida.siltecnologia.com.br/tbwhomologacao/services/Abrasf10?wsdl"
+
+    cert, key = extract_cert_and_key_from_pfx(certificado.pfx, certificado.password)
+    cert, key = save_cert_key(cert, key)
+
+    disable_warnings()
+    session = Session()
+    session.cert = (cert, key)
+    session.verify = False
+    transport = Transport(session=session)
+
+    client = Client(base_url, transport=transport)
+
+    xml_send = kwargs["xml"]
+    response = client.service[method](xml_send)
+    response, obj = sanitize_response(response)
+    return {"sent_xml": xml_send, "received_xml": response, "object": obj}
+
+
+def xml_recepcionar_lote_rps(certificado, **kwargs):
+    return _render(certificado, "recepcionarLoteRps", **kwargs)
+
+
+def recepcionar_lote_rps(certificado, **kwargs):
+    if "xml" not in kwargs:
+        kwargs["xml"] = xml_recepcionar_lote_rps(certificado, **kwargs)
+    return {"sent_xml": kwargs["xml"]}
+    # return _send(certificado, 'recepcionarLoteRps', **kwargs)
+
+
+def xml_consultar_lote_rps(certificado, **kwargs):
+    return _render(certificado, "consultarLoteRps", **kwargs)
+
+
+def consultar_lote_rps(certificado, **kwargs):
+    if "xml" not in kwargs:
+        kwargs["xml"] = xml_consultar_lote_rps(certificado, **kwargs)
+    return _send(certificado, "consultarLoteRps", **kwargs)
+
+
+def xml_cancelar_nfse(certificado, **kwargs):
+    return _render(certificado, "cancelarNfse", **kwargs)
+
+
+def cancelar_nfse(certificado, **kwargs):
+    if "xml" not in kwargs:
+        kwargs["xml"] = xml_cancelar_nfse(certificado, **kwargs)
+    return _send(certificado, "cancelarNfse", **kwargs)
diff --git a/pytrustnfe/nfse/natal/templates/Rps.xml b/pytrustnfe/nfse/natal/templates/Rps.xml
new file mode 100644
index 0000000..9dcd23c
--- /dev/null
+++ b/pytrustnfe/nfse/natal/templates/Rps.xml
@@ -0,0 +1,101 @@
+<Rps>
+    <InfRps Id="{{ rps.numero }}">
+        <IdentificacaoRps>
+            <Numero>{{ rps.numero }}</Numero>
+            <Serie>{{ rps.serie }}</Serie>
+            <Tipo>{{ rps.tipo_rps }}</Tipo>
+        </IdentificacaoRps>
+        <DataEmissao>{{ rps.data_emissao }}</DataEmissao>
+        <NaturezaOperacao>{{ rps.natureza_operacao }}</NaturezaOperacao>
+        <RegimeEspecialTributacao>{{ rps.regime_tributacao }}</RegimeEspecialTributacao>
+        <OptanteSimplesNacional>{{ rps.optante_simples }}</OptanteSimplesNacional>
+        <IncentivadorCultural>{{ rps.incentivador_cultural }}</IncentivadorCultural>
+        <Servico>
+            <Valores>
+                <ValorServicos>{{ rps.servico.valor_servico }}</ValorServicos> 
+                <IssRetido>{{ rps.servico.iss_retido }}</IssRetido>
+                <BaseCalculo>{{ rps.servico.base_calculo }}</BaseCalculo>
+            </Valores>
+            <ItemListaServico>{{ rps.servico.codigo_servico }}</ItemListaServico>
+            <CodigoCnae>{{ rps.servico.cnae_servico }}</CodigoCnae>
+            <CodigoTributacaoMunicipio>{{ rps.servico.codigo_tributacao_municipio }}</CodigoTributacaoMunicipio>
+            <Discriminacao>{{ rps.servico.descricao }}</Discriminacao>
+            <CodigoMunicipio>{{ rps.servico.codigo_municipio }}</CodigoMunicipio>
+        </Servico>
+        <PrestadorServico>
+            <IdentificacaoPrestador>
+               <Cnpj>{{ rps.prestador.cnpj }}</Cnpj>
+               <InscricaoMunicipal>{{ rps.prestador.inscricao_municipal }}</InscricaoMunicipal>
+            </IdentificacaoPrestador>
+            <RazaoSocial>{{ rps.prestador.razaosocial }}</RazaoSocial>
+            <NomeFantasia>{{ rps.prestador.fantasia }}</NomeFantasia>
+            <Endereco>
+               <Endereco>{{ rps.prestador.endereco }}</Endereco>
+               <Numero>{{ rps.prestador.numero }}</Numero>
+               <Complemento>{{ rps.prestador.complemento }}</Complemento>
+               <Bairro>{{ rps.prestador.bairro }}</Bairro>
+               <CodigoMunicipio>{{ rps.prestador.codigomunicipal }}</CodigoMunicipio>
+               <Uf>{{ rps.prestador.uf }}</Uf>
+               <Cep>{{ rps.prestador.cep }}</Cep>
+            </Endereco>
+            <Contato>
+               <Telefone>{{ rps.prestador.telefone }}</Telefone>
+               <Email>{{ rps.prestador.email }}</Email>
+            </Contato>
+        </PrestadorServico>
+        <TomadorServico>
+            <IdentificacaoTomador>
+                <CpfCnpj>
+                    {% if rps.tomador.cnpj_cpf|length == 14  %}
+                    <Cnpj>{{ rps.tomador.cnpj_cpf }}</Cnpj>
+                    {% endif %}
+                    {% if rps.tomador.cnpj_cpf|length == 11  %}
+                    <Cpf>{{ rps.tomador.cnpj_cpf }}</Cpf>
+                    {% endif %}
+                </CpfCnpj>
+                {% if rps.tomador.inscricao_municipal is defined -%}
+                <InscricaoMunicipal>{{ rps.tomador.inscricao_municipal }}</InscricaoMunicipal>
+                {% endif %}
+            </IdentificacaoTomador>
+            <RazaoSocial>{{ rps.tomador.razao_social }}</RazaoSocial>
+            <Endereco>
+                <Endereco>{{ rps.tomador.logradouro }}</Endereco>
+                <Numero>{{ rps.tomador.numero }}</Numero>
+                {% if rps.tomador.complemento is defined -%}
+                <Complemento>{{ rps.tomador.complemento }}</Complemento>
+                {% endif %}
+                <Bairro>{{ rps.tomador.bairro }}</Bairro>
+                <CodigoMunicipio>{{ rps.tomador.codigo_municipio }}</CodigoMunicipio>
+                <Uf>{{ rps.tomador.uf }}</Uf>
+                <Cep>{{ rps.tomador.cep }}</Cep>
+            </Endereco>
+            <Contato>
+                {% if rps.tomador.telefone is defined -%}
+                <Telefone>{{ rps.tomador.telefone }}</Telefone>
+                {% endif %}
+                {% if rps.tomador.email is defined -%}
+                <Telefone>{{ rps.tomador.email }}</Telefone>
+                {% endif %}
+            </Contato>
+            <OrgaoGerador>
+                <CodigoMunicipio>{{ rps.tomador.orgao_gerador.codigo_municipio }}</CodigoMunicipio>
+                <Uf>{{ rps.tomador.orgao_gerador.uf }}</Uf>
+            </OrgaoGerador>
+        </TomadorServico>
+        {% if rps.intermediario is defined -%}
+        <IntermediarioServico>
+            <RazaoSocial>{{ rps.intermediario.razao_social }}</RazaoSocial>
+            <CpfCnpj>
+                <Cnpj>{{ rps.intermediario.cnpj }}</Cnpj>
+            </CpfCnpj>
+            <InscricaoMunicipal>{{ rps.intermediario.inscricao_municipal }}</InscricaoMunicipal>
+        </IntermediarioServico>
+        {% endif %}
+        {% if rps.construcao_civil is defined -%}
+        <ContrucaoCivil>
+            <CodigoObra>{{ rps.construcao_civil.codigo_obra }}</CodigoObra>
+            <Art>{{ rps.construcao_civil.art }}</Art>
+        </ContrucaoCivil>
+        {% endif %}
+    </InfRps>
+</Rps>
diff --git a/pytrustnfe/nfse/natal/templates/cancelarNfse.xml b/pytrustnfe/nfse/natal/templates/cancelarNfse.xml
new file mode 100644
index 0000000..f5ec25b
--- /dev/null
+++ b/pytrustnfe/nfse/natal/templates/cancelarNfse.xml
@@ -0,0 +1,15 @@
+<CancelarNfseEnvio xmlns="http://nfse.abrasf.org.br">
+    <Pedido>
+        <InfPedidoCancelamento Id="1">
+            <IdentificacaoNfse>
+                <Numero>{{ cancelamento.numero_nfse }}</Numero>
+                <CpfCnpj>
+                    <Cnpj>{{ cancelamento.cnpj_prestador }}</Cnpj>
+                </CpfCnpj>
+                <InscricaoMunicipal>{{ cancelamento.inscricao_municipal }}</InscricaoMunicipal>
+                <CodigoMunicipio>{{ cancelamento.cidade }}</CodigoMunicipio>
+            </IdentificacaoNfse>
+            <CodigoCancelamento>{{ cancelamento.codigo_cancelamento }}</CodigoCancelamento>
+        </InfPedidoCancelamento>
+    </Pedido>
+</CancelarNfseEnvio>
diff --git a/pytrustnfe/nfse/natal/templates/consultarLoteRps.xml b/pytrustnfe/nfse/natal/templates/consultarLoteRps.xml
new file mode 100644
index 0000000..7e4df96
--- /dev/null
+++ b/pytrustnfe/nfse/natal/templates/consultarLoteRps.xml
@@ -0,0 +1,7 @@
+<ConsultarLoteRpsEnvio xmlns="http://nfse.abrasf.org.br">
+    <Prestador>
+        <Cnpj>{{ consulta.cnpj_prestador }}</Cnpj>
+        <InscricaoMunicipal>{{ consulta.inscricao_municipal }}</InscricaoMunicipal>
+    </Prestador>
+    <Protocolo>{{ consulta.protocolo }}</Protocolo>
+</ConsultarLoteRpsEnvio>
diff --git a/pytrustnfe/nfse/natal/templates/recepcionarLoteRps.xml b/pytrustnfe/nfse/natal/templates/recepcionarLoteRps.xml
new file mode 100644
index 0000000..0756b0e
--- /dev/null
+++ b/pytrustnfe/nfse/natal/templates/recepcionarLoteRps.xml
@@ -0,0 +1,13 @@
+<EnviarLoteRpsEnvio xmlns="http://www.abrasf.org.br/ABRASF/arquivos/nfse.xsd">
+    <LoteRps Id="{{ nfse.numero_lote }}">
+        <NumeroLote>{{ nfse.numero_lote }}</NumeroLote>
+        <Cnpj>{{ nfse.cnpj_prestador }}</Cnpj>
+        <InscricaoMunicipal>{{ nfse.inscricao_municipal }}</InscricaoMunicipal>
+        <QuantidadeRps>{{ nfse.lista_rps|length }}</QuantidadeRps>
+        <ListaRps>
+            {% for rps in nfse.lista_rps -%}
+            {% include 'Rps.xml' %}
+            {% endfor %}
+        </ListaRps>
+    </LoteRps>
+</EnviarLoteRpsEnvio>