tested-auth-refactoring

orm/notification.py

@@ -1,35 +1,120 @@
 import enum
-import time
+from datetime import datetime
+from enum import Enum, auto
 
-from sqlalchemy import JSON, Column, ForeignKey, Integer, String
+from sqlalchemy import JSON, Column, DateTime, ForeignKey, Integer, String
+from sqlalchemy import Enum as SQLAlchemyEnum
 from sqlalchemy.orm import relationship
 
-from auth.orm import Author
-from services.db import BaseModel as Base
+from orm.author import Author
+from orm.base import BaseModel as Base
+from services.logger import root_logger as logger
+
+
+class NotificationStatus(Enum):
+    """Статусы уведомлений."""
+
+    UNREAD = auto()
+    READ = auto()
+    ARCHIVED = auto()
+
+    @classmethod
+    def from_string(cls, value: str) -> "NotificationStatus":
+        """
+        Создает экземпляр статуса уведомления из строки.
+
+        Args:
+            value (str): Строковое представление статуса.
+
+        Returns:
+            NotificationStatus: Экземпляр статуса уведомления.
+        """
+        try:
+            return cls[value.upper()]
+        except KeyError:
+            logger.error(f"Неверный статус уведомления: {value}")
+            raise ValueError("Неверный статус уведомления")  # noqa: B904
+
+
+class NotificationKind(Enum):
+    """Типы уведомлений."""
+
+    COMMENT = auto()
+    MENTION = auto()
+    REACTION = auto()
+    FOLLOW = auto()
+    INVITE = auto()
+
+    @classmethod
+    def from_string(cls, value: str) -> "NotificationKind":
+        """
+        Создает экземпляр типа уведомления из строки.
+
+        Args:
+            value (str): Строковое представление типа.
+
+        Returns:
+            NotificationKind: Экземпляр типа уведомления.
+        """
+        try:
+            return cls[value.upper()]
+        except KeyError:
+            logger.error(f"Неверный тип уведомления: {value}")
+            raise ValueError("Неверный тип уведомления")  # noqa: B904
 
 
 class NotificationEntity(enum.Enum):
-    REACTION = "reaction"
+    """Сущности, связанные с уведомлениями."""
+
+    TOPIC = "topic"
+    COMMENT = "comment"
     SHOUT = "shout"
-    FOLLOWER = "follower"
+    AUTHOR = "author"
     COMMUNITY = "community"
 
     @classmethod
-    def from_string(cls, value):
-        return cls(value)
+    def from_string(cls, value: str) -> "NotificationEntity":
+        """
+        Создает экземпляр сущности уведомления из строки.
+
+        Args:
+            value (str): Строковое представление сущности.
+
+        Returns:
+            NotificationEntity: Экземпляр сущности уведомления.
+        """
+        try:
+            return cls(value)
+        except ValueError:
+            logger.error(f"Неверная сущность уведомления: {value}")
+            raise ValueError("Неверная сущность уведомления")  # noqa: B904
 
 
 class NotificationAction(enum.Enum):
+    """Действия в уведомлениях."""
+
     CREATE = "create"
     UPDATE = "update"
     DELETE = "delete"
-    SEEN = "seen"
-    FOLLOW = "follow"
-    UNFOLLOW = "unfollow"
+    MENTION = "mention"
+    REACT = "react"
 
     @classmethod
-    def from_string(cls, value):
-        return cls(value)
+    def from_string(cls, value: str) -> "NotificationAction":
+        """
+        Создает экземпляр действия уведомления из строки.
+
+        Args:
+            value (str): Строковое представление действия.
+
+        Returns:
+            NotificationAction: Экземпляр действия уведомления.
+        """
+        try:
+            return cls(value)
+        except ValueError:
+            logger.error(f"Неверное действие уведомления: {value}")
+            raise ValueError("Неверное действие уведомления")  # noqa: B904
 
 
 class NotificationSeen(Base):
@@ -42,22 +127,31 @@ class NotificationSeen(Base):
 class Notification(Base):
     __tablename__ = "notification"
 
-    id = Column(Integer, primary_key=True, autoincrement=True)
-    created_at = Column(Integer, server_default=str(int(time.time())))
+    id = Column(Integer, primary_key=True, index=True)
+    created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+
     entity = Column(String, nullable=False)
     action = Column(String, nullable=False)
     payload = Column(JSON, nullable=True)
 
+    status = Column(SQLAlchemyEnum(NotificationStatus), default=NotificationStatus.UNREAD)
+    kind = Column(SQLAlchemyEnum(NotificationKind), nullable=False)
+
     seen = relationship(Author, secondary="notification_seen")
 
     def set_entity(self, entity: NotificationEntity):
-        self.entity = entity.value  # type: ignore[assignment]
+        """Устанавливает сущность уведомления."""
+        self.entity = entity.value
 
     def get_entity(self) -> NotificationEntity:
+        """Возвращает сущность уведомления."""
         return NotificationEntity.from_string(self.entity)
 
     def set_action(self, action: NotificationAction):
-        self.action = action.value  # type: ignore[assignment]
+        """Устанавливает действие уведомления."""
+        self.action = action.value
 
     def get_action(self) -> NotificationAction:
+        """Возвращает действие уведомления."""
         return NotificationAction.from_string(self.action)