slightly cleaned the functions and changed os by platform, that seem …

…a more robust solution
eolus87 · Jun 13, 2023 · c66b1d6 · c66b1d6
1 parent 7e5de94
commit c66b1d6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/control/control_functions.py b/control/control_functions.py
@@ -1,24 +1,22 @@
 __author__ = "Nicolas Gutierrez"
 
 # Standard libraries
-import os
-import time
+import platform
 import asyncio
-from typing import Tuple
 # Third party libraries
 import kasa
 # Custom libraries
 
-if os.name == 'nt':
+if platform.system() == 'Windows':
     asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
 
 async def switch_on(target: str) -> bool:
     plug = kasa.SmartPlug(target)
     try:
-        await plug.turn_on()
-        await asyncio.sleep(0.1)
         await plug.update()
+        await asyncio.sleep(0.1)
+        await plug.turn_on()
         is_successful = plug.is_on
     except Exception as inst:
         print(f"Switch on function failed with error: {inst}")
@@ -35,9 +33,9 @@ def switch_on_function(target: str) -> bool:
 async def switch_off(target: str) -> bool:
     plug = kasa.SmartPlug(target)
     try:
-        await plug.turn_off()
-        await asyncio.sleep(0.1)
         await plug.update()
+        await asyncio.sleep(0.1)
+        await plug.turn_off()
         is_successful = plug.is_off
     except Exception as inst:
         print(f"Switch off function failed with error: {inst}")

diff --git a/power/power_function.py b/power/power_function.py
@@ -1,14 +1,14 @@
 __author__ = "Nicolas Gutierrez"
 
 # Standard libraries
-import os
+import platform
 import asyncio
 from typing import Tuple
 # Third party libraries
 import kasa
 # Custom libraries
 
-if os.name == 'nt':
+if platform.system() == 'Windows':
     asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())