src/feed/models.py

import uuid
from datetime import datetime
from enum import Enum
from typing import Optional

from sqlalchemy import UniqueConstraint
from sqlmodel import Field, SQLModel


class PostType(str, Enum):
    BIRTHDAY = "Birthday"
    NOTICE = "Notice"
    BANNER = "Banner"
    JOB_REQUEST = "Job Request"


class PostCategory(str, Enum):
    TEAM = "Team"
    GLOBAL = "Global"


class Posts(SQLModel, table=True):
    __tablename__ = "posts"
    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
    type: PostType = Field(default=PostType.NOTICE)
    category: PostCategory = Field(default=PostCategory.GLOBAL)
    caption: Optional[str] = None
    image: Optional[str] = None
    created_at: datetime = Field(default_factory=datetime.now, nullable=False)
    edited_at: datetime = Field(default_factory=datetime.now)


class Comments(SQLModel, table=True):
    __tablename__ = "comments"
    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
    post_id: uuid.UUID = Field(foreign_key="posts.id", nullable=False)
    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
    comment: str = Field(nullable=False)
    created_at: datetime = Field(default_factory=datetime.now, nullable=False)


class Likes(SQLModel, table=True):
    __tablename__ = "likes"
    __table_args__ = (UniqueConstraint("user_id", "post_id"),)
    id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
    post_id: uuid.UUID = Field(foreign_key="posts.id", nullable=False)
    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
    liked_at: datetime = Field(default_factory=datetime.now, nullable=False)