Merge pull request #28 from netosjb/fix-password-to-binary

[FIX]  Passphrase Warning

Author: renatonlima

src/erpbrasil/assinatura/certificado.py

@@ -19,7 +19,7 @@ class Certificado(object):
     def __init__(self, arquivo, senha, raise_expirado=True):
         """Permite informar um arquivo PFX binario ou o path do arquivo"""
 
-        self._senha = senha
+        self._senha = self._encode_senha(senha)
 
         try:
             try:
@@ -66,7 +66,7 @@ def _load_key_and_certificates(self):
         """
         return load_key_and_certificates(
             data=self._arquivo,
-            password=self._senha.encode(),
+            password=self._senha,
             backend=default_backend()
         )
 
@@ -113,6 +113,11 @@ def pkcs12(self):
         """Retorna o arquivo pfx no formato binario pkc12"""
         return self._pkcs12
 
+    def _encode_senha(self, senha):
+        if type(senha) == str:
+            return senha.encode()
+        else:
+            return senha
 
 class ArquivoCertificado(object):
     """ Classe para ser utilizada quando for necessário salvar o arquivo

tests/test_erpbrasil_certificado.py

@@ -43,3 +43,9 @@ def test_abertura_arquivo_temporariamente(self):
             self.assertTrue(os.path.exists(caminho_cert))
         self.assertFalse(os.path.exists(caminho_key))
         self.assertFalse(os.path.exists(caminho_cert))
+
+    def test_senha(self):
+        senha = self.certificado._encode_senha("123456")
+        self.assertTrue(type(senha) is bytes)
+        senha = self.certificado._encode_senha("123456".encode())
+        self.assertTrue(type(senha) is bytes)