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

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

from ..base import Base


class Repository(Base.Model):
    __tablename__ = "repository"
    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)
    isActive = Base.Column(Base.Boolean, nullable=False, default=False)
    # Foreign Keys
    owner_id = Base.Column(Base.String, Base.ForeignKey("user.email"), nullable=False)
    # Relationships
    owner = Base.relationship("User", back_populates="owner_of")
    authorizations = Base.relationship("Authorization", back_populates="repository")
    tweets = Base.relationship("Composed", back_populates="repository")
    alerts = Base.relationship("Alert", back_populates="repository")
    uses = Base.relationship("Uses", back_populates="repository")

    def to_json(self):
        return {"id": self.id, "name": self.name, "start": (self.start.isoformat() if self.start else None),
                "isActive": self.isActive, "end": (self.end.isoformat() if self.end else None),
                "owner": self.owner.to_json()}