def __init__(self):
self.statusicon = StatusIcon.StatusIcon(self)
self.CredentialManager = Credentials.CredentialManager(self)
self.browser = Browser()
self.browser.addheaders = [("User-agent", USER_AGENT)]
self.logged = False
self.check_initial_state()
if current_system == "windows":
self.MainLoop = loop(self)
else:
self.MainLoop = loop()
class Manager:
def __init__(self):
self.statusicon = StatusIcon.StatusIcon(self)
self.CredentialManager = Credentials.CredentialManager(self)
self.browser = Browser()
self.browser.addheaders = [("User-agent", USER_AGENT)]
self.logged = False
self.check_initial_state()
if current_system == "windows":
self.MainLoop = loop(self)
else:
self.MainLoop = loop()
# when we get a signal from the wifi event loop we need to do some stuff
def on_signal(self, state):
if state == 0:
print "disconnected signal"
self.statusicon.set_visibility(False)
self.logged = False
elif state == 1:
print "connection established signal"
self.on_connection()
elif state == 2:
print "acquiring connection signal"
self.statusicon.set_blinking(True)
def on_password_change(self):
print "our stored password changed"
print self.CredentialManager.get_username()
print self.CredentialManager.get_password()
self.statusicon.set_blinking(True)
if self.wwu_auth():
self.logged = True
self.statusicon.set_blinking(False)
def on_connection(self):
print "on connection"
self.statusicon.set_blinking(False)
if is_wwu_wifi():
print "on wwu wifi"
self.statusicon.set_visibility(True)
if self.need_auth():
print "we need to auth"
self.statusicon.set_blinking(True)
if not self.wwu_auth():
print "we failed to auth"
# tries to log on if it failed this executes
self.CredentialManager.ask_user()
else:
print "we succeeded with auth"
self.logged = True
self.statusicon.set_blinking(False)
else:
print "we're already logged on"
self.logged = True
self.statusicon.set_blinking(False)
else:
print "not wwu wifi"
self.statusicon.set_visibility(False)
self.logged = False
def check_initial_state(self):
# we need to figure out whether we're already on a wifi network
# and if so whether it's a wwu wireless network
# and if so whether we need to log on
# this might eventually have some other stuff in it but right now we can
# just use the on_connection function
self.on_connection()
def need_auth(self):
response = self.browser.open("http://www.blankwebpage.com/")
return get_title(response.read()) == "Web Authentication"
def wwu_auth(self, statusicon="stat"):
self.browser.open(LOGIN_ADDRESS)
self.browser.select_form("MyForm")
self.browser.form.set_all_readonly(False)
self.browser["username"] = self.CredentialManager.get_username()
self.browser["password"] = self.CredentialManager.get_password()
self.browser["buttonClicked"] = "4"
self.browser.submit()
return not (self.need_auth())
def wwu_de_auth(self, statusicon="stat"):
self.statusicon.set_blinking(True)
response = self.browser.open(LOGOUT_ADDRESS)
self.browser.select_form(nr=0)
self.browser.submit()
def quit(self, statusicon="stat"):
self.MainLoop.quit()