Commit f927ed09 authored by Raphael Beer's avatar Raphael Beer

Refactor: clearer names for shutdown/cleanup handlers

parent af360ffc
...@@ -131,12 +131,12 @@ if args.debug is True: ...@@ -131,12 +131,12 @@ if args.debug is True:
else: else:
set_log_level('info') set_log_level('info')
async def close_sessions(app): async def shut_down(app):
log.info("Closing %s guest sessions" % len(TwitterSession.guest_sessions)) log.info("Closing %s guest sessions" % len(TwitterSession.guest_sessions))
for session in TwitterSession.guest_sessions: for session in TwitterSession.guest_sessions:
await session.close() await session.close()
async def close_database(app): async def clean_up(app):
global db global db
log.info("Closing database connection") log.info("Closing database connection")
db.close() db.close()
...@@ -151,13 +151,16 @@ def run(): ...@@ -151,13 +151,16 @@ def run():
username=args.mongo_username, username=args.mongo_username,
password=args.mongo_password password=args.mongo_password
) )
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
loop.run_until_complete(login_accounts(accounts, args.cookie_dir)) loop.run_until_complete(login_accounts(accounts, args.cookie_dir))
loop.run_until_complete(login_guests()) loop.run_until_complete(login_guests())
app = web.Application() app = web.Application()
app.add_routes(routes) app.add_routes(routes)
app.on_shutdown.append(close_sessions) app.on_shutdown.append(shut_down)
app.on_cleanup.append(close_database) app.on_cleanup.append(clean_up)
web.run_app(app, host=args.host, port=args.port) web.run_app(app, host=args.host, port=args.port)
if args.daemon: if args.daemon:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment