elite_api/companion.py

import easygui
from edmc import companion as c
import platform

companion_session = c.Session()

def login(username, password):
  try:
    companion_session.login(username, password)
  except c.VerificationRequired:
    if platform.system() == 'Windows':
      code = easygui.enterbox("Input Verification Code (check your email)",
                              "Verification Required")
    else:
      code = raw_input("Input Verification Code (check your email): ")
      companion_session.verify(code)


def get_data():
  return companion_session.query()