#!/usr/bin/python

from edmc import companion
from inara.inara import InaraSession
import utils

settings = utils.get_settings()
companion_session = companion.Session()
inara_session = InaraSession(settings.get('inara', 'username'), settings.get('inara', 'password'))

try:
  companion_session.login(settings.get('ed_companion', 'username'), settings.get('ed_companion', 'password'))
except companion.VerificationRequired:
  code = raw_input("Input Verification Code: ")
  companion_session.verify(code)

data = companion_session.query()
inara_session.update_credits(data['commander']['credits'])
inara_session.update_location(data['lastSystem']['name'])

companion_session.close()