update

Author: jedmeng

LICENSE

@@ -0,0 +1 @@
+from .error import *

mock/__init__.py

@@ -0,0 +1,55 @@
+import asyncio
+import logging
+import argparse
+import sys
+
+
+def main(event_loop):
+    parser = argparse.ArgumentParser(prog="konkeio")
+    parser.add_argument('device', help='device type')
+
+    args = parser.parse_args()
+    device_type = args.device.lower()
+
+    if device_type == 'k1':
+        from .mock_k1 import MockK1
+        device = MockK1()
+    elif device_type == 'k2':
+        from .mock_k2 import MockK2
+        device = MockK2()
+    elif device_type == 'minik':
+        from .mock_minik import MockMiniK
+        device = MockMiniK()
+    elif device_type == 'kbulb':
+        from .mock_kbulb import MockKBulb
+        device = MockKBulb()
+    elif device_type == 'klight':
+        from .mock_klight import MockKLight
+        device = MockKLight()
+    elif device_type == 'micmul':
+        from .mock_micmul import MockMicMul
+        device = MockMicMul()
+    elif device_type == 'mul':
+        from .mock_mul import MockMul
+        device = MockMul()
+    else:
+        logging.error('Device not support: %s', device_type)
+        return False
+
+    device.start(event_loop)
+    return True
+
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.DEBUG,
+                        datefmt='%Y/%m/%d %H:%M:%S',
+                        format='%(asctime)s %(levelname)s %(message)s')
+    loop = asyncio.get_event_loop()
+
+    if not main(loop):
+        sys.exit(0)
+
+    try:
+        loop.run_forever()
+    except KeyboardInterrupt:
+        sys.exit(0)

mock/__main__.py

@@ -21,7 +21,7 @@ def set_is_online(self, is_online):
     def send_message(self, src, action, action_type, password='nopassword'):
         self.socket.send(src, self.mac, password, action, action_type)
 
-    def _message_handler(self, src, mac, password, action, action_type):
+    def _message_handler(self, src, mac, _, action, action_type):
         if not self.online:
             return
         elif action_type == 'heart':

tests/mock/mock_base_device.py mock/mock_base_device.pytests/mock/mock_base_device.py renamed to mock/mock_base_device.py

@@ -12,4 +12,3 @@ def message_handler(self, src, action, device_type):
         elif action in ['open', 'close'] and device_type == 'relay':
             self.status = action
             self.send_message(src, self.status, 'rack')
-

tests/mock/mock_k1.py mock/mock_k1.pytests/mock/mock_k1.py renamed to mock/mock_k1.py

@@ -30,4 +30,3 @@ def message_handler(self, src, action, device_type):
             self.last_power = random.randint(0, 3000) / 100
             power = 'dl_module#power#%.2f' % self.last_power
             self.send_message(src, power, 'pack')
-

tests/mock/mock_k2.py mock/mock_k2.pytests/mock/mock_k2.py renamed to mock/mock_k2.py

@@ -22,4 +22,3 @@ def message_handler(self, src, action, device_type):
             elif name == 'ctp':
                 self.ct = value
             self.send_message(src, 'x', 'kback')
-

tests/mock/mock_kbulb.py mock/mock_kbulb.pytests/mock/mock_kbulb.py renamed to mock/mock_kbulb.py

@@ -22,4 +22,3 @@ def message_handler(self, src, action, device_type):
             self.color = (r, g, b)
             self.brightness = brightness
             self.send_message(src, 'x', 'klack')
-

tests/mock/mock_klight.py mock/mock_klight.pytests/mock/mock_klight.py renamed to mock/mock_klight.py

@@ -12,4 +12,3 @@ def message_handler(self, src, action, device_type):
         elif action in ['open', 'close'] and device_type == 'relay':
             self.status = action
             self.send_message(src, self.status, 'rack')
-