iartamonov/mautrix-telegram
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Add SIGINT/SIGTERM handler

Browse Source
This commit is contained in:
Tulir Asokan
2018-10-20 21:17:36 +03:00
parent a770e1f67e
commit 2756873c53
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mautrix_telegram/__main__.py
+5 -1
View File
@@ -20,6 +20,7 @@ import asyncio
import logging.config import logging.config
import sys import sys
import copy import copy
import signal
from sqlalchemy import orm from sqlalchemy import orm
import sqlalchemy as sql import sqlalchemy as sql
@@ -126,13 +127,16 @@ with appserv.run(config["appservice.hostname"], config["appservice.port"]) as st
if context.bot: if context.bot:
startup_actions.append(context.bot.start()) startup_actions.append(context.bot.start())
signal.signal(signal.SIGINT, signal.default_int_handler)
signal.signal(signal.SIGTERM, signal.default_int_handler)
try: try:
log.debug("Initialization complete, running startup actions") log.debug("Initialization complete, running startup actions")
loop.run_until_complete(asyncio.gather(*startup_actions, loop=loop)) loop.run_until_complete(asyncio.gather(*startup_actions, loop=loop))
log.debug("Startup actions complete, now running forever") log.debug("Startup actions complete, now running forever")
loop.run_forever() loop.run_forever()
except KeyboardInterrupt: except KeyboardInterrupt:
log.debug("Keyboard interrupt received, stopping clients") log.debug("Interrupt received, stopping clients")
loop.run_until_complete( loop.run_until_complete(
asyncio.gather(*[user.stop() for user in User.by_tgid.values()], loop=loop)) asyncio.gather(*[user.stop() for user in User.by_tgid.values()], loop=loop))
log.debug("Clients stopped, shutting down") log.debug("Clients stopped, shutting down")