You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
237 lines
10 KiB
Python
237 lines
10 KiB
Python
from sqlalchemy import Boolean, Column, DateTime, Float, ForeignKey, Integer, String, Text, UniqueConstraint
|
|
from sqlalchemy.sql import func
|
|
|
|
from app.core.time_utils import utc_now
|
|
from app.db.mock_database import MockBase
|
|
|
|
|
|
class Vehicle(MockBase):
|
|
__tablename__ = "vehicles"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
modelo = Column(String(120), nullable=False)
|
|
categoria = Column(String(50), nullable=False, index=True)
|
|
preco = Column(Float, nullable=False, index=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
|
|
|
|
class Customer(MockBase):
|
|
__tablename__ = "customers"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
cpf = Column(String(11), unique=True, nullable=False, index=True)
|
|
nome = Column(String(120), nullable=False)
|
|
score = Column(Integer, nullable=False)
|
|
limite_credito = Column(Float, nullable=False)
|
|
possui_restricao = Column(Boolean, nullable=False, default=False)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
|
|
|
|
class User(MockBase):
|
|
__tablename__ = "users"
|
|
__table_args__ = (
|
|
UniqueConstraint("channel", "external_id", name="uq_mock_users_channel_external_id"),
|
|
)
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
channel = Column(String(40), nullable=False, index=True)
|
|
external_id = Column(String(120), nullable=False, index=True)
|
|
name = Column(String(120), nullable=True)
|
|
username = Column(String(120), nullable=True)
|
|
cpf = Column(String(11), ForeignKey("customers.cpf"), nullable=True, index=True)
|
|
phone = Column(String(30), nullable=True)
|
|
email = Column(String(255), nullable=True, index=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
updated_at = Column(
|
|
DateTime,
|
|
server_default=func.current_timestamp(),
|
|
onupdate=func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class Order(MockBase):
|
|
__tablename__ = "orders"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
numero_pedido = Column(String(40), unique=True, nullable=False, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
|
|
cpf = Column(String(11), ForeignKey("customers.cpf"), nullable=False, index=True)
|
|
vehicle_id = Column(Integer, ForeignKey("vehicles.id"), nullable=True, index=True)
|
|
modelo_veiculo = Column(String(120), nullable=True)
|
|
valor_veiculo = Column(Float, nullable=True)
|
|
status = Column(String(20), nullable=False, default="Ativo")
|
|
motivo_cancelamento = Column(Text, nullable=True)
|
|
data_cancelamento = Column(DateTime, nullable=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
updated_at = Column(
|
|
DateTime,
|
|
server_default=func.current_timestamp(),
|
|
onupdate=func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class ReviewSchedule(MockBase):
|
|
__tablename__ = "review_schedules"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
protocolo = Column(String(50), unique=True, nullable=False, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
|
|
placa = Column(String(10), nullable=False, index=True)
|
|
data_hora = Column(DateTime, nullable=False)
|
|
status = Column(String(20), nullable=False, default="agendado")
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
|
|
|
|
class RentalVehicle(MockBase):
|
|
__tablename__ = "rental_vehicles"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
placa = Column(String(10), unique=True, nullable=False, index=True)
|
|
modelo = Column(String(120), nullable=False, index=True)
|
|
categoria = Column(String(50), nullable=False, index=True)
|
|
ano = Column(Integer, nullable=False, index=True)
|
|
valor_diaria = Column(Float, nullable=False, index=True)
|
|
status = Column(String(20), nullable=False, default="disponivel", index=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
|
|
|
|
class RentalContract(MockBase):
|
|
__tablename__ = "rental_contracts"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
contrato_numero = Column(String(50), unique=True, nullable=False, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
|
|
cpf = Column(String(11), ForeignKey("customers.cpf"), nullable=True, index=True)
|
|
rental_vehicle_id = Column(Integer, ForeignKey("rental_vehicles.id"), nullable=False, index=True)
|
|
placa = Column(String(10), nullable=False, index=True)
|
|
modelo_veiculo = Column(String(120), nullable=False)
|
|
categoria = Column(String(50), nullable=False, index=True)
|
|
data_inicio = Column(DateTime, nullable=False, index=True)
|
|
data_fim_prevista = Column(DateTime, nullable=False, index=True)
|
|
data_devolucao = Column(DateTime, nullable=True, index=True)
|
|
valor_diaria = Column(Float, nullable=False)
|
|
valor_previsto = Column(Float, nullable=False)
|
|
valor_final = Column(Float, nullable=True)
|
|
status = Column(String(20), nullable=False, default="ativa", index=True)
|
|
observacoes = Column(Text, nullable=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
updated_at = Column(
|
|
DateTime,
|
|
server_default=func.current_timestamp(),
|
|
onupdate=func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class RentalPayment(MockBase):
|
|
__tablename__ = "rental_payments"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
protocolo = Column(String(50), unique=True, nullable=False, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
|
|
rental_contract_id = Column(Integer, ForeignKey("rental_contracts.id"), nullable=True, index=True)
|
|
contrato_numero = Column(String(50), nullable=True, index=True)
|
|
placa = Column(String(10), nullable=True, index=True)
|
|
valor = Column(Float, nullable=False)
|
|
data_pagamento = Column(DateTime, nullable=True, index=True)
|
|
favorecido = Column(String(120), nullable=True)
|
|
identificador_comprovante = Column(String(120), nullable=True, index=True)
|
|
observacoes = Column(Text, nullable=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
|
|
|
|
class RentalFine(MockBase):
|
|
__tablename__ = "rental_fines"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
protocolo = Column(String(50), unique=True, nullable=False, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
|
|
rental_contract_id = Column(Integer, ForeignKey("rental_contracts.id"), nullable=True, index=True)
|
|
contrato_numero = Column(String(50), nullable=True, index=True)
|
|
placa = Column(String(10), nullable=True, index=True)
|
|
auto_infracao = Column(String(60), nullable=True, index=True)
|
|
orgao_emissor = Column(String(120), nullable=True)
|
|
valor = Column(Float, nullable=False)
|
|
data_infracao = Column(DateTime, nullable=True, index=True)
|
|
vencimento = Column(DateTime, nullable=True, index=True)
|
|
status = Column(String(20), nullable=False, default="registrada", index=True)
|
|
observacoes = Column(Text, nullable=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
|
|
|
|
class IntegrationRoute(MockBase):
|
|
__tablename__ = "integration_routes"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"event_type",
|
|
"provider",
|
|
"recipient_email",
|
|
name="uq_integration_routes_target",
|
|
),
|
|
)
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
event_type = Column(String(80), nullable=False, index=True)
|
|
provider = Column(String(40), nullable=False, index=True)
|
|
enabled = Column(Boolean, nullable=False, default=True, index=True)
|
|
recipient_email = Column(String(255), nullable=False, index=True)
|
|
recipient_name = Column(String(120), nullable=True)
|
|
subject_template = Column(Text, nullable=True)
|
|
body_template = Column(Text, nullable=True)
|
|
provider_config_json = Column(Text, nullable=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
updated_at = Column(
|
|
DateTime,
|
|
server_default=func.current_timestamp(),
|
|
onupdate=func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class IntegrationDelivery(MockBase):
|
|
__tablename__ = "integration_deliveries"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
route_id = Column(Integer, ForeignKey("integration_routes.id"), nullable=False, index=True)
|
|
event_type = Column(String(80), nullable=False, index=True)
|
|
provider = Column(String(40), nullable=False, index=True)
|
|
status = Column(String(20), nullable=False, default="pending", index=True)
|
|
payload_json = Column(Text, nullable=False)
|
|
recipient_email = Column(String(255), nullable=True, index=True)
|
|
recipient_name = Column(String(120), nullable=True)
|
|
rendered_subject = Column(Text, nullable=True)
|
|
rendered_body = Column(Text, nullable=True)
|
|
provider_message_id = Column(String(120), nullable=True, index=True)
|
|
idempotency_key = Column(String(64), unique=True, nullable=False, index=True)
|
|
attempts = Column(Integer, nullable=False, default=0)
|
|
last_error = Column(Text, nullable=True)
|
|
dispatched_at = Column(DateTime, nullable=True, index=True)
|
|
created_at = Column(DateTime, server_default=func.current_timestamp())
|
|
updated_at = Column(
|
|
DateTime,
|
|
server_default=func.current_timestamp(),
|
|
onupdate=func.current_timestamp(),
|
|
)
|
|
|
|
|
|
class ConversationTurn(MockBase):
|
|
__tablename__ = "conversation_turns"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
request_id = Column(String(80), unique=True, nullable=False, index=True)
|
|
conversation_id = Column(String(120), nullable=False, index=True)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
|
|
channel = Column(String(40), nullable=True, index=True)
|
|
external_id = Column(String(120), nullable=True, index=True)
|
|
username = Column(String(120), nullable=True)
|
|
user_message = Column(Text, nullable=False)
|
|
assistant_response = Column(Text, nullable=True)
|
|
turn_status = Column(String(20), nullable=False, default="completed", index=True)
|
|
intent = Column(String(80), nullable=True, index=True)
|
|
domain = Column(String(40), nullable=True, index=True)
|
|
action = Column(String(80), nullable=True)
|
|
tool_name = Column(String(120), nullable=True, index=True)
|
|
tool_arguments = Column(Text, nullable=True)
|
|
error_detail = Column(Text, nullable=True)
|
|
elapsed_ms = Column(Float, nullable=True)
|
|
started_at = Column(DateTime, nullable=False, default=utc_now, index=True)
|
|
completed_at = Column(DateTime, nullable=True, index=True)
|