openpilot_comma/selfdrive/registration.py

import json
import subprocess

from selfdrive.swaglog import cloudlog
from selfdrive.version import version, training_version
from common.api import api_get
from common.params import Params

def get_imei():
  ret = subprocess.check_output(["getprop", "oem.device.imeicache"]).strip()
  if ret == "":
    ret = "000000000000000"
  return ret

def get_serial():
  return subprocess.check_output(["getprop", "ro.serialno"]).strip()

def get_git_commit():
  return subprocess.check_output(["git", "rev-parse", "HEAD"]).strip()

def get_git_branch():
  return subprocess.check_output(["git", "rev-parse", "--abbrev-ref", "HEAD"]).strip()

def get_git_remote():
  local_branch = subprocess.check_output(["git", "name-rev", "--name-only", "HEAD"]).strip()
  tracking_remote = subprocess.check_output(["git", "config", "branch."+local_branch+".remote"]).strip()
  return subprocess.check_output(["git", "config", "remote."+tracking_remote+".url"]).strip()

def register():
  params = Params()
  params.put("Version", version)
  params.put("TrainingVersion", training_version)
  params.put("GitCommit", get_git_commit())
  params.put("GitBranch", get_git_branch())
  params.put("GitRemote", get_git_remote())

  dongle_id, access_token = params.get("DongleId"), params.get("AccessToken")

  try:
    if dongle_id is None or access_token is None:
      cloudlog.info("getting pilotauth")
      resp = api_get("v1/pilotauth/", method='POST', timeout=15,
                     imei=get_imei(), serial=get_serial())
      dongleauth = json.loads(resp.text)
      dongle_id, access_token = dongleauth["dongle_id"].encode('ascii'), dongleauth["access_token"].encode('ascii')

      params.put("DongleId", dongle_id)
      params.put("AccessToken", access_token)
    return dongle_id, access_token
  except Exception:
    cloudlog.exception("failed to authenticate")
    return None

if __name__ == "__main__":
  print(api_get("").text)
  print(register())