Fix #10

Fixes #10

Author: undera

pylgbst/comms/cgatt.py

@@ -100,7 +100,7 @@ def connect(self, hub_mac=None):
             for dev in devices:
                 address = dev.mac_address
                 name = dev.alias()
-                if name == LEGO_MOVE_HUB or hub_mac == address:
+                if (not hub_mac and name == LEGO_MOVE_HUB) or hub_mac == address:
                     logging.info("Found %s at %s", name, address)
                     self._device = CustomDevice(address, self._manager)
                     break

pylgbst/comms/cgattlib.py

@@ -70,7 +70,7 @@ def connect(self, hub_mac=None):
             log.debug("Devices: %s", devices)
 
             for address, name in devices.items():
-                if name == LEGO_MOVE_HUB or hub_mac == address:
+                if (not hub_mac and name == LEGO_MOVE_HUB) or hub_mac == address:
                     logging.info("Found %s at %s", name, address)
                     self.requester = Requester(address, True, self._iface)
                     break

pylgbst/comms/cpygatt.py

@@ -34,7 +34,7 @@ def connect(self, hub_mac=None):
             for dev in devices:
                 address = dev['address']
                 name = dev['name']
-                if name == LEGO_MOVE_HUB or hub_mac == address:
+                if (not hub_mac and name == LEGO_MOVE_HUB) or hub_mac == address:
                     logging.info("Found %s at %s", name, address)
                     self._conn_hnd = adapter.connect(address)
                     break