Source code for system.version

#!/usr/bin/env python3
import os
import subprocess
from typing import List, Optional
from functools import lru_cache

from common.basedir import BASEDIR
from system.swaglog import cloudlog

TESTED_BRANCHES = ['devel', 'release3-staging', 'dashcam3-staging', 'release3', 'dashcam3']

training_version: bytes = b"0.2.0"
terms_version: bytes = b"2"

[docs]def cache(user_function, /): return lru_cache(maxsize=None)(user_function)
[docs]def run_cmd(cmd: List[str]) -> str: return subprocess.check_output(cmd, encoding='utf8').strip()
[docs]def run_cmd_default(cmd: List[str], default: Optional[str] = None) -> Optional[str]: try: return run_cmd(cmd) except subprocess.CalledProcessError: return default
[docs]@cache def get_commit(branch: str = "HEAD", default: Optional[str] = None) -> Optional[str]: return run_cmd_default(["git", "rev-parse", branch], default=default)
[docs]@cache def get_short_branch(default: Optional[str] = None) -> Optional[str]: return run_cmd_default(["git", "rev-parse", "--abbrev-ref", "HEAD"], default=default)
[docs]@cache def get_branch(default: Optional[str] = None) -> Optional[str]: return run_cmd_default(["git", "rev-parse", "--abbrev-ref", "--symbolic-full-name", "@{u}"], default=default)
[docs]@cache def get_origin(default: Optional[str] = None) -> Optional[str]: try: local_branch = run_cmd(["git", "name-rev", "--name-only", "HEAD"]) tracking_remote = run_cmd(["git", "config", "branch." + local_branch + ".remote"]) return run_cmd(["git", "config", "remote." + tracking_remote + ".url"]) except subprocess.CalledProcessError: # Not on a branch, fallback return run_cmd_default(["git", "config", "--get", "remote.origin.url"], default=default)
[docs]@cache def get_normalized_origin(default: Optional[str] = None) -> Optional[str]: origin: Optional[str] = get_origin() if origin is None: return default return origin.replace("git@", "", 1) \ .replace(".git", "", 1) \ .replace("https://", "", 1) \ .replace(":", "/", 1)
[docs]@cache def get_version() -> str: with open(os.path.join(BASEDIR, "common", "version.h")) as _versionf: version ='"')[1] return version
[docs]@cache def get_short_version() -> str: return get_version().split('-')[0] # type: ignore
[docs]@cache def is_prebuilt() -> bool: return os.path.exists(os.path.join(BASEDIR, 'prebuilt'))
[docs]@cache def is_comma_remote() -> bool: # note to fork maintainers, this is used for release metrics. please do not # touch this to get rid of the orange startup alert. there's better ways to do that origin: Optional[str] = get_origin() if origin is None: return False return origin.startswith('') or origin.startswith('')
[docs]@cache def is_tested_branch() -> bool: return get_short_branch() in TESTED_BRANCHES
[docs]@cache def is_dirty() -> bool: origin = get_origin() branch = get_branch() if (origin is None) or (branch is None): return True dirty = False try: # Actually check dirty files if not is_prebuilt(): # This is needed otherwise touched files might show up as modified try: subprocess.check_call(["git", "update-index", "--refresh"]) except subprocess.CalledProcessError: pass dirty = (["git", "diff-index", "--quiet", branch, "--"]) != 0) except subprocess.CalledProcessError: cloudlog.exception("git subprocess failed while checking dirty") dirty = True return dirty
if __name__ == "__main__": from common.params import Params params = Params() params.put("TermsVersion", terms_version) params.put("TrainingVersion", training_version) print(f"Dirty: {is_dirty()}") print(f"Version: {get_version()}") print(f"Short version: {get_short_version()}") print(f"Origin: {get_origin()}") print(f"Normalized origin: {get_normalized_origin()}") print(f"Branch: {get_branch()}") print(f"Short branch: {get_short_branch()}") print(f"Prebuilt: {is_prebuilt()}")