mirror of https://gitlab.com/litecord/litecord.git
run.py: sanity of mind tweaks
This commit is contained in:
parent
790dde9c87
commit
6ee5b2fa15
61
run.py
61
run.py
|
|
@ -159,21 +159,22 @@ async def app_before_request():
|
||||||
|
|
||||||
@app.after_request
|
@app.after_request
|
||||||
async def app_after_request(resp):
|
async def app_after_request(resp):
|
||||||
|
"""Handle CORS headers."""
|
||||||
origin = request.headers.get('Origin', '*')
|
origin = request.headers.get('Origin', '*')
|
||||||
resp.headers['Access-Control-Allow-Origin'] = origin
|
resp.headers['Access-Control-Allow-Origin'] = origin
|
||||||
|
resp.headers['Access-Control-Allow-Headers'] = (
|
||||||
resp.headers['Access-Control-Allow-Headers'] = ('*, X-Super-Properties, '
|
'*, X-Super-Properties, '
|
||||||
'X-Fingerprint, '
|
'X-Fingerprint, '
|
||||||
'X-Context-Properties, '
|
'X-Context-Properties, '
|
||||||
'X-Failed-Requests, '
|
'X-Failed-Requests, '
|
||||||
'X-Debug-Options, '
|
'X-Debug-Options, '
|
||||||
'Content-Type, '
|
'Content-Type, '
|
||||||
'Authorization, '
|
'Authorization, '
|
||||||
'Origin, '
|
'Origin, '
|
||||||
'If-None-Match')
|
'If-None-Match'
|
||||||
# resp.headers['Access-Control-Allow-Methods'] = '*'
|
)
|
||||||
resp.headers['Access-Control-Allow-Methods'] = \
|
resp.headers['Access-Control-Allow-Methods'] = \
|
||||||
resp.headers.get('allow', '*')
|
resp.headers.get('allow', '*')
|
||||||
|
|
||||||
return resp
|
return resp
|
||||||
|
|
||||||
|
|
@ -203,40 +204,40 @@ async def app_set_ratelimit_headers(resp):
|
||||||
return resp
|
return resp
|
||||||
|
|
||||||
|
|
||||||
async def init_app_db(app):
|
async def init_app_db(app_):
|
||||||
"""Connect to databases.
|
"""Connect to databases.
|
||||||
|
|
||||||
Also spawns the job scheduler.
|
Also spawns the job scheduler.
|
||||||
"""
|
"""
|
||||||
log.info('db connect')
|
log.info('db connect')
|
||||||
app.db = await asyncpg.create_pool(**app.config['POSTGRES'])
|
app_.db = await asyncpg.create_pool(**app.config['POSTGRES'])
|
||||||
|
|
||||||
app.sched = JobManager()
|
app_.sched = JobManager()
|
||||||
|
|
||||||
|
|
||||||
def init_app_managers(app):
|
def init_app_managers(app_):
|
||||||
"""Initialize singleton classes."""
|
"""Initialize singleton classes."""
|
||||||
app.loop = asyncio.get_event_loop()
|
app_.loop = asyncio.get_event_loop()
|
||||||
app.ratelimiter = RatelimitManager(app.config.get('_testing'))
|
app_.ratelimiter = RatelimitManager(app.config.get('_testing'))
|
||||||
app.state_manager = StateManager()
|
app_.state_manager = StateManager()
|
||||||
|
|
||||||
app.storage = Storage(app)
|
app_.storage = Storage(app)
|
||||||
app.user_storage = UserStorage(app.storage)
|
app_.user_storage = UserStorage(app.storage)
|
||||||
|
|
||||||
app.icons = IconManager(app)
|
app_.icons = IconManager(app)
|
||||||
|
|
||||||
app.dispatcher = EventDispatcher(app)
|
app_.dispatcher = EventDispatcher(app)
|
||||||
|
|
||||||
# TODO: only pass app
|
# TODO: only pass app
|
||||||
app.presence = PresenceManager(
|
app_.presence = PresenceManager(
|
||||||
app.storage, app.user_storage,
|
app.storage, app.user_storage,
|
||||||
app.state_manager, app.dispatcher
|
app.state_manager, app.dispatcher
|
||||||
)
|
)
|
||||||
|
|
||||||
app.storage.presence = app.presence
|
app_.storage.presence = app.presence
|
||||||
|
|
||||||
|
|
||||||
async def api_index(app):
|
async def api_index(app_):
|
||||||
to_find = {}
|
to_find = {}
|
||||||
found = []
|
found = []
|
||||||
|
|
||||||
|
|
@ -252,7 +253,7 @@ async def api_index(app):
|
||||||
method = method.strip()
|
method = method.strip()
|
||||||
to_find[(path, method)] = name
|
to_find[(path, method)] = name
|
||||||
|
|
||||||
for rule in app.url_map.rules:
|
for rule in app_.url_map.rules:
|
||||||
path = rule.rule
|
path = rule.rule
|
||||||
|
|
||||||
# convert the path to the discord_endpoints file's style
|
# convert the path to the discord_endpoints file's style
|
||||||
|
|
@ -289,10 +290,10 @@ async def api_index(app):
|
||||||
log.debug('missing: {}', missing)
|
log.debug('missing: {}', missing)
|
||||||
|
|
||||||
|
|
||||||
async def post_app_start(app):
|
async def post_app_start(app_):
|
||||||
# we'll need to start a billing job
|
# we'll need to start a billing job
|
||||||
app.sched.spawn(payment_job(app))
|
app_.sched.spawn(payment_job(app_))
|
||||||
app.sched.spawn(api_index(app))
|
app_.sched.spawn(api_index(app_))
|
||||||
|
|
||||||
|
|
||||||
@app.before_serving
|
@app.before_serving
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue