diff --git a/companion.py b/companion.py new file mode 100644 index 0000000..961d46e --- /dev/null +++ b/companion.py @@ -0,0 +1,20 @@ +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()