def validate(cls, v):
if not isinstance(v, str):
raise TypeError("string required")
cpf, cnpj = validate_docbr.CPF(), validate_docbr.CNPJ()
if not (cpf.validate(v) or cnpj.validate(v)):
raise TypeError("Invalid CPF / CNPJ")
return "".join(filter(str.isdigit, v))
def validate_type_entity(cls, v, values):
if "cpf_cnpj" not in values:
return None
cpf_cnpj = values["cpf_cnpj"]
cpf, cnpj = validate_docbr.CPF(), validate_docbr.CNPJ()
if cpf.validate(cpf_cnpj):
return EntityTypeEnum.pf
if cnpj.validate(cpf_cnpj):
return EntityTypeEnum.pj
return None
def validate_document_number(document_number: str):
document_number = validate_string(min_length=11,
max_length=14)(document_number)
CPF_validator = validate_docbr.CPF()
is_valid_CPF = CPF_validator.validate(document_number)
CNPJ_validator = validate_docbr.CNPJ()
is_valid_CNPJ = CNPJ_validator.validate(document_number)
if is_valid_CPF or is_valid_CNPJ:
return document_number
raise Exception(CONSTANTS['MESSAGE']['INVALID_DOCUMENT_NUMBER'])
def format(self):
mascara = validate_docbr.CNPJ()
mascara = mascara.mask(self.cnpj)
return mascara
def valida(self,documento):
validacao = validate_docbr.CNPJ()
validacao = validacao.validate(documento)
return (validacao)
def setUp(self):
""" Inicia novo objeto em todo os testes """
self.cnpj = docbr.CNPJ()