Olá, diferente da aula, eu decidi fazer uma classe especifica para CPF e outra para CNPJ, na aula o professor fez uma classe misturando os dois, falando em organização de codigo, o meu está bom? ou seria melhor juntar as duas classe numa só?
from validate_docbr import CPF
from validate_docbr import CNPJ
class Cpf:
def __init__(self, documento):
documento = str(documento)
if self.cpfValido(documento):
self.cpf = documento
else:
raise ValueError('Cpf invalido')
def __str__(self):
return self.format_cpf()
def cpfValido(self, documento):
if len(str(documento)) == 11:
validador = CPF()
return validador.validate(documento)
else:
raise ValueError('Quantidade de digitos invalida')
def format_cpf(self):
mascara = CPF()
return mascara.mask(self.cpf)
class Cnpj:
def __init__(self, documento):
documento = str(documento)
if self.cnpj_valido(documento):
self.documento = documento
else:
raise ValueError('Cnpj invalido')
def __str__(self):
return self.format_cnpj()
def cnpj_valido(self, documento):
if len(documento) == 14:
cnpj = CNPJ()
if cnpj.validate(documento):
return True
else:
raise ValueError('Quantidade de digitos invalida!')
def format_cnpj(self):
cnpj = CNPJ()
return cnpj.mask(self.documento)