WIP: handler.py

FRESH-SHIN · Dec 17, 2023 · 26a2712 · 26a2712
1 parent 4ee3e76
commit 26a2712
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/nlabosesamebot/sesame/handler.py b/nlabosesamebot/sesame/handler.py
@@ -44,13 +44,13 @@ async def connect(self):
         await self.device.wait_for_login()
     async def unlock(self):
         logging.debug("opening")
-        if self.device is None:
+        if self.device is None or not (self.device.getDeviceStatus() == CHSesame2Status.Unlocked or self.device.getDeviceStatus() == CHSesame2Status.Locked) :
             await self.connect()
-        await self.device.unlock(history_tag="My Script")
+        await self.device.unlock(history_tag="Discord")
 
     async def lock(self):
         logging.debug("closing")
-        if self.device is None:
+        if self.device is None  or not (self.device.getDeviceStatus() == CHSesame2Status.Unlocked or self.device.getDeviceStatus() == CHSesame2Status.Locked) :
             await self.connect()
-        await self.device.lock(history_tag="My Script")
+        await self.device.lock(history_tag="Discord")