Merge pull request #27 from arduino/battery_charge_ctrl

added battery alert system

Author: gbr1

arduino_alvik/arduino_alvik.py

@@ -42,6 +42,7 @@ def __init__(self):
         self.right_led = self.DL2 = _ArduinoAlvikRgbLed(self._packeter, 'right', self._led_state,
                                                         rgb_mask=[0b00100000, 0b01000000, 0b10000000])
         self._battery_perc = None
+        self._battery_is_charging = None
         self._touch_byte = None
         self._move_byte = None
         self._behaviour = None
@@ -92,18 +93,20 @@ def is_on() -> bool:
         return CHECK_STM32.value() == 1
 
     @staticmethod
-    def _progress_bar(percentage: float) -> None:
+    def _print_battery_status(percentage: float, is_charging) -> None:
         """
-        Prints a progressbar
-        :param percentage:
+        Pretty prints the battery status
+        :param percentage: SOC of the battery
+        :param is_charging: True if the battery is charging
         :return:
         """
         sys.stdout.write(bytes('\r'.encode('utf-8')))
         if percentage > 97:
             marks_str = ' \U0001F50B'
         else:
             marks_str = ' \U0001FAAB'
-        word = marks_str + f" {percentage}% \t"
+        charging_str = f' \U0001F50C' if is_charging else ' \U000026A0WARNING: battery is discharging!'
+        word = marks_str + f" {percentage}%" + charging_str + " \t"
         sys.stdout.write(bytes((word.encode('utf-8'))))
 
     def _lenghty_op(self, iterations=10000000) -> int:
@@ -140,8 +143,9 @@ def _idle(self, delay_=1, check_on_thread=False, blocking=False) -> None:
 
                 soc_raw = struct.unpack('h', self.i2c.readfrom(0x36, 2))[0]
                 soc_perc = soc_raw * 0.00390625
-                self._battery_perc = soc_perc
-                self._progress_bar(round(soc_perc))
+                self._battery_is_charging = soc_perc > 0
+                self._battery_perc = abs(soc_perc)
+                self._print_battery_status(round(soc_perc), self._battery_is_charging)
                 if blocking:
                     self._lenghty_op(10000)
                 else:
@@ -219,6 +223,7 @@ def begin(self) -> int:
         self._snake_robot(2000)
         self.set_illuminator(True)
         self.set_behaviour(1)
+        self.set_behaviour(2)
         self._set_color_reference()
         if self._has_events_registered():
             print('Starting events thread')
@@ -671,7 +676,9 @@ def _parse_message(self) -> int:
             _, self._ax, self._ay, self._az, self._gx, self._gy, self._gz = self._packeter.unpacketC6F()
         elif code == ord('p'):
             # battery percentage
-            _, self._battery_perc = self._packeter.unpacketC1F()
+            _, battery_perc = self._packeter.unpacketC1F()
+            self._battery_is_charging = battery_perc > 0
+            self._battery_perc = abs(battery_perc)
         elif code == ord('d'):
             # distance sensor
             _, self._left_tof, self._center_tof, self._right_tof = self._packeter.unpacketC3I()
@@ -726,6 +733,13 @@ def get_battery_charge(self) -> int | None:
             return 100
         return round(self._battery_perc)
 
+    def is_battery_charging(self) -> bool:
+        """
+        Returns True if the device battery is charging
+        :return:
+        """
+        return self._battery_is_charging
+
     @property
     def _touch_bits(self) -> int:
         """