initial commit - LeDiscord plateforme des copains

backend/models/album.py

@@ -0,0 +1,58 @@
+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")