prod-tech
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
from uuid import UUID
|
|
from uuid import uuid4
|
|
|
|
from sqlalchemy import BigInteger
|
|
from sqlalchemy import Column
|
|
from sqlalchemy import ForeignKey
|
|
from sqlmodel import Field
|
|
from sqlmodel import Relationship
|
|
from sqlmodel import SQLModel
|
|
|
|
from app.models.transactions import Transaction
|
|
|
|
from .links import EventUserLink
|
|
from .user import User
|
|
|
|
|
|
class BaseEvent(SQLModel):
|
|
name: str = Field(nullable=False, default='Change name pls')
|
|
|
|
|
|
class Event(BaseEvent, table=True):
|
|
id: UUID = Field(primary_key=True, default_factory=uuid4)
|
|
owner_id: int = Field(
|
|
sa_column=Column(BigInteger(), ForeignKey('user.id'))
|
|
)
|
|
invite: str | None = Field(default_factory=uuid4)
|
|
users: list['User'] = Relationship(
|
|
back_populates='events', link_model=EventUserLink
|
|
)
|
|
transactions: list['Transaction'] = Relationship(
|
|
back_populates='event', cascade_delete=True
|
|
)
|
|
|
|
async def add_user(self, user: User):
|
|
self.users.append(user)
|
|
|
|
async def add_transaction(self, transaction: Transaction):
|
|
self.transactions.append(transaction)
|
|
|
|
|
|
class OutputEvent(BaseEvent):
|
|
id: UUID
|
|
owner: int
|
|
users: list[User]
|
|
invite: str | None
|
|
name: str
|
|
|
|
|
|
class AddUserRequest(SQLModel):
|
|
user_id: int
|