precreate-tables-fix

main.py

@@ -6,20 +6,42 @@ 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 DEV_SERVER_PID_FILE_NAME, MODE, SENTRY_DSN
+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())
-    print(task)
+    logger.info(task)
 
     if MODE == "dev":
         if exists(DEV_SERVER_PID_FILE_NAME):
@@ -40,8 +62,7 @@ async def start_up():
                 ],
             )
         except Exception as e:
-            print("[sentry] init error")
-            print(e)
+            logger.error("sentry init error", e)
 
 
 async def shutdown():