Refactor.

1 files changed, 92 insertions, 0 deletions

diff --git a/bluetooth_paired_devices.py b/bluetooth_paired_devices.py
new file mode 100644
index 0000000..139ca3d
--- /dev/null
+++ b/bluetooth_paired_devices.py
@@ -0,0 +1,92 @@
+import sys
+import re
+import os
+
+from resource_manager import GenericResourceManager
+
+if sys.platform == "darwin":
+    from binplist import binplist
+    import lightblue
+else:
+    import bluetooth
+    if sys.platform == "win32":
+        import _winreg
+
+
+def get_parrot_zik_mac():
+        p = re.compile('90:03:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}|'
+                       'A0:14:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}:[0-9A-Fa-f]{2}')
+        if sys.platform == "linux2":
+            out = os.popen("bluez-test-device list").read()
+            res = p.findall(out)
+            if len(res) > 0:
+                return res[0]
+
+        elif sys.platform == "darwin":
+            fd = open("/Library/Preferences/com.apple.Bluetooth.plist", "rb")
+            plist = binplist.BinaryPlist(file_obj=fd)
+            parsed_plist = plist.Parse()
+            try:
+                for mac in parsed_plist['PairedDevices']:
+                    if p.match(mac.replace("-", ":")):
+                        return mac.replace("-", ":")
+            except Exception:
+                pass
+
+        elif sys.platform == "win32":
+            aReg = _winreg.ConnectRegistry(None, _winreg.HKEY_LOCAL_MACHINE)
+            aKey = _winreg.OpenKey(
+                aReg, 'SYSTEM\CurrentControlSet\Services\
+                BTHPORT\Parameters\Devices')
+            for i in range(10):
+                try:
+                    asubkey_name = _winreg.EnumKey(aKey, i)
+                    mac = ':'.join(asubkey_name[i:i+2] for i in range(0, 12, 2))
+                    res = p.findall(mac)
+                    if len(res) > 0:
+                        return res[0]
+
+                except EnvironmentError:
+                    pass
+
+
+def connect(addr=None):
+    if sys.platform == "darwin":
+        service_matches = lightblue.findservices(
+            name="Parrot RFcomm service", addr=addr)
+    else:
+        uuids = ["0ef0f502-f0ee-46c9-986c-54ed027807fb",
+                 "8B6814D3-6CE7-4498-9700-9312C1711F63"]
+        service_matches = []
+        for uuid in uuids:
+            service_matches = bluetooth.find_service(uuid=uuid, address=addr)
+            if service_matches:
+                break
+
+    if len(service_matches) == 0:
+        print "Failed to find Parrot Zik RFCOMM service"
+        return GenericResourceManager(None)
+
+    if sys.platform == "darwin":
+        first_match = service_matches[0]
+        port = first_match[1]
+        name = first_match[2]
+        host = first_match[0]
+    else:
+        first_match = service_matches[0]
+        port = first_match["port"]
+        name = first_match["name"]
+        host = first_match["host"]
+
+    print "Connecting to \"%s\" on %s" % (name, host)
+
+    if sys.platform == "darwin":
+        sock = lightblue.socket()
+    else:
+        sock = bluetooth.BluetoothSocket(bluetooth.RFCOMM)
+
+    sock.connect((host, port))
+
+    sock.send('\x00\x03\x00')
+    sock.recv(1024)
+    return GenericResourceManager(sock)