117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
"""Vela Platform — SQLAlchemy models."""
|
|
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import (
|
|
Boolean,
|
|
Column,
|
|
DateTime,
|
|
Float,
|
|
ForeignKey,
|
|
Integer,
|
|
String,
|
|
Text,
|
|
UniqueConstraint,
|
|
create_engine,
|
|
)
|
|
from sqlalchemy.orm import DeclarativeBase, relationship
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
email = Column(String(255), unique=True, nullable=False)
|
|
password_hash = Column(String(255), nullable=False)
|
|
active = Column(Boolean, default=True, nullable=False)
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class Service(Base):
|
|
__tablename__ = "services"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
name = Column(String(100), unique=True, nullable=False)
|
|
description = Column(Text, default="")
|
|
sell_price = Column(Float, default=0.0)
|
|
cost_price = Column(Float, default=0.0) # sum of component unit_cost * quantity
|
|
active = Column(Boolean, default=True, nullable=False)
|
|
sort_order = Column(Integer, default=0)
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime, default=lambda: datetime.now(timezone.utc),
|
|
onupdate=lambda: datetime.now(timezone.utc))
|
|
|
|
components = relationship("ServiceComponent", back_populates="service",
|
|
cascade="all, delete-orphan")
|
|
|
|
|
|
class ServiceComponent(Base):
|
|
__tablename__ = "service_components"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
service_id = Column(Integer, ForeignKey("services.id", ondelete="CASCADE"), nullable=False)
|
|
name = Column(String(100), nullable=False)
|
|
unit_cost = Column(Float, default=0.0)
|
|
unit_sell = Column(Float, default=0.0)
|
|
quantity = Column(Integer, default=1)
|
|
notes = Column(Text, default="")
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
|
|
service = relationship("Service", back_populates="components")
|
|
|
|
|
|
class Transaction(Base):
|
|
"""Frozen transaction — snapshots prices at time of sale."""
|
|
__tablename__ = "transactions"
|
|
__table_args__ = (
|
|
UniqueConstraint("source_type", "source_id", name="uq_transaction_source"),
|
|
)
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
source_type = Column(String(20), default="manual") # "manual" or "order"
|
|
source_id = Column(Integer, nullable=True)
|
|
|
|
service_id = Column(Integer, ForeignKey("services.id"), nullable=True)
|
|
service_name_snapshot = Column(String(100), nullable=False)
|
|
quantity = Column(Integer, default=1)
|
|
|
|
unit_sell_price_snapshot = Column(Float, nullable=False)
|
|
unit_cost_snapshot = Column(Float, nullable=False)
|
|
revenue_total = Column(Float, nullable=False)
|
|
cogs_total = Column(Float, nullable=False)
|
|
gross_profit_total = Column(Float, nullable=False)
|
|
|
|
month_start_date = Column(String(10), nullable=False) # "2026-06"
|
|
|
|
payment_status = Column(String(20), default="unpaid") # unpaid, partial, paid, refunded
|
|
notes = Column(Text, default="")
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
|
|
|
|
class Expense(Base):
|
|
__tablename__ = "expenses"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
name = Column(String(200), nullable=False)
|
|
amount = Column(Float, nullable=False)
|
|
category = Column(String(100), default="general")
|
|
month_start_date = Column(String(10), nullable=False) # "2026-06"
|
|
notes = Column(Text, default="")
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|
|
deleted_at = Column(DateTime, nullable=True)
|
|
|
|
|
|
class ExpenseTemplate(Base):
|
|
"""Recurring expense templates — generate actual expenses each month."""
|
|
__tablename__ = "expense_templates"
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
name = Column(String(200), nullable=False)
|
|
amount = Column(Float, nullable=False)
|
|
category = Column(String(100), default="general")
|
|
active = Column(Boolean, default=True, nullable=False)
|
|
created_at = Column(DateTime, default=lambda: datetime.now(timezone.utc))
|