59 lines
2.2 KiB
Python
59 lines
2.2 KiB
Python
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Text, Enum as SQLEnum
|
|
from sqlalchemy.orm import relationship
|
|
from datetime import datetime
|
|
import enum
|
|
from config.database import Base
|
|
|
|
class MediaType(enum.Enum):
|
|
IMAGE = "image"
|
|
VIDEO = "video"
|
|
|
|
class Album(Base):
|
|
__tablename__ = "albums"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
title = Column(String, nullable=False)
|
|
description = Column(Text)
|
|
creator_id = Column(Integer, ForeignKey("users.id"), nullable=False)
|
|
event_id = Column(Integer, ForeignKey("events.id"), nullable=True)
|
|
cover_image = Column(String)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
# Relationships
|
|
creator = relationship("User", back_populates="albums")
|
|
event = relationship("Event", back_populates="albums")
|
|
media = relationship("Media", back_populates="album", cascade="all, delete-orphan")
|
|
|
|
class Media(Base):
|
|
__tablename__ = "media"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
album_id = Column(Integer, ForeignKey("albums.id"), nullable=False)
|
|
file_path = Column(String, nullable=False)
|
|
thumbnail_path = Column(String)
|
|
media_type = Column(SQLEnum(MediaType), nullable=False)
|
|
caption = Column(Text)
|
|
file_size = Column(Integer) # in bytes
|
|
width = Column(Integer)
|
|
height = Column(Integer)
|
|
duration = Column(Integer) # in seconds for videos
|
|
likes_count = Column(Integer, default=0)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
# Relationships
|
|
album = relationship("Album", back_populates="media")
|
|
likes = relationship("MediaLike", back_populates="media", cascade="all, delete-orphan")
|
|
|
|
class MediaLike(Base):
|
|
__tablename__ = "media_likes"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
media_id = Column(Integer, ForeignKey("media.id"), nullable=False)
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
# Relationships
|
|
media = relationship("Media", back_populates="likes")
|
|
user = relationship("User", back_populates="media_likes")
|