Fix aiomqtt on Windows

sokolpezinok · Jun 9, 2024 · fa8e468 · fa8e468
1 parent 733b048
commit fa8e468
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/python/yaroc/scripts/mqtt_forwarder.py b/python/yaroc/scripts/mqtt_forwarder.py
@@ -1,5 +1,7 @@
 import asyncio
 import logging
+import os
+import sys
 import tomllib
 
 from ..sources.mqtt import MqttForwader
@@ -33,4 +35,8 @@ async def main():
     await forwarder.loop()
 
 
+if sys.platform.lower() == "win32" or os.name.lower() == "nt":
+    from asyncio import WindowsSelectorEventLoopPolicy, set_event_loop_policy
+
+    set_event_loop_policy(WindowsSelectorEventLoopPolicy())
 asyncio.run(main())
diff --git a/python/yaroc/scripts/send_punch.py b/python/yaroc/scripts/send_punch.py
@@ -1,7 +1,9 @@
 import asyncio
 import datetime
 import logging
+import os
 import socket
+import sys
 import time
 import tomllib
 
@@ -115,4 +117,8 @@ async def main():
     await ps.loop()
 
 
+if sys.platform.lower() == "win32" or os.name.lower() == "nt":
+    from asyncio import WindowsSelectorEventLoopPolicy, set_event_loop_policy
+
+    set_event_loop_policy(WindowsSelectorEventLoopPolicy())
 asyncio.run(main())