105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
"""
|
|
CRUD completo de CLIENTE. Única tabela cadastral (ID autoincrement é a PK
|
|
técnica; CODIGO é o identificador de negócio usado pelo resto do sistema —
|
|
ver licenciamento-n8n/docs/MAPEAMENTO_BANCO.md).
|
|
"""
|
|
import pymysql
|
|
|
|
from .connection import get_connection
|
|
|
|
CAMPOS_CLIENTE = [
|
|
"CODIGO", "CNPJ", "RAZAO", "FANTASIA", "IE", "ENDERECO", "CIDADE",
|
|
"BAIRRO", "UF", "CONTATO", "TELEFONE", "CELULAR", "CEP", "IMPLANTACAO",
|
|
"MENSALIDADE", "JUROS_MORA", "JUROS_MULTA", "DATA_CONTRATO", "EMAIL",
|
|
"OBS", "VENCIMENTO_CONTRATO", "DIA_VENCIMENTO",
|
|
]
|
|
|
|
_COLUNAS_SELECT = ", ".join(["ID", *CAMPOS_CLIENTE])
|
|
|
|
|
|
class ClienteComDependenciasError(Exception):
|
|
"""Levantada ao tentar excluir um CLIENTE que ainda tem CLIENTE_CONTATO."""
|
|
|
|
|
|
def listar_clientes(busca: str | None = None, limite: int = 100) -> list[dict]:
|
|
conn = get_connection()
|
|
try:
|
|
with conn.cursor() as cur:
|
|
if busca:
|
|
sql = (
|
|
f"SELECT {_COLUNAS_SELECT} FROM CLIENTE "
|
|
"WHERE CODIGO LIKE %s OR CNPJ LIKE %s OR RAZAO LIKE %s OR FANTASIA LIKE %s "
|
|
"ORDER BY RAZAO LIMIT %s"
|
|
)
|
|
termo = f"%{busca}%"
|
|
cur.execute(sql, (termo, termo, termo, termo, limite))
|
|
else:
|
|
cur.execute(f"SELECT {_COLUNAS_SELECT} FROM CLIENTE ORDER BY RAZAO LIMIT %s", (limite,))
|
|
return cur.fetchall()
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def obter_cliente(cliente_id: int) -> dict | None:
|
|
conn = get_connection()
|
|
try:
|
|
with conn.cursor() as cur:
|
|
cur.execute(f"SELECT {_COLUNAS_SELECT} FROM CLIENTE WHERE ID = %s", (cliente_id,))
|
|
return cur.fetchone()
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def codigo_em_uso(codigo: str, ignorar_id: int | None = None) -> bool:
|
|
conn = get_connection()
|
|
try:
|
|
with conn.cursor() as cur:
|
|
sql = "SELECT ID FROM CLIENTE WHERE CODIGO = %s"
|
|
params = [codigo]
|
|
if ignorar_id is not None:
|
|
sql += " AND ID != %s"
|
|
params.append(ignorar_id)
|
|
cur.execute(sql, params)
|
|
return cur.fetchone() is not None
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def criar_cliente(dados: dict) -> int:
|
|
colunas = ", ".join(CAMPOS_CLIENTE)
|
|
marcadores = ", ".join(["%s"] * len(CAMPOS_CLIENTE))
|
|
valores = [dados[campo] for campo in CAMPOS_CLIENTE]
|
|
conn = get_connection()
|
|
try:
|
|
with conn.cursor() as cur:
|
|
cur.execute(f"INSERT INTO CLIENTE ({colunas}) VALUES ({marcadores})", valores)
|
|
return cur.lastrowid
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def atualizar_cliente(cliente_id: int, dados: dict) -> None:
|
|
atribuicoes = ", ".join(f"{campo} = %s" for campo in CAMPOS_CLIENTE)
|
|
valores = [dados[campo] for campo in CAMPOS_CLIENTE]
|
|
valores.append(cliente_id)
|
|
conn = get_connection()
|
|
try:
|
|
with conn.cursor() as cur:
|
|
cur.execute(f"UPDATE CLIENTE SET {atribuicoes} WHERE ID = %s", valores)
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def excluir_cliente(cliente_id: int) -> None:
|
|
conn = get_connection()
|
|
try:
|
|
with conn.cursor() as cur:
|
|
try:
|
|
cur.execute("DELETE FROM CLIENTE WHERE ID = %s", (cliente_id,))
|
|
except pymysql.err.IntegrityError as exc:
|
|
raise ClienteComDependenciasError(
|
|
"Este cliente tem contatos cadastrados (CLIENTE_CONTATO) — remova-os antes de excluir."
|
|
) from exc
|
|
finally:
|
|
conn.close()
|