Fix using manual_stop while startup task is running

mautrix · Jun 25, 2023 · 654b215 · 654b215
1 parent c93d24d
commit 654b215
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mautrix/util/program.py b/mautrix/util/program.py
@@ -214,6 +214,7 @@ def _run(self) -> None:
         signal.signal(signal.SIGINT, signal.default_int_handler)
         signal.signal(signal.SIGTERM, signal.default_int_handler)
 
+        self._stop_task = self.loop.create_future()
         exit_code = 0
         try:
             self.log.debug("Running startup actions...")
@@ -224,7 +225,6 @@ def _run(self) -> None:
                 f"Startup actions complete in {round(end_ts - start_ts, 2)} seconds, "
                 "now running forever"
             )
-            self._stop_task = self.loop.create_future()
             exit_code = self.loop.run_until_complete(self._stop_task)
             self.log.debug("manual_stop() called, stopping...")
         except KeyboardInterrupt: