Merge pull request #1 from dherrada/adding-ams

Adding ams (I actually tested this one, believe it or not)

Author: evaherrada

CircuitPython_Pyloton/code.py

@@ -18,32 +18,29 @@
 pyloton = pyloton.Pyloton(ble, display, 84.229)
 pyloton.show_splash()
 
+pyloton.ams_connect()
+
 hr_connection = None
-# Start with a fresh connection.
-if ble.connected:
-    for connection in ble.connections:
-        if HeartRateService in connection:
-            connection.disconnect()
-        break
 
 start = time.time()
 hr_connection = None
-speed_cad_connection = []
+speed_cad_connections = []
+radio = None
 while True:
     if not hr_connection:
         print("Running hr_connection")
         hr_connection = pyloton.heart_connect()
         ble.stop_scan()
-    if not speed_cad_connection:
+    if not speed_cad_connections:
         print("Running speed_cad_connection")
-        speed_cad_connection = pyloton.speed_cad_connect()
+        speed_cad_connections = pyloton.speed_cad_connect()
 
     if time.time()-start >= 45:
         pyloton.timeout()
         break
     # Stop scanning whether or not we are connected.
     ble.stop_scan()
-    if hr_connection and hr_connection.connected and speed_cad_connection:
+    if hr_connection and hr_connection.connected and speed_cad_connections:
         print("Fetch connection")
         hr_service = hr_connection[HeartRateService]
         print("Location:", hr_service.location)

CircuitPython_Pyloton/pyloton.py

@@ -1,5 +1,7 @@
 import time
+import adafruit_ble
 from adafruit_ble.advertising.standard import ProvideServicesAdvertisement
+from adafruit_ble.advertising.standard import SolicitServicesAdvertisement
 import displayio
 import adafruit_imageload
 from adafruit_ble_cycling_speed_and_cadence import CyclingSpeedAndCadenceService
@@ -8,6 +10,8 @@
 from adafruit_display_shapes.rect import Rect
 from adafruit_display_text import label
 
+from adafruit_ble_apple_media import AppleMediaService
+
 
 class Pyloton:
 
@@ -148,6 +152,29 @@ def heart_connect(self):
                 break
         return self.hr_connection
 
+    def ams_connect(self):
+        self._status_update("Connect your phone now")
+        radio = adafruit_ble.BLERadio()
+        a = SolicitServicesAdvertisement()
+        a.solicited_services.append(AppleMediaService)
+        radio.start_advertising(a)
+
+        while not radio.connected:
+            pass
+
+        self._status_update("Connected")
+
+        known_notifications = set()
+
+        for connection in radio.connections:
+            if not connection.paired:
+                connection.pair()
+                self._status_update("paired")
+
+        self.ams = connection[AppleMediaService]
+
+        self.radio = radio
+
 
     def speed_cad_connect(self):
         """
@@ -246,11 +273,13 @@ def icon_maker(self, n, icon_x, icon_y):
         return sprite
 
 
-    def _label_maker(self, text, x, y):
+    def _label_maker(self, text, x, y, font=None):
         """
         Generates labels
         """
-        return label.Label(font=self.arial24, x=x, y=y, text=text, color=self.WHITE)
+        if not font:
+            font = self.arial24
+        return label.Label(font=font, x=x, y=y, text=text, color=self.WHITE)
 
 
     def _get_y(self):
@@ -338,13 +367,15 @@ def update_display(self, hr_service):
                 self.splash.append(cad_label)
 
         if self.ams_enabled:
-            ams_label = self._label_maker('None', 50, 210) # 210
+            ams_label = self._label_maker('{}'.format(self.ams.title), 50, 210, font=self.arial16) # 210
+
             if self.setup:
                 self.splash[6] = ams_label
             else:
                 self.splash.append(ams_label)
 
 
         self.setup=True
+        time.sleep(0.1)
 
         self.display.show(self.splash)