cleaner-main

main.py

@@ -2,14 +2,8 @@ import os
 from importlib import import_module
 from os.path import exists
 
-import sentry_sdk
 from ariadne import load_schema_from_path, make_executable_schema
 from ariadne.asgi import GraphQL
-from sentry_sdk.integrations.aiohttp import AioHttpIntegration
-from sentry_sdk.integrations.ariadne import AriadneIntegration
-from sentry_sdk.integrations.redis import RedisIntegration
-from sentry_sdk.integrations.sqlalchemy import SqlalchemyIntegration
-from sentry_sdk.integrations.starlette import StarletteIntegration
 from starlette.applications import Starlette
 from starlette.routing import Route
 
@@ -18,57 +12,27 @@ from services.schema import resolvers
 from services.search import search_service
 from services.viewed import ViewedStorage
 from services.webhook import WebhookEndpoint
-from settings import DEV_SERVER_PID_FILE_NAME, MODE, SENTRY_DSN
+from settings import DEV_SERVER_PID_FILE_NAME, MODE
 
 
 import_module('resolvers')
-
 schema = make_executable_schema(load_schema_from_path('schema/'), resolvers)
 
 
-async def start_up():
-    print(f'[main] starting in {MODE} mode')
-
-    with sentry_sdk.start_transaction(op='task', name='Redis Connection'):
-        await redis.connect()
-
-    # start viewed service
-    await ViewedStorage.init()
-
-    # start search service
-    search_service.info()
-
-    if MODE == 'development':
-        # pid file management
-        if not exists(DEV_SERVER_PID_FILE_NAME):
-            with open(DEV_SERVER_PID_FILE_NAME, 'w', encoding='utf-8') as f:
-                f.write(str(os.getpid()))
-
-    if MODE == 'production':
-        # sentry monitoring
-        try:
-            sentry_sdk.init(
-                SENTRY_DSN,
-                enable_tracing=True,
-                integrations=[
-                    StarletteIntegration(),
-                    AriadneIntegration(),
-                    SqlalchemyIntegration(),
-                    RedisIntegration(),
-                    AioHttpIntegration(),
-                ],
-            )
-        except Exception as e:
-            print('[sentry] init error')
-            print(e)
-
-
-async def shutdown():
-    await redis.disconnect()
-
+async def dev_pid():
+    # pid file management
+    if MODE == 'development' and not exists(DEV_SERVER_PID_FILE_NAME):
+        with open(DEV_SERVER_PID_FILE_NAME, 'w', encoding='utf-8') as f:
+            f.write(str(os.getpid()))
+    print(f'[main] started in {MODE} mode')
 
 routes = [
     Route('/', GraphQL(schema, debug=True)),
     Route('/new-author', WebhookEndpoint),
 ]
-app = Starlette(routes=routes, debug=True, on_startup=[start_up], on_shutdown=[shutdown])
+app = Starlette(routes=routes, debug=True, on_startup=[
+        redis.connect,
+        ViewedStorage.init,
+        search_service.info,
+        dev_pid
+    ], on_shutdown=[redis.disconnect])