import datetime import functools import re from typing import List, Optional from tools.lib.auth_config import get_token from tools.lib.api import CommaApi from tools.lib.helpers import RE, timestamp_to_datetime @functools.total_ordering class Bootlog: def __init__(self, url: str): self._url = url r = re.search(RE.BOOTLOG_NAME, url) if not r: raise Exception(f"Unable to parse: {url}") self._dongle_id = r.group('dongle_id') self._timestamp = r.group('timestamp') @property def url(self) -> str: return self._url @property def dongle_id(self) -> str: return self._dongle_id @property def timestamp(self) -> str: return self._timestamp @property def datetime(self) -> datetime.datetime: return timestamp_to_datetime(self._timestamp) def __str__(self): return f"{self._dongle_id}|{self._timestamp}" def __eq__(self, b) -> bool: if not isinstance(b, Bootlog): return False return self.datetime == b.datetime def __lt__(self, b) -> bool: if not isinstance(b, Bootlog): return False return self.datetime < b.datetime def get_bootlog_from_id(bootlog_id: str) -> Optional[Bootlog]: # TODO: implement an API endpoint for this bl = Bootlog(bootlog_id) for b in get_bootlogs(bl.dongle_id): if b == bl: return b return None def get_bootlogs(dongle_id: str) -> List[Bootlog]: api = CommaApi(get_token()) r = api.get(f'v1/devices/{dongle_id}/bootlogs') return [Bootlog(b) for b in r]