Criação de transferência em lotes pelo PIX

README.rst

@@ -50,11 +50,11 @@ API's
 ---------------------
 
 - ☑ API de Cobrança (geração de boletos)
-- ☑ API PIX (recebimento PIX) {essa API ainda está instável e incompleta no BB}
+- ☑ API PIX (recebimento PIX)
 - ☐ API Arrecadação PIX {sem previsão de implementação}
-- ☑ API Lotes de Pagamentos {essa API ainda está instável e incompleta no BB}
+- ☑ API Lotes de Pagamentos
 
-  - ☐ Transferência PIX
+  - ☑ Transferência PIX
   - ☑ Transferência Bancária
   - ☐ Pagamento GPS
   - ☐ Pagamento GRU
@@ -70,6 +70,7 @@ Recursos auxiliares
 - ☑ Geração, validação e conversão de código de barras de tributos
 - ☑ Geração de QR Code PIX
 - ☑ Validação e limpeza de CPF/CNPJ
+- ☑ Validação do tipo de chave PIX 
 
 Exemplos disponíveis
 =====================

bb_wrapper/models/pagamentos.py

@@ -2,7 +2,6 @@
 from enum import IntEnum, Enum
 
 from pydantic import BaseModel, root_validator
-from pycpfcnpj import cpfcnpj
 
 from .perfis import TipoInscricaoEnum
 
@@ -30,7 +29,6 @@ class TipoChavePIX(IntEnum):
     2: Chave Pix tipo Email
     3: Chave Pix tipo CPF/CNPJ
     4: Chave Aleatória
-    5: Dados Bancários
     """
 
     telefone = 1
@@ -39,6 +37,18 @@ class TipoChavePIX(IntEnum):
     uuid = 4
 
 
+class TipoContaPIX(IntEnum):
+    """
+    1: Conta do tipo corrente
+    2: Conta do tipo pagamento
+    3: Conta do tipo poupança
+    """
+
+    conta_corrente = 1
+    conta_pagamento = 2
+    conta_poupanca = 3
+
+
 class FinalidadeTED(IntEnum):
     """
     1: Conta corrente outros bancos
@@ -51,17 +61,19 @@ class FinalidadeTED(IntEnum):
     poupanca = 11
 
 
-class TransferenciaPIX(BaseModel):
-    data: str
+class TransferenciaChavePIX(BaseModel):
+    data: int
     valor: float
     chave: str
     formaIdentificacao: Optional[TipoChavePIX]
+    descricaoPagamento: Optional[str]
     dddTelefone: Optional[int]
     telefone: Optional[int]
     email: Optional[str]
     cpf: Optional[int]
     cnpj: Optional[int]
     identificacaoAleatoria: Optional[str]
+    documento: Optional[int]
 
     # noinspection PyMethodParameters
     @root_validator
@@ -74,23 +86,70 @@ def _set_data(cls, values):
         from ..services.pix import PixService
 
         key = values.get("chave")
-
         key_type = PixService().identify_key_type(key)
-        values["formaIdentificacao"] = key_type
+        values["formaIdentificacao"] = key_type.value
 
         if key_type == TipoChavePIX.telefone:
             values["dddTelefone"] = int(key[:2])
             values["telefone"] = int(key[2:])
         elif key_type == TipoChavePIX.email:
+            PixService().verify_email(key)
             values["email"] = key
         elif key_type == TipoChavePIX.uuid:
             values["identificacaoAleatoria"] = key
         elif key_type == TipoChavePIX.documento:
-            key_value = cpfcnpj.clear_punctuation(key)
-            if len(key_value) == 1:
-                values["cpf"] = int(key_value)
-            else:
-                values["cnpj"] = int(key_value)
+            PixService().verify_document(key, values)
+
+        documento = values.pop("documento", None)
+        if documento is not None:
+            PixService().verify_document(documento, values)
+
+        PixService().remove_null_values(values)
+
+        values.pop("chave")
+        return values
+
+
+class TransferenciaDadosBancariosPIX(BaseModel):
+    data: int
+    valor: float
+    cpf: Optional[int]
+    cnpj: Optional[int]
+    tipoConta: TipoContaPIX
+    agencia: Optional[int]
+    conta: Optional[int]
+    digitoVerificadorConta: Optional[str]
+    formaIdentificacao: Optional[int]
+    descricaoPagamento: Optional[str]
+    cpf: Optional[int]
+    cnpj: Optional[int]
+    contaPagamento: Optional[str]
+    numeroCOMPE: int
+
+    # noinspection PyMethodParameters
+    @root_validator
+    def _set_data(cls, values):
+        """
+        Esse método realiza o processamento em cima do valor 'chave'
+        identificando que tipo de chave é e configurando-a corretamente
+        no objeto.
+        """
+        from ..services.pix import PixService
+
+        values["formaIdentificacao"] = 5
+
+        tipo_conta = values.get("tipo_conta_favorecido")
+        if TipoContaPIX.conta_pagamento == tipo_conta:
+            values.pop("agencia")
+            values.pop("conta")
+            values.pop("digitoVerificadorConta")
+        else:
+            values.pop("contaPagamento")
+
+        PixService().remove_null_values(values)
+        if "cnpj" not in values and "cpf" not in values:
+            raise ValueError("É necessário inserir o CNPj ou CPF")
+
         return values
 
 

bb_wrapper/services/__init__.py

@@ -4,6 +4,7 @@
 from .barcode import BarcodeService
 from .mod import ModService  # noqa: F401
 from .pixcode import PixCodeService  # noqa: F401
+from .pix import PixService  # noqa: F401
 
 
 parse_unicode_to_alphanumeric = UnicodeService().parse_unicode_to_alphanumeric

bb_wrapper/services/document.py

@@ -1,4 +1,5 @@
 from pycpfcnpj import cpfcnpj
+from bb_wrapper.models.perfis import TipoInscricaoEnum
 
 
 class DocumentoService:
@@ -12,9 +13,9 @@ def identifica_tipo(self, documento):
         """
         documento = self.valida(documento)
         if len(documento) == 11:
-            return 1
+            return TipoInscricaoEnum.cpf
         else:
-            return 2
+            return TipoInscricaoEnum.cnpj
 
     def valida(self, documento):
         """

bb_wrapper/services/pix.py

@@ -1,4 +1,5 @@
 import uuid
+import re
 
 from pycpfcnpj import cpfcnpj
 
@@ -36,7 +37,7 @@ def identify_key_type(self, key: str):
 
         # 4
         try:
-            is_uuid = uuid.UUID(key)
+            is_uuid = bool(uuid.UUID(key))
         except ValueError:
             is_uuid = False
 
@@ -52,3 +53,26 @@ def identify_key_type(self, key: str):
             return TipoChavePIX.uuid
         else:
             raise ValueError("Tipo de chave não identificado")
+
+    def verify_email(self, email: str, raise_exception=True):
+        regex = re.compile(
+            r"([A-Za-z0-9]+[.-_])*[A-Za-z0-9]+@[A-Za-z0-9-]+(\.[A-Z|a-z]{2,})+"
+        )
+
+        is_email_valid = re.fullmatch(regex, email)
+
+        if raise_exception and not is_email_valid:
+            raise ValueError("Email inválido!")
+        return is_email_valid
+
+    def verify_document(self, key, values: dict):
+        key_value = cpfcnpj.clear_punctuation(str(key))
+        if len(key_value) == 11:
+            values["cpf"] = int(key_value)
+        else:
+            values["cnpj"] = int(key_value)
+
+    def remove_null_values(self, values: dict):
+        keys_to_remove = [k for k, v in values.items() if v is None]
+        for key in keys_to_remove:
+            del values[key]