from edmc import companion as c
import platform
import easygui

companion_session = companion.Session()

def login(username, password):
  try:
    companion_session.login(username, password)
  except companion.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()