111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
import datetime
|
|
import decimal
|
|
import typing
|
|
|
|
from sqlalchemy import Column, ForeignKey, Numeric, Table
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
|
|
|
|
TABLE_PREFIX = "aps_"
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class Account(Base):
|
|
__tablename__ = TABLE_PREFIX + "account"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
name: Mapped[str] = mapped_column(nullable=False, unique=True)
|
|
|
|
users: Mapped[list["User"]] = relationship(
|
|
"User",
|
|
secondary=TABLE_PREFIX + "user_account_association",
|
|
back_populates="accounts",
|
|
)
|
|
transactions: Mapped[list["Transaction"]] = relationship("Transaction")
|
|
|
|
@property
|
|
def balance(self):
|
|
return sum(t.total_amount for t in self.transactions)
|
|
|
|
|
|
user_account_association = Table(
|
|
TABLE_PREFIX + "user_account_association",
|
|
Base.metadata,
|
|
Column("user_id", ForeignKey(TABLE_PREFIX + "user.id"), primary_key=True),
|
|
Column("account_id", ForeignKey(TABLE_PREFIX + "account.id"), primary_key=True),
|
|
)
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = TABLE_PREFIX + "user"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
username: Mapped[str] = mapped_column(nullable=False, unique=True)
|
|
display_name: Mapped[str] = mapped_column(nullable=False)
|
|
|
|
accounts: Mapped[list["Account"]] = relationship(
|
|
"Account", secondary=user_account_association, back_populates="users"
|
|
)
|
|
|
|
|
|
class Area(Base):
|
|
__tablename__ = TABLE_PREFIX + "area"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
name: Mapped[str] = mapped_column(nullable=False, unique=True)
|
|
description: Mapped[str] = mapped_column(nullable=True)
|
|
image_path: Mapped[str] = mapped_column(nullable=True)
|
|
|
|
products: Mapped[list["Product"]] = relationship("Product")
|
|
|
|
|
|
UnitsOfMeasure = typing.Literal[
|
|
"g",
|
|
"kg",
|
|
"piece",
|
|
]
|
|
|
|
|
|
class Product(Base):
|
|
__tablename__ = TABLE_PREFIX + "product"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
name: Mapped[str] = mapped_column(nullable=False, unique=True)
|
|
|
|
price: Mapped[decimal.Decimal] = mapped_column(Numeric(10, 2))
|
|
unit_of_measure: Mapped[UnitsOfMeasure] = mapped_column(nullable=False)
|
|
# TODO: limit this to actually used vat rates?
|
|
vat_rate: Mapped[decimal.Decimal] = mapped_column(Numeric(10, 2))
|
|
|
|
area_id: Mapped[int] = mapped_column(ForeignKey(TABLE_PREFIX + "area.id"))
|
|
area: Mapped["Area"] = relationship("Area", back_populates="products")
|
|
|
|
image_path: Mapped[str] = mapped_column(nullable=True)
|
|
|
|
|
|
TransactionTypes = typing.Literal[
|
|
"product",
|
|
"deposit",
|
|
"withdrawal",
|
|
"expense",
|
|
]
|
|
|
|
|
|
class Transaction(Base):
|
|
__tablename__ = TABLE_PREFIX + "transaction"
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
type: Mapped[TransactionTypes] = mapped_column(nullable=False)
|
|
quantity: Mapped[decimal.Decimal] = mapped_column(Numeric(10, 2), nullable=True)
|
|
timestamp: Mapped[datetime.datetime] = mapped_column(
|
|
nullable=False, default=datetime.datetime.now()
|
|
)
|
|
total_amount: Mapped[decimal.Decimal] = mapped_column(
|
|
Numeric(10, 2), nullable=False
|
|
)
|
|
|
|
product_id: Mapped[int] = mapped_column(
|
|
ForeignKey(TABLE_PREFIX + "product.id"), nullable=True
|
|
)
|
|
product: Mapped["Product"] = relationship("Product")
|
|
|
|
account_id: Mapped[int] = mapped_column(ForeignKey(TABLE_PREFIX + "account.id"))
|
|
account: Mapped["Account"] = relationship("Account", back_populates="transactions")
|