working with fake-sensor

Author: ddkohler

yaqd-core/yaqd_core/_is_daemon.py

@@ -192,13 +192,14 @@ def main(cls):
         # Run the event loop
         try:
             asyncio.run(cls._main(config_filepath, config_file, args))
-        except asyncio.CancelledError:
-            pass
+        except KeyboardInterrupt:
+            asyncio.run(cls.shutdown_all(signal.SIGINT, loop=cls.loop))
 
     @classmethod
     async def _main(cls, config_filepath, config_file, args=None):
         """Parse command line arguments, run event loop."""
         loop = asyncio.get_running_loop()
+        cls.loop = loop
         if sys.platform.startswith("win"):
             signals = ()
         else:
@@ -208,27 +209,24 @@ async def _main(cls, config_filepath, config_file, args=None):
                 s, lambda s=s: asyncio.create_task(cls.shutdown_all(s, loop))
             )
 
-        try:
+        cls.__servers = set()
+        for section in config_file:
+            if section == "shared-settings":
+                continue
+            try:
+                config = cls._parse_config(config_file, section, args)
+            except ValueError as e:
+                logger.error(str(e))
+                continue
+            logger.debug(f"Starting {section} with {config}")
+            await cls._start_daemon(section, config, config_filepath)
+
+        while cls.__servers:
+            awaiting = cls.__servers
             cls.__servers = set()
-            for section in config_file:
-                if section == "shared-settings":
-                    continue
-                try:
-                    config = cls._parse_config(config_file, section, args)
-                except ValueError as e:
-                    logger.error(str(e))
-                    continue
-                logger.debug(f"Starting {section} with {config}")
-                await cls._start_daemon(section, config, config_filepath)
-
-            while cls.__servers:
-                awaiting = cls.__servers
-                cls.__servers = set()
-                await asyncio.wait(awaiting)
-                await asyncio.sleep(1)
-            loop.stop()
-        except KeyboardInterrupt:  # handle windows ctrl+c
-            await cls.shutdown_all(signal.SIGINT, loop)
+            await asyncio.wait(awaiting)
+            await asyncio.sleep(1)
+        loop.stop()
 
     @classmethod
     async def _start_daemon(cls, name, config, config_filepath):

yaqd-fakes/yaqd_fakes/_fake_sensor.py

@@ -39,3 +39,10 @@ async def _update_measurements(self):
             out["measurement_id"] = self._measurement_id
             self._measured = out
             await asyncio.sleep(self._config["update_period"])
+
+    async def aclose(self):
+        await asyncio.sleep(0.1)
+        self.logger.info("finished async close routine")
+
+    def close(self):
+        asyncio.get_running_loop().create_task(self.aclose())