You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
119 lines
4.1 KiB
119 lines
4.1 KiB
# -*- coding: utf-8 -*-
|
|
# © 2017 Danimar Ribeiro, Trustcode
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
|
|
import os
|
|
import hashlib
|
|
import base64
|
|
import requests
|
|
from pytrustnfe.xml import render_xml, sanitize_response
|
|
from pytrustnfe.certificado import extract_cert_and_key_from_pfx, save_cert_key
|
|
from pytrustnfe.nfse.assinatura import Assinatura
|
|
|
|
URLS = {
|
|
'producao': {
|
|
'processar_nota': 'https://nfps-e.pmf.sc.gov.br/api/v1/processamento/notas/processa',
|
|
'cancelar_nota': 'https://nfps-e.pmf.sc.gov.br/api/v1/cancelamento/notas/cancela'
|
|
},
|
|
'homologacao': {
|
|
'processar_nota': 'https://nfps-e-hml.pmf.sc.gov.br/api/v1/processamento/notas/processa',
|
|
'cancelar_nota': 'https://nfps-e-hml.pmf.sc.gov.br/api/v1/cancelamento/notas/cancela'
|
|
}
|
|
}
|
|
|
|
|
|
def _render(certificado, method, **kwargs):
|
|
path = os.path.join(os.path.dirname(__file__), 'templates')
|
|
xml_send = render_xml(path, '%s.xml' % method, False, **kwargs)
|
|
|
|
cert, key = extract_cert_and_key_from_pfx(
|
|
certificado.pfx, certificado.password)
|
|
cert, key = save_cert_key(cert, key)
|
|
signer = Assinatura(cert, key, certificado.password)
|
|
xml_send = signer.assina_xml(xml_send, '')
|
|
return xml_send
|
|
|
|
|
|
def _get_oauth_token(**kwargs):
|
|
if kwargs['ambiente'] == 'producao':
|
|
url = 'https://nfps-e.pmf.sc.gov.br/api/v1/autenticacao/oauth/token'
|
|
else:
|
|
url = 'https://nfps-e-hml.pmf.sc.gov.br/api/v1/autenticacao/oauth/token'
|
|
|
|
m = hashlib.md5()
|
|
secret = "%s:%s" % (kwargs["client_id"], kwargs["secret_id"])
|
|
auth = base64.b64encode(secret.encode('utf-8'))
|
|
headers = {
|
|
"Content-Type": "application/x-www-form-urlencoded",
|
|
"Authorization": "Basic %s" % auth.decode('utf-8').replace('\n', '')
|
|
}
|
|
m.update(kwargs["password"].encode('utf-8'))
|
|
password = m.hexdigest().upper()
|
|
|
|
dados = "grant_type=password&username=%s&password=%s&client_id=%s&client_secret=%s" % (
|
|
kwargs["username"], password, kwargs["client_id"], kwargs["secret_id"])
|
|
r = requests.post(url, data=dados, headers=headers)
|
|
if r.status_code == 200:
|
|
return r.json()
|
|
else:
|
|
return r.json()
|
|
|
|
|
|
def _send(certificado, method, **kwargs):
|
|
url = URLS[kwargs['ambiente']][method]
|
|
xml_send = kwargs['xml']
|
|
|
|
token = _get_oauth_token(**kwargs)
|
|
if "access_token" not in token:
|
|
raise Exception("%s - %s: %s" % (token["status"], token["error"],
|
|
token["message"]))
|
|
kwargs.update({"numero": 1, 'access_token': token["access_token"]})
|
|
|
|
headers = {"Accept": "application/xml;charset=UTF-8",
|
|
"Content-Type": "application/xml",
|
|
"Authorization": "Bearer %s" % kwargs['access_token']}
|
|
r = requests.post(url, headers=headers, data=xml_send)
|
|
|
|
response, obj = sanitize_response(r.text.strip().encode('utf-8'))
|
|
return {
|
|
'sent_xml': xml_send,
|
|
'received_xml': response,
|
|
'object': obj,
|
|
'status_code': r.status_code,
|
|
}
|
|
|
|
|
|
def xml_processar_nota(certificado, **kwargs):
|
|
return _render(certificado, 'processar_nota', **kwargs)
|
|
|
|
|
|
def processar_nota(certificado, **kwargs):
|
|
if "xml" not in kwargs:
|
|
kwargs['xml'] = xml_processar_nota(certificado, **kwargs)
|
|
return _send(certificado, 'processar_nota', **kwargs)
|
|
|
|
|
|
def xml_cancelar_nota(certificado, **kwargs):
|
|
return _render(certificado, 'cancelar_nota', **kwargs)
|
|
|
|
|
|
def cancelar_nota(certificado, **kwargs):
|
|
if "xml" not in kwargs:
|
|
kwargs['xml'] = xml_cancelar_nota(certificado, **kwargs)
|
|
return _send(certificado, 'cancelar_nota', **kwargs)
|
|
|
|
|
|
def consultar_nota(certificado, **kwargs):
|
|
if kwargs['ambiente'] == 'producao':
|
|
url = "https://nfps-e.pmf.sc.gov.br/api/v1/consultas/notas/numero/%s" % (kwargs["numero"])
|
|
else:
|
|
url = "https://nfps-e-hml.pmf.sc.gov.br/api/v1/consultas/notas/numero/%s" % (kwargs["numero"])
|
|
|
|
headers = {"Accept": "application/json",
|
|
"Authorization": "Bearer %s" % kwargs['access_token']}
|
|
r = requests.get(url, headers=headers)
|
|
print(r.status_code)
|
|
if r.status_code == 200:
|
|
return r.text
|
|
else:
|
|
return r.text
|