def main(self):
self.app = wx.App()
self.vpnconn = OpenVPNConnection(self.on_openvpn_connected)
self.icon = Icon(self)
self.set_state(STATE_DISCONNECTED)
self.connect()
self.app.MainLoop()
self.exit_event.wait()
sys.exit()
class Program(object):
def __init__(self):
self.open_files_on_connection = False
self.quiting = False
self.exit_event = threading.Event()
self.config = Configuration()
def set_state(self, state):
self.state = state
if not self.quiting:
self.icon.set_state(state)
def on_openvpn_connected(self):
if onWindows:
# For unknown reason, it can take quite some time before
# "explorer \\10.18.0.1" works properly after connecting.
# We poll using "net view \\SMB_HOSTNAME" until we're connected.
# TODO is there a way to speed this up?
while True:
l.info("calling 'net view'")
# WTF subprocess.Popen behaves differently than
# subprocess.call. This difference only occurs with
# the custom startupinfo.
pipe = subprocess.Popen(['net', 'view', '\\\\' + SMB_HOSTNAME],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
startupinfo=subprocess_sui)
out, err = pipe.communicate()
if pipe.returncode in (0, 2):
break
l.debug("returncode: %s", pipe.returncode)
l.debug("Message: %s", repr(err))
time.sleep(0.5)
if self.quiting:
return
self.set_state(STATE_CONNECTED)
if not onWindows and which('gvfs-mount') is not None:
subprocess.call(['gvfs-mount', 'smb://'+ SMB_HOSTNAME])
if self.open_files_on_connection:
self._show_files()
self.open_files_on_connection = False
def _vpn_worker_entry(self):
self.vpnconn.run()
self.set_state(STATE_DISCONNECTED)
def on_exit(self):
self.quiting = True
if self.state != STATE_DISCONNECTED:
self.set_state(STATE_UNKNOWN)
self.vpnconn.stop()
self.app.ExitMainLoop()
self.exit_event.set()
def connect(self):
assert self.state == STATE_DISCONNECTED
self.set_state(STATE_UNKNOWN)
if 'got-creds' not in self.config:
self.set_state(STATE_PROMPTING_CREDS)
LoginDialog(self._loginDialog_callback)
else:
self._start_vpn_worker()
def _loginDialog_callback(self, creds):
if creds is None:
self.set_state(STATE_DISCONNECTED)
return
self.set_state(STATE_CHECKING_CREDS)
try:
fetchCert.fetch(*creds)
self.config['got-creds'] = True
except fetchCert.AuthFailed:
self.set_state(STATE_PROMPTING_CREDS)
LoginDialog(self._loginDialog_callback,
message="De gebruikersnaam of "+
"het wachtwoord is verkeerd")
return
self.set_state(STATE_UNKNOWN)
self._start_vpn_worker()
def _start_vpn_worker(self):
self.vpn_worker = threading.Thread(target=self._vpn_worker_entry)
self.vpn_worker.start()
def main(self):
self.app = wx.App()
self.vpnconn = OpenVPNConnection(self.on_openvpn_connected)
self.icon = Icon(self)
self.set_state(STATE_DISCONNECTED)
self.connect()
self.app.MainLoop()
self.exit_event.wait()
sys.exit()
def show_files(self):
if self.state == STATE_CONNECTED:
self._show_files()
else:
self.open_files_on_connection = True
if self.state == STATE_DISCONNECTED:
self.connect()
def _show_files(self):
if onWindows:
l.info("calling explorer")
# WTF adding startupinfo breaks this.
subprocess.call(['explorer', '\\\\' + SMB_HOSTNAME])
else:
attempts = [['gnome-open', 'smb://'+ SMB_HOSTNAME], ['nautilus', 'smb://'+ SMB_HOSTNAME]]
for args in attempts:
exepath = which(args[0])
if exepath is not None:
l.info("Starting %s", exepath)
subprocess.call([exepath] + args[1:])
return
l.warning("Could not found filebrowser")
def toggle_connection(self):
if self.state == STATE_CONNECTED:
self.set_state(STATE_UNKNOWN)
self.vpnconn.stop()
elif self.state == STATE_DISCONNECTED:
self.connect()