39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import os
|
|
from importlib import import_module
|
|
from os.path import exists
|
|
|
|
from ariadne import load_schema_from_path, make_executable_schema
|
|
from ariadne.asgi import GraphQL
|
|
from starlette.applications import Starlette
|
|
from starlette.routing import Route
|
|
|
|
from services.rediscache import redis
|
|
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
|
|
|
|
|
|
import_module('resolvers')
|
|
schema = make_executable_schema(load_schema_from_path('schema/'), resolvers)
|
|
|
|
|
|
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=[
|
|
redis.connect,
|
|
ViewedStorage.init,
|
|
search_service.info,
|
|
dev_pid
|
|
], on_shutdown=[redis.disconnect])
|