from pydantic import BaseModel


###################
## POST /empresas #
###################
class Endereco(BaseModel):
    codigoPais: str
    descricaoPais: str
    tipoLogradouro: str | None = None
    logradouro: str
    numero: str
    complemento: str | None = None
    tipoBairro: str | None = None
    bairro: str
    codigoCidade: str
    descricaoCidade: str
    estado: str
    cep: str

class Rps(BaseModel):
    lote: int
    numeracao: list

class Email(BaseModel):
    envio: bool

class Config(BaseModel):
    rps: Rps | None = None
    email: Email | None = None
    producao: bool | None = None
    envioLimitado: bool | None = None
    integracoes: list | None = None
    prefeitura: dict | str | int | None = None

class Nfse(BaseModel):
    config: Config | None = None
    ativo: bool = False
    tipoContrato: int | None = 0

class Dfe(BaseModel):
    primeiroNsu: bool

class ConfigNfe(BaseModel):
    dfe: Dfe | None = None
    email: Email | None = None
    integracoes: dict | None = None
    producao: bool | None = None
    impressaoFcp: bool | None = None
    impressaoPartilha: bool | None = None
    versaoManual: str | None = None
    versaoEsquema: str | None = None
    numeracao: list | None = None
    calculoAutomaticoIbpt: bool | None = True

class Nfe(BaseModel):
    config: ConfigNfe | None = None
    ativo: bool = False
    tipoContrato: int | None = 0

class ConfigNfce(BaseModel):
    email: Email | None = None
    integracoes: dict | None = None
    producao: bool | None = None
    versaoManual: str | None = None
    versaoEsquema: str | None = None
    numeracao: list | None = None

class Nfce(BaseModel):
    config: ConfigNfce | None = None
    ativo: bool = False
    tipoContrato: int | None = 0


# Entidade Empresa que cadastra a credencial emissora no sistema da PlugNotas
class Empresa(BaseModel):
    endereco: Endereco
    nfse: Nfse | None = None
    nfe: Nfe | None = None
    nfce: Nfce | None = None
    incentivoFiscal: bool
    incentivadorCultural: bool
    cpfCnpj: str
    inscricaoMunicipal: str
    inscricaoEstadual: str | None = None
    certificado: str
    razaoSocial: str
    nomeFantasia: str | None = None
    simplesNacional: bool
    regimeTributario: int
    regimeTributarioEspecial: int
    id: str | None = None
    banco: str | None = None
    email: str | None = None
    id_empresa: int | None = None