pds-2021-g2-nest/code/backend/nest_backend/database/tables/Repository.py

"""
This module defines the :class:`.Repository` database class.
"""

from ..base import Base
from .Enums import ConditionMode


class Repository(Base.Model):
    __tablename__ = "repository"

    # Columns
    id = Base.Column(Base.Integer, primary_key=True)
    name = Base.Column(Base.String, nullable=False)
    start = Base.Column(Base.DateTime, nullable=True)
    end = Base.Column(Base.DateTime, nullable=True)
    is_active = Base.Column(Base.Boolean, nullable=False, default=False)
    evaluation_mode = Base.Column(Base.Enum(ConditionMode), default=ConditionMode.all_or.value)

    # Foreign Keys
    owner_id = Base.Column(Base.String, Base.ForeignKey("user.email", ondelete="CASCADE"), nullable=False)

    # Relationships
    owner = Base.relationship("User", back_populates="owner_of")
    authorizations = Base.relationship("Authorization", back_populates="repository", cascade="all, delete")
    tweets = Base.relationship("Composed", back_populates="repository", cascade="all, delete")
    alerts = Base.relationship("Alert", back_populates="repository", cascade="all, delete")
    conditions = Base.relationship("Condition", back_populates="repository")

    def to_json(self):
        return {
            "id": self.id,
            "name": self.name,
            "start": self.start.isoformat() if self.start else None,
            "is_active": self.is_active,
            "end": self.end.isoformat() if self.end else None,
            "owner": self.owner.to_json(),
            "evaluation_mode": self.evaluation_mode,
            "conditions": [c.to_json() for c in self.conditions]
        }