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()