feat: CRUD de CLIENTE
This commit is contained in:
@@ -0,0 +1,104 @@
|
||||
"""
|
||||
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()
|
||||
Reference in New Issue
Block a user