diff options
-rwxr-xr-x | extra/usb_updater/servo_updater.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/extra/usb_updater/servo_updater.py b/extra/usb_updater/servo_updater.py index 89f3f651ad..575f4dbbdb 100755 --- a/extra/usb_updater/servo_updater.py +++ b/extra/usb_updater/servo_updater.py @@ -251,8 +251,26 @@ def _extract_version(boardname, binfile): return newvers +def get_firmware_channel(bname, version): + """Find out which channel |version| for |bname| came from. -def get_files_and_version(cname, fname, channel=DEFAULT_CHANNEL): + Args: + bname: board name + version: current version string + + Returns: + one of the channel names if |version| came from one of those, or None + """ + for channel in CHANNELS: + # Pass |bname| as cname to find the board specific file, and pass None as + # fname to ensure the default directory is searched + _, _, vers = get_files_and_version(bname, None, channel=channel) + if version == vers: + return channel + # None of the channels matched. This firmware is currently unknown. + return None + +def get_files_and_version(cname, fname=None, channel=DEFAULT_CHANNEL): """Select config and firmware binary files. This checks default file names and paths. |