def attempt_login(cursor):
username, password = "", ""
# Try to get it from CGI, failing that try cookies
# Don't try to get it from CGI if it's mode=edit_user
if common_f.get_val('mode',"") != "edit_user":
username = common_f.get_val('username', "")
password = common_f.get_val('password', "")
from_cookie = False
# Cookies method
if username == "" and password == "":
username = html_f.get_cookie('profiteer_username', "")
password = html_f.get_cookie('profiteer_password', "")
from_cookie = True
# Still nothing?
if username == "" and password == "":
if os.environ.get('REMOTE_ADDR') == "::1" or os.environ.get('REMOTE_ADDR') == None:
u = common_q.get_one(cursor, User, id=1)
common_f.cache['user'] = u
return u
return ""
response = get_user(cursor, username, password, from_cookie)
if type(response) == User:
html_f.set_cookie("profiteer_username", username)
html_f.set_cookie("profiteer_password", response.password)
return response
def test_cookies(self):
self.test_targets.append(html_f.set_cookie)
self.test_targets.append(html_f.delete_cookie)
self.test_targets.append(html_f.get_cookie)
# Wipe the cookies to start with
html_f.cookies = cookies.SimpleCookie()
self.assertEqual(html_f.get_cookie("nonexistant_cookie"), None)
# Lets try setting one
html_f.set_cookie("imaginary_cookie", "math.sqrt(-1)")
self.assertEqual(html_f.get_cookie("imaginary_cookie"), "math.sqrt(-1)")
# Delete
html_f.delete_cookie("imaginary_cookie")
self.assertEqual(html_f.get_cookie("nonexistant_cookie"), None)
def logout(cursor):
html_f.set_cookie("profiteer_username", "")
html_f.set_cookie("profiteer_password", "")