Weekly FW version check on RVR program startup

Author: Angel Campos

.gitignore

@@ -6,6 +6,7 @@
 /.idea
 /.python-version
 /docs/_build/
+/.fw
 *.ipynb_checkpoints
 *.h5
 *.py~

sphero_sdk/__init__.py

@@ -9,23 +9,23 @@
 from sphero_sdk.common.enums.sphero_rvr_enums import SpheroRvrLedBitmasks
 from sphero_sdk.common.rvr_streaming_services import RvrStreamingServices
 
+
 from sphero_sdk.common.enums.drive_enums import RawMotorModesEnum
 from sphero_sdk.common.enums.drive_enums import StabilizationIndexesEnum
-from sphero_sdk.common.enums.drive_enums import MotorIndexesEnum
+from sphero_sdk.common.enums.drive_enums import MotorIndexesEnum as DriveMotorIndexesEnum
 from sphero_sdk.common.enums.drive_enums import DriveFlagsBitmask
 from sphero_sdk.common.enums.io_enums import SpecdrumsColorPaletteIndiciesEnum
-from sphero_sdk.common.enums.io_enums import FrameInfoTypesEnum
 from sphero_sdk.common.enums.power_enums import BatteryVoltageStatesEnum
 from sphero_sdk.common.enums.power_enums import BatteryVoltageReadingTypesEnum
 from sphero_sdk.common.enums.power_enums import AmplifierIdsEnum
-from sphero_sdk.common.enums.sensor_enums import MotorIndexesEnum
+from sphero_sdk.common.enums.sensor_enums import MotorIndexesEnum as SensorMotorIndexesEnum
 from sphero_sdk.common.enums.sensor_enums import ThermalProtectionStatusEnum
 from sphero_sdk.common.enums.sensor_enums import StreamingDataSizesEnum
-from sphero_sdk.common.enums.sensor_enums import StreamingServiceStatesEnum
 from sphero_sdk.common.enums.sensor_enums import GyroMaxFlagsBitmask
 from sphero_sdk.common.enums.sensor_enums import LocatorFlagsBitmask
 from sphero_sdk.common.enums.sensor_enums import InfraredSensorLocationsBitmask
 
+from sphero_sdk.asyncio.client.firmware.rvr_fw_check_async import RvrFwCheckAsync
 from sphero_sdk.asyncio.controls.led_control_async import LedControlAsync
 from sphero_sdk.asyncio.controls.drive_control_async import DriveControlAsync
 from sphero_sdk.asyncio.controls.infrared_control_async import InfraredControlAsync
@@ -35,6 +35,7 @@
 from sphero_sdk.asyncio.client.dal.restful_async_dal import RestfulAsyncDal
 
 from sphero_sdk.observer.observer_base import Observer
+from sphero_sdk.observer.client.firmware.rvr_fw_check_observer import RvrFwCheckObserver
 from sphero_sdk.observer.events.event_dispatcher import EventDispatcher
 from sphero_sdk.observer.controls.led_control_observer import LedControlObserver
 from sphero_sdk.observer.controls.drive_control_observer import DriveControlObserver

sphero_sdk/asyncio/client/firmware/__init__.py

@@ -0,0 +1,57 @@
+import os
+import logging
+from sphero_sdk import SpheroRvrTargets
+from sphero_sdk.common.firmware.cms_fw_check_base import CmsFwCheckBase
+
+logger = logging.getLogger(__name__)
+
+
+class RvrFwCheckAsync(CmsFwCheckBase):
+
+    def __init__(self):
+        CmsFwCheckBase.__init__(self)
+        self.__rvr = self
+
+    async def _check_rvr_fw(self):
+        """Checks the RVR's firmware on the Nordic and ST chips against the CMS, if an internet connection is available.
+        """
+        root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../../../'))
+
+        # Have more than 7 days elapsed since last check?
+        if not self._should_run_fw_check():
+            return
+
+        print('Checking RVR firmware versions...')
+
+        rvr_nordic_version = await self.__rvr.get_main_application_version(target=SpheroRvrTargets.primary.value, timeout=3)
+        rvr_st_version = await self.__rvr.get_main_application_version(target=SpheroRvrTargets.secondary.value, timeout=3)
+
+        if rvr_nordic_version is None or rvr_st_version is None:
+            logger.error('Unable to retrieve Nordic and/or ST versions from RVR.')
+            return
+
+        print('Checking CMS firmware versions...')
+        cms_nordic_version = await self._get_cms_fw_version(self._nordic_cms_url)
+        logger.info('CMS Nordic Version:', cms_nordic_version)
+
+        cms_st_version = await self._get_cms_fw_version(self._st_cms_url)
+        logger.info('CMS ST Version:', cms_st_version)
+
+        # Proceed only if both versions are acquired from the CMS.
+        if cms_nordic_version is not None or cms_st_version is not None:
+
+            # Record timestamp of this check
+            self._write_timestamp()
+
+            warning_displayed = self._check_update_available(rvr_nordic_version, cms_nordic_version)
+
+            # Check ST version only if Nordic version check return false
+            if not warning_displayed:
+                self._check_update_available(rvr_st_version, cms_st_version)
+
+        print('Firmware check complete.')
+
+
+
+
+