From 0e7cfb28cb5373dd64da00ae0056e846a83f0287 Mon Sep 17 00:00:00 2001 From: nitromaster101 Date: Mon, 13 Jul 2009 17:15:30 +0000 Subject: added test utility git-svn-id: http://freebase-python.googlecode.com/svn/trunk@191 5914aa95-5b3a-0410-a3b5-7b719e7fe9b2 --- test/getlogindetails.py | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 test/getlogindetails.py (limited to 'test/getlogindetails.py') 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 -- cgit v1.3.1