dragonpilot/selfdrive/version.py

#!/usr/bin/env python3
import os
import subprocess
from selfdrive.swaglog import cloudlog


def get_git_commit():
  return subprocess.check_output(["git", "rev-parse", "HEAD"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg


def get_git_branch():
  return subprocess.check_output(["git", "rev-parse", "--abbrev-ref", "HEAD"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg


def get_git_full_branchname():
  return subprocess.check_output(["git", "rev-parse", "--abbrev-ref", "--symbolic-full-name", "@{u}"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg


def get_git_remote():
  try:
    local_branch = subprocess.check_output(["git", "name-rev", "--name-only", "HEAD"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg
    tracking_remote = subprocess.check_output(["git", "config", "branch." + local_branch + ".remote"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg
    return subprocess.check_output(["git", "config", "remote." + tracking_remote + ".url"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg
  except subprocess.CalledProcessError:
    # Not on a branch, fallback
    return subprocess.check_output(["git", "config", "--get", "remote.origin.url"], encoding='utf8').strip()  # pylint: disable=unexpected-keyword-arg


with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), "common", "version.h")) as _versionf:
  version = _versionf.read().split('"')[1]

try:
  origin = get_git_remote()
  if origin.startswith('git@github.com:commaai') or origin.startswith('https://github.com/commaai'):
    if origin.endswith('/one.git'):
      dirty = True
    else:
      branch = get_git_full_branchname()

      # This is needed otherwise touched files might show up as modified
      try:
        subprocess.check_call(["git", "update-index", "--refresh"])
      except subprocess.CalledProcessError:
        pass

      dirty = subprocess.call(["git", "diff-index", "--quiet", branch, "--"]) != 0
      if dirty:
        dirty_files = subprocess.check_output(["git", "diff-index", branch, "--"], encoding='utf8')  # pylint: disable=unexpected-keyword-arg
        commit = subprocess.check_output(["git", "rev-parse", "--verify", "HEAD"], encoding='utf8').rstrip()  # pylint: disable=unexpected-keyword-arg
        origin_commit = subprocess.check_output(["git", "rev-parse", "--verify", branch], encoding='utf8').rstrip()  # pylint: disable=unexpected-keyword-arg
        cloudlog.event("dirty comma branch", vesion=version, dirty=dirty, origin=origin, branch=branch, dirty_files=dirty_files, commit=commit, origin_commit=origin_commit)

  else:
    dirty = True
except subprocess.CalledProcessError:
  try:
    cloudlog.exception("git subprocess failed while finding version")
  except:
    pass
  dirty = True

training_version = b"0.1.0"
terms_version = b"2"

if __name__ == "__main__":
  print("Dirty: %s" % dirty)
  print("Version: %s" % version)
  print("Remote: %s" % origin)

  try:
    print("Branch %s" % branch)
  except NameError:
    pass