discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

notification-check
All checks were successful
Deploy on push / deploy (push) Successful in 34s
Details

Browse Source
This commit is contained in:
Untone 2024-04-09 17:51:23 +03:00
parent 9e6f81606b
commit 3fd94dc0fa
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
orm/notification.py
View File

@ -3,9 +3,11 @@ from enum import Enum as Enumeration
from sqlalchemy import JSON, Column, ForeignKey, Integer, String from sqlalchemy import JSON, Column, ForeignKey, Integer, String
from sqlalchemy.orm import relationship from sqlalchemy.orm import relationship
from sqlalchemy.exc import ProgrammingError
from orm.author import Author from orm.author import Author
from services.db import Base from services.db import Base, engine
from services.logger import root_logger as logger
class NotificationEntity(Enumeration): class NotificationEntity(Enumeration):
@ -39,3 +41,11 @@ class Notification(Base):
payload = Column(JSON, nullable=True) payload = Column(JSON, nullable=True)
seen = relationship(lambda: Author, secondary='notification_seen') seen = relationship(lambda: Author, secondary='notification_seen')
try:
Notification.__table__.create(engine)
logger.info("Table `notification` was created.")
except ProgrammingError:
# Handle the exception here, for example by printing a message
logger.info("Table `notification` already exists.")