diff options
Diffstat (limited to 'test/getlogindetails.py')
| -rw-r--r-- | test/getlogindetails.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/test/getlogindetails.py b/test/getlogindetails.py new file mode 100644 index 0000000..2bd91ee --- /dev/null +++ b/test/getlogindetails.py @@ -0,0 +1,44 @@ +import os +import os.path +import getpass + +import freebase +from freebase.api.session import MetawebError + +passwordfile = "test/.password.txt" + +def main(create_password_file=False): + USERNAME, PASSWORD = "", "" + + if not os.path.isfile(passwordfile): + print "In order to run the tests, we need to use a valid freebase username and password" + USERNAME = raw_input("Please enter your username: ") + try: + PASSWORD = getpass.getpass("Please enter your password: ") + except getpass.GetPassWarning: + PASSWORD = raw_input("Please enter your password: ") + + freebase.login(USERNAME, PASSWORD) + + print "Thanks!" + + if create_password_file: + writepassword(passwordfile, USERNAME, PASSWORD) + + else: + pf = open(passwordfile, "r") + USERNAME, PASSWORD = pf.read().split("\n") + pf.close() + try: + freebase.login(USERNAME, PASSWORD) + except MetawebError, me: + print "The username/password in your .password.txt file are incorrect" + raise me + + return USERNAME, PASSWORD + + +def writepassword(passwordfile, username, password): + fh = open(passwordfile, "w") + fh.write(USERNAME + "\n" + PASSWORD) + fh.close()
\ No newline at end of file |