notifier/main.py

import asyncio
import os
from os.path import exists

from sentry_sdk.integrations.aiohttp import AioHttpIntegration
from sentry_sdk.integrations.redis import RedisIntegration
from sentry_sdk.integrations.sqlalchemy import SqlalchemyIntegration
from sentry_sdk.integrations.strawberry import StrawberryIntegration
from sqlalchemy import inspect
from sqlalchemy.engine.create import create_engine
from starlette.applications import Starlette
from strawberry.asgi import GraphQL
from orm.notification import NotificationSeen

from resolvers.listener import notifications_worker
from resolvers.model import Notification
from resolvers.schema import schema
from services.db import Base
from services.rediscache import redis
from settings import DB_URL, DEV_SERVER_PID_FILE_NAME, MODE, SENTRY_DSN
import logging


logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger("\t[main]\t")
logger.setLevel(logging.DEBUG)


async def start_up():
    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.__table__, NotificationSeen.__table__])
        logger.info("Notification table was created.")
    else:
        logger.info("Notification table already exists.")

    task = asyncio.create_task(notifications_worker())
    logger.info(task)

    if MODE == "dev":
        if exists(DEV_SERVER_PID_FILE_NAME):
            with open(DEV_SERVER_PID_FILE_NAME, "w", encoding="utf-8") as f:
                f.write(str(os.getpid()))
    else:
        try:
            import sentry_sdk

            sentry_sdk.init(
                SENTRY_DSN,
                enable_tracing=True,
                integrations=[
                    StrawberryIntegration(async_execution=True),
                    SqlalchemyIntegration(),
                    RedisIntegration(),
                    AioHttpIntegration(),
                ],
            )
        except Exception as e:
            logger.error("sentry init error", e)


async def shutdown():
    await redis.disconnect()


app = Starlette(debug=True, on_startup=[start_up], on_shutdown=[shutdown])
app.mount("/", GraphQL(schema, debug=True))