update-api-3

orm/invite.py

@@ -1,4 +1,4 @@
-from enum import Enum as Enumeration
+import enum
 
 from sqlalchemy import Column, ForeignKey, String
 from sqlalchemy.orm import relationship
@@ -6,11 +6,15 @@ from sqlalchemy.orm import relationship
 from services.db import Base
 
 
-class InviteStatus(Enumeration):
+class InviteStatus(enum.Enum):
     PENDING = "PENDING"
     ACCEPTED = "ACCEPTED"
     REJECTED = "REJECTED"
 
+    @classmethod
+    def from_string(cls, value):
+        return cls(value)
+
 
 class Invite(Base):
     __tablename__ = "invite"
@@ -20,6 +24,12 @@ class Invite(Base):
     shout_id = Column(ForeignKey("shout.id"), primary_key=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")
+    inviter = relationship("Author", foreign_keys=[inviter_id])
+    author = relationship("Author", foreign_keys=[author_id])
+    shout = relationship("Shout")
+
+    def set_status(self, status: InviteStatus):
+        self.status = status.value
+
+    def get_status(self) -> InviteStatus:
+        return InviteStatus.from_string(self.status)