Fix migration errors.

This commit is contained in:
Anna Rose 2016-01-21 14:44:23 -05:00
parent 0654796a1a
commit 588997e312

View File

@ -2,12 +2,12 @@ from edmc import companion as c
import platform import platform
import easygui import easygui
companion_session = companion.Session() companion_session = c.Session()
def login(username, password): def login(username, password):
try: try:
companion_session.login(username, password) companion_session.login(username, password)
except companion.VerificationRequired: except c.VerificationRequired:
if platform.system() == 'Windows': if platform.system() == 'Windows':
code = easygui.enterbox("Input Verification Code (check your email)", code = easygui.enterbox("Input Verification Code (check your email)",
"Verification Required") "Verification Required")