Minor fixes

Author: dala318

lib/device.py

@@ -28,6 +28,7 @@ def run(self):
         passcode_lenth = response[9]
         passcode_raw = response[10 : 10 + passcode_lenth]
         passcode = passcode_raw.decode("utf-8")
+        print(passcode)
 
         data = bytes.fromhex('000000030f00000800') + passcode_lenth.to_bytes(1, 'little') + passcode_raw
         self.socket.sendall(data)
@@ -146,14 +147,3 @@ def __enter__(self):
 
     def __exit__(self, type, value, traceback):
         self.socket.close()
-
-
-if __name__ == '__main__':
-    #with Device('192.168.1.89') as d:
-    #    print(d.get_result())
-
-    loop = asyncio.get_event_loop()
-    device = Device('192.168.1.89')
-    loop.run_until_complete(device.run_async())
-    print(device.get_result())
-    loop.close()

lib/discovery.py

@@ -68,7 +68,7 @@ def run(self):
         version_raw = data[idx_start: idx]
         self.result['result']['version_server'] = version_raw.decode("utf-8")
 
-        self.result['status'] = ['success']
+        self.result['status'] = ['Success']
 
     def close(self):
         self.socket.close()
@@ -82,13 +82,3 @@ def __enter__(self):
 
     def __exit__(self, type, value, traceback):
         self.close()
-
-if __name__ == '__main__':
-    with Discovery() as d:
-        print(d.get_result())
-
-    loop = asyncio.get_event_loop()
-    discovery = Discovery()
-    loop.run_until_complete(discovery.run_async())
-    print(discovery.get_result())
-    loop.close()

main.py

@@ -8,6 +8,10 @@
 result = discover.get_result()
 print(result)
 
+if result['status'][0] != 'Success':
+    print(result)
+    exit(1)
+
 dev = device.Device(result['result']['ip'])
 # listener = loop.create_task(dev.run_async())
 # asyncio.wait([listener])