29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
from datetime import datetime, date
|
|
from typing import Optional
|
|
from sqlalchemy import String, Text, Float, Date, DateTime, ForeignKey, Enum, func
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
import enum
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class EntryType(str, enum.Enum):
|
|
add = "add"
|
|
remove = "remove"
|
|
|
|
|
|
class Entry(Base):
|
|
__tablename__ = "entries"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False, index=True)
|
|
entry_type: Mapped[EntryType] = mapped_column(Enum(EntryType), nullable=False)
|
|
date: Mapped[date] = mapped_column(Date, nullable=False)
|
|
bourbon_name: Mapped[Optional[str]] = mapped_column(String(255))
|
|
proof: Mapped[Optional[float]] = mapped_column(Float)
|
|
amount_shots: Mapped[float] = mapped_column(Float, nullable=False, default=1.0)
|
|
notes: Mapped[Optional[str]] = mapped_column(Text)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
|
|
user: Mapped["User"] = relationship("User", back_populates="entries")
|