From 70cabd69bb6c6f8e3cc5d169be59f480e9f1bc5b Mon Sep 17 00:00:00 2001 From: timotheyca Date: Mon, 20 Dec 2021 16:50:21 +0300 Subject: [PATCH] serve --- v6d2ctx/serve.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/v6d2ctx/serve.py b/v6d2ctx/serve.py index ea77eb6..1883f89 100644 --- a/v6d2ctx/serve.py +++ b/v6d2ctx/serve.py @@ -15,12 +15,11 @@ def serve(main, client: discord.Client, loop: asyncio.AbstractEventLoop): def sigtermed(*_args): loop.create_task(aclose()) - if __name__ == '__main__': - signal.signal(signal.SIGTERM, sigtermed) - try: - loop.run_until_complete(main) - except (KeyboardInterrupt, InterruptedError, RuntimeError): - pass - finally: - if not client.is_closed(): - close() + signal.signal(signal.SIGTERM, sigtermed) + try: + loop.run_until_complete(main) + except (KeyboardInterrupt, InterruptedError, RuntimeError): + pass + finally: + if not client.is_closed(): + close()