jyoung/selfdrive/registration.py

import os
import json
import subprocess

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

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

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 register():
  params = Params()
  try:
    params.put("Version", version)
    params.put("GitCommit", get_git_commit())
    params.put("GitBranch", get_git_branch())

    dongle_id, access_token = params.get("DongleId"), params.get("AccessToken")
    if dongle_id is None or access_token is None:
      resp = api_get("v1/pilotauth/", method='POST', 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()