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

tablenames-fix-2

Browse Source
This commit is contained in:
Untone 2024-01-22 22:15:49 +03:00
parent 4722bc2300
commit e0a13423cc
1 changed files with 1 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
main.py
View File

@ -6,18 +6,13 @@ from sentry_sdk.integrations.aiohttp import AioHttpIntegration
from sentry_sdk.integrations.redis import RedisIntegration from sentry_sdk.integrations.redis import RedisIntegration
from sentry_sdk.integrations.sqlalchemy import SqlalchemyIntegration from sentry_sdk.integrations.sqlalchemy import SqlalchemyIntegration
from sentry_sdk.integrations.strawberry import StrawberryIntegration from sentry_sdk.integrations.strawberry import StrawberryIntegration
from sqlalchemy import inspect
from sqlalchemy.engine.create import create_engine
from starlette.applications import Starlette from starlette.applications import Starlette
from strawberry.asgi import GraphQL from strawberry.asgi import GraphQL
from orm.notification import NotificationSeen
from resolvers.listener import notifications_worker from resolvers.listener import notifications_worker
from resolvers.model import Notification
from resolvers.schema import schema from resolvers.schema import schema
from services.db import Base
from services.rediscache import redis from services.rediscache import redis
from settings import DB_URL, DEV_SERVER_PID_FILE_NAME, MODE, SENTRY_DSN from settings import DEV_SERVER_PID_FILE_NAME, MODE, SENTRY_DSN
import logging import logging
@ -29,17 +24,6 @@ logger.setLevel(logging.DEBUG)
async def start_up(): async def start_up():
await redis.connect() await redis.connect()
engine = create_engine(DB_URL)
# Check if the table already exists
inspector = inspect(engine)
if not inspector.has_table("notification"):
# Create the Notification table
Base.metadata.create_all(bind=engine, tables=["notification", "notification_seen"])
logger.info("Notification table was created.")
else:
logger.info("Notification table already exists.")
task = asyncio.create_task(notifications_worker()) task = asyncio.create_task(notifications_worker())
logger.info(task) logger.info(task)