wifiradar/gui/g2/__init__.py

   1 #!/usr/bin/python
   2 # -*- coding: utf-8 -*-
   3 #
   4 #   gui/g2/__init__.py - collection of classes for main UI with PyGTK
   5 #
   6 #   Part of WiFi Radar: A utility for managing WiFi profiles on GNU/Linux.
   7 #
   8 #   Copyright (C) 2004-2005 Ahmad Baitalmal <ahmad@baitalmal.com>
   9 #   Copyright (C) 2005 Nicolas Brouard <nicolas.brouard@mandrake.org>
  10 #   Copyright (C) 2005-2009 Brian Elliott Finley <brian@thefinleys.com>
  11 #   Copyright (C) 2006 David Decotigny <com.d2@free.fr>
  12 #   Copyright (C) 2006 Simon Gerber <gesimu@gmail.com>
  13 #   Copyright (C) 2006-2007 Joey Hurst <jhurst@lucubrate.org>
  14 #   Copyright (C) 2012 Anari Jalakas <anari.jalakas@gmail.com>
  15 #   Copyright (C) 2006, 2009 Ante Karamatic <ivoks@ubuntu.com>
  16 #   Copyright (C) 2009-2010,2014 Sean Robinson <seankrobinson@gmail.com>
  17 #   Copyright (C) 2010 Prokhor Shuchalov <p@shuchalov.ru>
  18 #
  19 #   This program is free software; you can redistribute it and/or modify
  20 #   it under the terms of the GNU General Public License as published by
  21 #   the Free Software Foundation; version 2 of the License.
  22 #
  23 #   This program is distributed in the hope that it will be useful,
  24 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
  25 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  26 #   GNU General Public License in LICENSE.GPL for more details.
  27 #
  28 #   You should have received a copy of the GNU General Public License
  29 #   along with this program; if not, write to the Free Software
  30 #   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  31 #
  32
  33
  34 import errno
  35 import logging
  36 import sys
  37
  38 import glib
  39 import gtk
  40
  41 from wifiradar.config import make_section_name
  42 import wifiradar.connections as connections
  43 import wifiradar.misc as misc
  44 from wifiradar.pubsub import Message
  45 from . import prefs
  46 from . import profile as profile_ed
  47 from . import transients
  48
  49 # create a logger
  50 logger = logging.getLogger(__name__)
  51
  52
  53 # Create a bunch of icons from files in the package.
  54 known_profile_icon = gtk.gdk.pixbuf_new_from_file("pixmaps/known_profile.png")
  55 unknown_profile_icon = gtk.gdk.pixbuf_new_from_file("pixmaps/unknown_profile.png")
  56 signal_none_pb = gtk.gdk.pixbuf_new_from_file("pixmaps/signal_none.xpm")
  57 signal_low_pb = gtk.gdk.pixbuf_new_from_file("pixmaps/signal_low.xpm")
  58 signal_barely_pb = gtk.gdk.pixbuf_new_from_file("pixmaps/signal_barely.xpm")
  59 signal_ok_pb = gtk.gdk.pixbuf_new_from_file("pixmaps/signal_ok.xpm")
  60 signal_best_pb = gtk.gdk.pixbuf_new_from_file("pixmaps/signal_best.xpm")
  61
  62 def pixbuf_from_known(known):
  63     """ Return a :class:`gtk.gdk.Pixbuf` icon to represent :data:`known`.
  64         Any true :data:`known` value returns the icon showing previous
  65         familiarity.
  66     """
  67     if known:
  68         return known_profile_icon
  69     return unknown_profile_icon
  70
  71 def pixbuf_from_signal(signal):
  72     """ Return a :class:`gtk.gdk.Pixbuf` icon to indicate the :data:`signal`
  73         level.  :data:`signal` is as reported by iwlist (may be arbitrary
  74         scale in 0-100 or -X dBm)
  75     """
  76     signal = int(signal)
  77     # Shift signal up by 80 to convert dBm scale to arbitrary scale.
  78     if signal < 0:
  79         signal = signal + 80
  80     # Find an icon...
  81     if signal < 3:
  82         return signal_none_pb
  83     elif signal < 12:
  84         return signal_low_pb
  85     elif signal < 20:
  86         return signal_barely_pb
  87     elif signal < 35:
  88         return signal_ok_pb
  89     elif signal >= 35:
  90         return signal_best_pb
  91
  92
  93 class RadarWindow:
  94     def __init__(self, msg_pipe):
  95         """ Create a new RadarWindow wanting to communicate through
  96             :data:`msg_pipe`, a :class:`multiprocessing.Connection`.
  97         """
  98         self.msg_pipe = msg_pipe
  99
 100         gtk.gdk.threads_init()
 101
 102         self.icon = gtk.gdk.pixbuf_new_from_file("pixmaps/wifi-radar.png")
 103
 104         self.window = gtk.Dialog('WiFi Radar', None, gtk.DIALOG_MODAL)
 105         self.window.set_icon(self.icon)
 106         self.window.set_border_width(10)
 107         self.window.set_size_request(550, 300)
 108         self.window.set_title("WiFi Radar")
 109         self.window.connect('delete_event', self.delete_event)
 110         # let's create all our widgets
 111         self.current_network = gtk.Label()
 112         self.current_network.set_property('justify', gtk.JUSTIFY_CENTER)
 113         self.current_network.show()
 114         self.close_button = gtk.Button("Close", gtk.STOCK_CLOSE)
 115         self.close_button.show()
 116         self.close_button.connect('clicked', self.delete_event, None)
 117         self.about_button = gtk.Button("About", gtk.STOCK_ABOUT)
 118         self.about_button.show()
 119         self.about_button.connect('clicked', self.show_about_info, None)
 120         self.preferences_button = gtk.Button("Preferences", gtk.STOCK_PREFERENCES)
 121         self.preferences_button.show()
 122         self.preferences_button.connect('clicked', self.request_preferences_edit, None)
 123         #                           essid  bssid  known_icon      known  available  wep_icon  signal_level    mode  protocol  channel
 124         self.pstore = gtk.ListStore(str,   str,   gtk.gdk.Pixbuf, bool,  bool,      str,      gtk.gdk.Pixbuf, str,  str,      str)
 125         self.plist = gtk.TreeView(self.pstore)
 126         # The icons column, known and encryption
 127         self.pix_cell = gtk.CellRendererPixbuf()
 128         self.wep_cell = gtk.CellRendererPixbuf()
 129         self.icons_cell = gtk.CellRendererText()
 130         self.icons_col = gtk.TreeViewColumn()
 131         self.icons_col.pack_start(self.pix_cell, False)
 132         self.icons_col.pack_start(self.wep_cell, False)
 133         self.icons_col.add_attribute(self.pix_cell, 'pixbuf', 2)
 134         self.icons_col.add_attribute(self.wep_cell, 'stock-id', 5)
 135         self.plist.append_column(self.icons_col)
 136         # The AP column
 137         self.ap_cell = gtk.CellRendererText()
 138         self.ap_col = gtk.TreeViewColumn("Access Point")
 139         self.ap_col.pack_start(self.ap_cell, True)
 140         self.ap_col.set_cell_data_func(self.ap_cell, self._set_ap_col_value)
 141         self.plist.append_column(self.ap_col)
 142         # The signal column
 143         self.sig_cell = gtk.CellRendererPixbuf()
 144         self.signal_col = gtk.TreeViewColumn("Signal")
 145         self.signal_col.pack_start(self.sig_cell, True)
 146         self.signal_col.add_attribute(self.sig_cell, 'pixbuf', 6)
 147         self.plist.append_column(self.signal_col)
 148         # The mode column
 149         self.mode_cell = gtk.CellRendererText()
 150         self.mode_col = gtk.TreeViewColumn("Mode")
 151         self.mode_col.pack_start(self.mode_cell, True)
 152         self.mode_col.add_attribute(self.mode_cell, 'text', 7)
 153         self.plist.append_column(self.mode_col)
 154         # The protocol column
 155         self.prot_cell = gtk.CellRendererText()
 156         self.protocol_col = gtk.TreeViewColumn("802.11")
 157         self.protocol_col.pack_start(self.prot_cell, True)
 158         self.protocol_col.add_attribute(self.prot_cell, 'text', 8)
 159         self.plist.append_column(self.protocol_col)
 160         # The channel column
 161         self.channel_cell = gtk.CellRendererText()
 162         self.channel_col = gtk.TreeViewColumn("Channel")
 163         self.channel_col.pack_start(self.channel_cell, True)
 164         self.channel_col.add_attribute(self.channel_cell, 'text', 9)
 165         self.plist.append_column(self.channel_col)
 166         # DnD Ordering
 167         self.plist.set_reorderable(True)
 168         # detect d-n-d of AP in round-about way, since rows-reordered does not work as advertised
 169         self.pstore.connect('row-deleted', self.update_auto_profile_order)
 170         # enable/disable buttons based on the selected network
 171         self.selected_network = self.plist.get_selection()
 172         self.selected_network.connect('changed', self.on_network_selection, None)
 173         # the list scroll bar
 174         sb = gtk.VScrollbar(self.plist.get_vadjustment())
 175         sb.show()
 176         self.plist.show()
 177         # Add New button
 178         self.new_button = gtk.Button("_New")
 179         self.new_button.connect('clicked', self.create_new_profile, misc.get_new_profile())
 180         self.new_button.show()
 181         # Add Configure button
 182         self.edit_button = gtk.Button("C_onfigure")
 183         self.edit_button.connect('clicked', self.request_profile_edit, None)
 184         self.edit_button.show()
 185         self.edit_button.set_sensitive(False)
 186         # Add Delete button
 187         self.delete_button = gtk.Button("_Delete")
 188         self.delete_button.connect('clicked', self.delete_profile_with_check, None)
 189         self.delete_button.show()
 190         self.delete_button.set_sensitive(False)
 191         # Add Connect button
 192         self.connect_button = gtk.Button("Co_nnect")
 193         self.connect_button.connect('clicked', self.connect_profile, None)
 194         # Add Disconnect button
 195         self.disconnect_button = gtk.Button("D_isconnect")
 196         self.disconnect_button.connect('clicked', self.disconnect_profile, None)
 197         # lets add our widgets
 198         rows = gtk.VBox(False, 3)
 199         net_list = gtk.HBox(False, 0)
 200         listcols = gtk.HBox(False, 0)
 201         prows = gtk.VBox(False, 0)
 202         # lets start packing
 203         # the network list
 204         net_list.pack_start(self.plist, True, True, 0)
 205         net_list.pack_start(sb, False, False, 0)
 206         # the rows level
 207         rows.pack_start(net_list , True, True, 0)
 208         rows.pack_start(self.current_network, False, True, 0)
 209         # the list columns
 210         listcols.pack_start(rows, True, True, 0)
 211         listcols.pack_start(prows, False, False, 5)
 212         # the list buttons
 213         prows.pack_start(self.new_button, False, False, 2)
 214         prows.pack_start(self.edit_button, False, False, 2)
 215         prows.pack_start(self.delete_button, False, False, 2)
 216         prows.pack_end(self.connect_button, False, False, 2)
 217         prows.pack_end(self.disconnect_button, False, False, 2)
 218
 219         self.window.action_area.pack_start(self.about_button)
 220         self.window.action_area.pack_start(self.preferences_button)
 221         self.window.action_area.pack_start(self.close_button)
 222
 223         rows.show()
 224         prows.show()
 225         listcols.show()
 226         self.window.vbox.add(listcols)
 227         self.window.vbox.set_spacing(3)
 228         self.window.show_all()
 229         #
 230         # Now, immediately hide these two.  The proper one will be
 231         # displayed later, based on interface state. -BEF-
 232         self.disconnect_button.hide()
 233         self.connect_button.hide()
 234         self.connect_button.set_sensitive(False)
 235
 236         # set up status window for later use
 237         self.status_window = transients.StatusWindow(self)
 238         self.status_window.cancel_button.connect('clicked', self.disconnect_profile, "cancel")
 239
 240         self._running = True
 241         # Check for incoming messages every 25 ms, a.k.a. 40 Hz.
 242         glib.timeout_add(25, self.run)
 243
 244         with gtk.gdk.lock:
 245             gtk.main()
 246
 247     def run(self):
 248         """ Watch for incoming messages.
 249         """
 250         if self.msg_pipe.poll():
 251             try:
 252                 msg = self.msg_pipe.recv()
 253             except (EOFError, IOError) as e:
 254                 # This is bad, really bad.
 255                 logger.critical('read on closed ' +
 256                     'Pipe ({}), failing...'.format(rfd))
 257                 raise misc.PipeError(e)
 258             else:
 259                 self._check_message(msg)
 260         return self._running
 261
 262     def _check_message(self, msg):
 263         """ Process incoming messages.
 264         """
 265         if msg.topic == 'EXIT':
 266             self.delete_event()
 267         elif msg.topic == 'CONFIG-UPDATE':
 268             # Replace configuration manager with the one in msg.details.
 269             self.config = msg.details
 270         elif msg.topic == 'PROFILE-EDIT':
 271             with gtk.gdk.lock:
 272                 self.edit_profile(msg.details)
 273         elif msg.topic == 'PROFILE-UPDATE':
 274             with gtk.gdk.lock:
 275                 self.update_plist_items(msg.details)
 276         elif msg.topic == 'PROFILE-UNLIST':
 277             with gtk.gdk.lock:
 278                 self.remove_profile(msg.details)
 279         elif msg.topic == 'PROFILE-MOVE':
 280             new_position, profile = msg.details
 281             with gtk.gdk.lock:
 282                 if profile['roaming']:
 283                     old_position = self.get_row_by_ap(profile['essid'])
 284                 else:
 285                     old_position = self.get_row_by_ap(profile['essid'],
 286                         profile['bssid'])
 287                 self.pstore.move_before(old_position, self.pstore[new_position].iter)
 288         elif msg.topic == 'PREFS-EDIT':
 289             with gtk.gdk.lock:
 290                 self.edit_preferences(msg.details)
 291         elif msg.topic == 'ERROR':
 292             with gtk.gdk.lock:
 293                 error_dlg = transients.ErrorDialog(self.window, msg.details)
 294                 del error_dlg
 295         else:
 296             logger.warning('unrecognized Message: "{}"'.format(msg))
 297
 298     def destroy(self, widget=None):
 299         """ Quit the Gtk event loop.  :data:`widget` is the widget
 300             sending the signal, but it is ignored.
 301         """
 302         if self.status_window:
 303             self.status_window.destroy()
 304         gtk.main_quit()
 305
 306     def delete_event(self, widget=None, data=None):
 307         """ Shutdown the application.  :data:`widget` is the widget sending
 308             the signal and :data:`data` is a list of arbitrary arguments,
 309             both are ignored.  Always returns False to not propigate the
 310             signal which called :func:`delete_event`.
 311         """
 312         self._running = False
 313         self.msg_pipe.send(Message('EXIT', ''))
 314         self.msg_pipe.close()
 315         self.window.hide()
 316         # process GTK events so that window hides more quickly
 317         if sys.modules.has_key("gtk"):
 318             while gtk.events_pending():
 319                 gtk.main_iteration(False)
 320         self.destroy()
 321         return False
 322
 323     def update_network_info(self, profile=None, ip=None):
 324         """ Update the current ip and essid shown to the user.
 325         """
 326         if (profile is None) and (ip is None):
 327             self.current_network.set_text("Not Connected.")
 328         else:
 329             self.current_network.set_text('Connected to {}\nIP Address {}'.format(profile, ip))
 330
 331     def update_connect_buttons(self, connected=False):
 332         """ Set the state of connect/disconnect buttons to reflect the
 333             current connected state.
 334         """
 335         if connected:
 336             self.connect_button.hide()
 337             self.disconnect_button.show()
 338         else:
 339             self.disconnect_button.hide()
 340             self.connect_button.show()
 341
 342     def update_plist_items(self, profile):
 343         """ Updates the display of :data:`profile`.
 344         """
 345         if profile['roaming']:
 346             prow_iter = self.get_row_by_ap(profile['essid'])
 347         else:
 348             prow_iter = self.get_row_by_ap(profile['essid'], profile['bssid'])
 349
 350         if prow_iter is None:
 351             # the AP is not in the list of APs on the screen
 352             self.add_profile(profile)
 353         else:
 354             # the AP is in the list of APs on the screen
 355             wep = None
 356             if profile['encrypted']:
 357                 wep = gtk.STOCK_DIALOG_AUTHENTICATION
 358             # Update the Gtk objects.
 359             self.pstore.set_value(prow_iter, 2, pixbuf_from_known(profile['known']))
 360             self.pstore.set_value(prow_iter, 3, profile['known'])
 361             self.pstore.set_value(prow_iter, 4, profile['available'])
 362             self.pstore.set_value(prow_iter, 5, wep)
 363             self.pstore.set_value(prow_iter, 6, pixbuf_from_signal(profile['signal']))
 364             self.pstore.set_value(prow_iter, 7, profile['mode'])
 365             self.pstore.set_value(prow_iter, 8, profile['protocol'])
 366             self.pstore.set_value(prow_iter, 9, profile['channel'])
 367
 368     def _set_ap_col_value(self, column, cell, model, iter):
 369         """ Set the text attribute of :data:`column` to the first two
 370             :data:`model` values joined by a newline.  This is for
 371             displaying the :data:`essid` and :data:`bssid` in a single
 372             cell column.
 373         """
 374         essid = model.get_value(iter, 0)
 375         bssid = model.get_value(iter, 1)
 376         cell.set_property('text', '\n'.join([essid, bssid]))
 377
 378     def get_row_by_ap(self, essid, bssid='  Multiple APs'):
 379         """ Returns a :class:`gtk.TreeIter` for the row which holds
 380             :data:`essid` and :data:`bssid`.
 381
 382             :data:`bssid` is optional.  If not given, :func:`get_row_by_ap`
 383             will try to match a roaming profile with the given :data:`essid`.
 384
 385             If no match is found, it returns None.
 386         """
 387         for row in self.pstore:
 388             if (row[0] == essid) and (row[1] == bssid):
 389                 return row.iter
 390         return None
 391
 392     def on_network_selection(self, widget=None, data=None):
 393         """ Enable/disable buttons based on the selected network.
 394             :data:`widget` is the widget sending the signal and :data:`data`
 395             is a list of arbitrary arguments, both are ignored.
 396         """
 397         store, selected_iter = self.selected_network.get_selected()
 398         if selected_iter is None:
 399             # No row is selected, disable all buttons except New.
 400             # This occurs after a drag-and-drop.
 401             self.edit_button.set_sensitive(False)
 402             self.delete_button.set_sensitive(False)
 403             self.connect_button.set_sensitive(False)
 404         else:
 405             # One row is selected, so enable or disable buttons.
 406             self.connect_button.set_sensitive(True)
 407             if store.get_value(selected_iter, 3):
 408                 # Known profile.
 409                 self.edit_button.set_sensitive(True)
 410                 self.delete_button.set_sensitive(True)
 411             else:
 412                 # Unknown profile.
 413                 self.edit_button.set_sensitive(True)
 414                 self.delete_button.set_sensitive(False)
 415
 416     def show_about_info(self, widget=None, data=None):
 417         """ Handle the life-cycle of the About dialog.  :data:`widget` is
 418             the widget sending the signal and :data:`data` is a list of
 419             arbitrary arguments, both are ignored.
 420         """
 421         about = transients.AboutDialog()
 422         about.run()
 423         about.destroy()
 424
 425     def request_preferences_edit(self, widget=None, data=None):
 426         """ Respond to a request to edit the application preferences.
 427             :data:`widget` is the widget sending the signal and :data:`data`
 428             is a list of arbitrary arguments, both are ignored.
 429         """
 430         self.msg_pipe.send(Message('PREFS-EDIT-REQUEST', ''))
 431
 432     def edit_preferences(self, config):
 433         """ Allow the user to edit :data:`config`.
 434         """
 435         # get raw strings from config file
 436         config.raw = True
 437         prefs_editor = prefs.PreferencesEditor(self, config)
 438         response = prefs_editor.run()
 439         if response == gtk.RESPONSE_ACCEPT:
 440             prefs_editor.save()
 441         prefs_editor.destroy()
 442         # get cooked strings from config file
 443         config.raw = False
 444         self.msg_pipe.send(Message('PREFS-UPDATE', config))
 445
 446     def add_profile(self, profile):
 447         """ Add :data:`profile` to the list of APs shown to the user.
 448         """
 449         if profile['roaming']:
 450             profile['bssid'] = '  Multiple APs'
 451
 452         wep = None
 453         if profile['encrypted']:
 454             wep = gtk.STOCK_DIALOG_AUTHENTICATION
 455
 456         self.pstore.append([profile['essid'], profile['bssid'],
 457             known_profile_icon, profile['known'], profile['available'],
 458             wep, signal_none_pb, profile['mode'], profile['protocol'],
 459             profile['channel']])
 460
 461     def remove_profile(self, profile):
 462         """ Remove :data:`profile` from the list of APs shown to the user.
 463         """
 464         if profile['roaming']:
 465             prow_iter = self.get_row_by_ap(profile['essid'])
 466         else:
 467             prow_iter = self.get_row_by_ap(profile['essid'], profile['bssid'])
 468         if prow_iter is not None:
 469                 self.pstore.remove(prow_iter)
 470
 471     def create_new_profile(self, widget=None, profile=None, data=None):
 472         """ Respond to a user request to create a new AP profile.
 473             :data:`widget` is the widget sending the signal.  :data:profile`
 474             is an AP profile to use as the basis for the new profile.  It
 475             is likely empty or mostly empty.  :data:`data` is a list of
 476             arbitrary arguments.  :data:`widget` and "data"`data` are both
 477             ignored.
 478
 479             The order of parameters is important.  Because when this method
 480             is called from a signal handler, :data:`widget` is always the
 481             first argument.
 482         """
 483         profile_editor = profile_ed.ProfileEditor(self, profile)
 484         try:
 485             profile = profile_editor.run()
 486         except ValueError:
 487             self.msg_pipe.send(Message('ERROR', 'Cannot save empty ESSID'))
 488         else:
 489             if profile:
 490                 self.msg_pipe.send(Message('PROFILE-EDITED', profile))
 491         finally:
 492             profile_editor.destroy()
 493
 494     def request_profile_edit(self, widget=None, data=None):
 495         """ Respond to a request to edit an AP profile.  :data:`widget`
 496             is the widget sending the signal and :data:`data` is a list
 497             of arbitrary arguments, both are ignored.
 498         """
 499         store, selected_iter = self.plist.get_selection().get_selected()
 500         if selected_iter is not None:
 501             essid = self.pstore.get_value(selected_iter, 0)
 502             bssid = self.pstore.get_value(selected_iter, 1)
 503             if bssid == '  Multiple APs':
 504                 # AP list says this is a roaming profile
 505                 bssid = ''
 506             self.msg_pipe.send(Message('PROFILE-EDIT-REQUEST', (essid, bssid)))
 507
 508     def edit_profile(self, profile):
 509         """ Allow the user to edit :data:`profile`.
 510         """
 511         profile_editor = profile_ed.ProfileEditor(self, profile)
 512         edited_profile = profile_editor.run()
 513         profile_editor.destroy()
 514
 515         if edited_profile is not None:
 516             # Replace old profile.
 517             self.msg_pipe.send(Message('PROFILE-EDITED',
 518                 (edited_profile, profile)))
 519
 520     def delete_profile(self, apname):
 521         """ Delete the profile associated with :data:`apname` (i.e. make
 522             the profile unknown).
 523         """
 524         self.msg_pipe.send(Message('PROFILE-REMOVE', apname))
 525
 526     def delete_profile_with_check(self, widget=None, data=None):
 527         """ Respond to a request to delete an AP profile (i.e. make the
 528             profile unknown)  Check with the user first.  :data:`widget`
 529             is the widget sending the signal and :data:`data` is a list of
 530             arbitrary arguments, both are ignored.
 531         """
 532         store, selected_iter = self.plist.get_selection().get_selected()
 533         if selected_iter is None:
 534             return
 535
 536         essid = self.pstore.get_value(selected_iter, 0)
 537         bssid = self.pstore.get_value(selected_iter, 1)
 538         if bssid == '  Multiple APs':
 539             apname = make_section_name(essid, '')
 540         else:
 541             apname = make_section_name(essid, bssid)
 542         profile = self.config.get_profile(apname)
 543         if profile['roaming']:
 544             dlg = gtk.MessageDialog(self.window, gtk.DIALOG_DESTROY_WITH_PARENT | gtk.DIALOG_MODAL, gtk.MESSAGE_QUESTION, gtk.BUTTONS_YES_NO, "Are you sure you want to delete the %s profile?" % (essid, ))
 545         else:
 546             dlg = gtk.MessageDialog(self.window, gtk.DIALOG_DESTROY_WITH_PARENT | gtk.DIALOG_MODAL, gtk.MESSAGE_QUESTION, gtk.BUTTONS_YES_NO, "Are you sure you want to delete the %s (%s) profile?" % (essid, bssid))
 547         known = store.get_value(selected_iter, 3)
 548         if not known: return
 549         res = dlg.run()
 550         dlg.destroy()
 551         del dlg
 552         if res == gtk.RESPONSE_NO:
 553             return
 554         self.delete_profile(apname)
 555
 556     def connect_profile(self, widget, profile, data=None):
 557         """ Respond to a request to connect to an AP.
 558
 559         Parameters:
 560
 561             'widget' -- gtk.Widget - The widget sending the event.
 562
 563             'profile' -- dictionary - The AP profile to which to connect.
 564
 565             'data' -- tuple - list of arbitrary arguments (not used)
 566
 567         Returns:
 568
 569             nothing
 570         """
 571         store, selected_iter = self.plist.get_selection().get_selected()
 572         if selected_iter is None:
 573             return
 574         essid = self.pstore.get_value(selected_iter, 0)
 575         bssid = self.pstore.get_value(selected_iter, 1)
 576         known = store.get_value(selected_iter, 3)
 577         if not known:
 578             dlg = gtk.MessageDialog(self.window, gtk.DIALOG_DESTROY_WITH_PARENT | gtk.DIALOG_MODAL, gtk.MESSAGE_QUESTION, gtk.BUTTONS_YES_NO, "This network does not have a profile configured.\n\nWould you like to create one now?")
 579             res = dlg.run()
 580             dlg.destroy()
 581             del dlg
 582             if res == gtk.RESPONSE_NO:
 583                 return
 584             profile = misc.get_new_profile()
 585             profile['essid'] = essid
 586             profile['bssid'] = bssid
 587             if not self.create_new_profile(widget, profile, data):
 588                 return
 589         else:
 590             # Check for roaming profile.
 591             ap_name = make_section_name(essid, '')
 592             profile = self.config.get_profile(ap_name)
 593             if not profile:
 594                 # Check for normal profile.
 595                 ap_name = make_section_name(essid, bssid)
 596                 profile = self.config.get_profile(ap_name)
 597                 if not profile:
 598                     # No configured profile
 599                     return
 600             profile['bssid'] = self.access_points[ap_name]['bssid']
 601             profile['channel'] = self.access_points[ap_name]['channel']
 602         self.msg_pipe.send(Message('CONNECT', profile))
 603
 604     def disconnect_profile(self, widget=None, data=None):
 605         """ Respond to a request to disconnect by sending a message to
 606             ConnectionManager.  :data:`widget` is the widget sending the
 607             signal and :data:`data` is a list of arbitrary arguments, both
 608             are ignored.
 609         """
 610         self.msg_pipe.send(Message('DISCONNECT', ''))
 611         if data == "cancel":
 612             self.status_window.update_message("Canceling connection...")
 613             if sys.modules.has_key("gtk"):
 614                 while gtk.events_pending():
 615                     gtk.main_iteration(False)
 616
 617     def profile_order_updater(self, model, path, iter, auto_profile_order):
 618         """
 619         """
 620         if model.get_value(iter, 3) is True:
 621             essid = self.pstore.get_value(iter, 0)
 622             bssid = self.pstore.get_value(iter, 1)
 623             if bssid == '  Multiple APs':
 624                 bssid = ''
 625             apname = make_section_name(essid, bssid)
 626             auto_profile_order.append(apname)
 627
 628     def update_auto_profile_order(self, widget=None, data=None, data2=None):
 629         """ Update the config file auto profile order from the on-screen
 630             order.  :data:`widget` is the widget sending the signal and
 631             :data:`data` and :data:`data2` is a list of arbitrary arguments,
 632             all are ignored.
 633         """
 634         # recreate the auto_profile_order
 635         auto_profile_order = []
 636         self.pstore.foreach(self.profile_order_updater, auto_profile_order)
 637         self.msg_pipe.send(Message('PROFILE-ORDER-UPDATE', auto_profile_order))
 638         # Request saving the configuration file after the update.
 639         self.msg_pipe.send(Message('CONFIG-SAVE', ''))
 640
 641
 642 # Make so we can be imported
 643 if __name__ == "__main__":
 644     pass
 645