core/orm/invite.py

from sqlalchemy import Column, ForeignKey, Enum, String
from sqlalchemy.orm import relationship
from services.db import Base
from orm.author import Author
from orm.shout import Shout
from enum import Enum as Enumeration


class InviteStatus(Enumeration):
    PENDING = "PENDING"
    ACCEPTED = "ACCEPTED"
    REJECTED = "REJECTED"


class Invite(Base):
    __tablename__ = "invite"

    inviter_id = Column(ForeignKey("author.id"), nullable=False, index=True)
    author_id = Column(ForeignKey("author.id"), nullable=False, index=True)
    shout_id = Column(ForeignKey("shout.id"), nullable=False, index=True)
    status = Column(String, default=InviteStatus.PENDING.value)

    inviter = relationship(Author, foreign_keys=[inviter_id])
    author = relationship(Author, foreign_keys=[author_id])
    shout = relationship(Shout)