50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
from importlib import import_module
|
|
from ariadne import load_schema_from_path, make_executable_schema
|
|
from ariadne.asgi import GraphQL
|
|
from starlette.applications import Starlette
|
|
from starlette.middleware import Middleware
|
|
from starlette.middleware.authentication import AuthenticationMiddleware
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
from starlette.routing import Route
|
|
from auth.authenticate import JWTAuthenticate
|
|
from auth.oauth import oauth_login, oauth_authorize
|
|
from auth.email import email_authorize
|
|
from base.redis import redis
|
|
from base.resolvers import resolvers
|
|
from resolvers.zine import ShoutsCache
|
|
from storages.viewed import ViewedStorage
|
|
# from storages.gittask import GitTask
|
|
from storages.topicstat import TopicStat
|
|
from storages.shoutauthor import ShoutAuthorStorage
|
|
from storages.reactions import ReactionsStorage
|
|
import asyncio
|
|
|
|
import_module('resolvers')
|
|
schema = make_executable_schema(load_schema_from_path("schema.graphql"), resolvers)
|
|
|
|
middleware = [
|
|
Middleware(AuthenticationMiddleware, backend=JWTAuthenticate()),
|
|
Middleware(SessionMiddleware, secret_key="!secret")
|
|
]
|
|
|
|
async def start_up():
|
|
await redis.connect()
|
|
viewed_storage_task = asyncio.create_task(ViewedStorage.worker())
|
|
shouts_cache_task = asyncio.create_task(ShoutsCache.worker())
|
|
reaction_stat_task = asyncio.create_task(ReactionsStorage.worker())
|
|
shout_author_task = asyncio.create_task(ShoutAuthorStorage.worker())
|
|
topic_stat_task = asyncio.create_task(TopicStat.worker())
|
|
# FIXME git_task = asyncio.create_task(GitTask.git_task_worker())
|
|
|
|
async def shutdown():
|
|
await redis.disconnect()
|
|
|
|
routes = [
|
|
Route("/oauth/{provider}", endpoint=oauth_login), # TODO: isolate auth microservice
|
|
Route("/oauth_authorize", endpoint=oauth_authorize),
|
|
Route("/email_authorize", endpoint=email_authorize)
|
|
]
|
|
|
|
app = Starlette(debug=True, on_startup=[start_up], on_shutdown=[shutdown], middleware=middleware, routes=routes)
|
|
app.mount("/", GraphQL(schema, debug=True))
|