mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-18 20:01:52 +03:00
0782df5e37
# Description - Notification module ## Checklist before requesting a review Please delete options that are not relevant. - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented hard-to-understand areas - [ ] I have ideally added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged ## Screenshots (if appropriate):
26 lines
616 B
Python
26 lines
616 B
Python
from enum import Enum
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class NotificationsStatusEnum(str, Enum):
|
|
Pending = "Pending"
|
|
Done = "Done"
|
|
|
|
|
|
class Notification(BaseModel):
|
|
id: UUID
|
|
datetime: str
|
|
chat_id: Optional[UUID]
|
|
message: Optional[str]
|
|
action: str
|
|
status: NotificationsStatusEnum
|
|
|
|
def dict(self, *args, **kwargs):
|
|
notification_dict = super().dict(*args, **kwargs)
|
|
if notification_dict.get("chat_id"):
|
|
notification_dict["chat_id"] = str(notification_dict.get("chat_id"))
|
|
return notification_dict
|