850 lines
24 KiB
Python
850 lines
24 KiB
Python
# vi: ts=4 ai noet
|
|
#
|
|
# Copyright (C) 2006, Red Hat, Inc.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
import dbus
|
|
import dbus.glib
|
|
import dbus.decorators
|
|
import gobject
|
|
import gtk
|
|
import logging
|
|
import os
|
|
from gettext import gettext as _
|
|
|
|
import hippo
|
|
from sugar.graphics.menu import Menu
|
|
from sugar.graphics import style
|
|
from sugar.graphics.iconcolor import IconColor
|
|
from sugar.graphics.timeline import Timeline
|
|
from bubble import Bubble
|
|
|
|
import nminfo
|
|
|
|
|
|
NM_DEVICE_STAGE_STRINGS=("Unknown",
|
|
"Prepare",
|
|
"Config",
|
|
"Need Users Key",
|
|
"IP Config",
|
|
"IP Config Get",
|
|
"IP Config Commit",
|
|
"Activated",
|
|
"Failed",
|
|
"Cancled"
|
|
)
|
|
|
|
NM_SERVICE = 'org.freedesktop.NetworkManager'
|
|
NM_IFACE = 'org.freedesktop.NetworkManager'
|
|
NM_IFACE_DEVICES = 'org.freedesktop.NetworkManager.Devices'
|
|
NM_PATH = '/org/freedesktop/NetworkManager'
|
|
|
|
DEVICE_TYPE_UNKNOWN = 0
|
|
DEVICE_TYPE_802_3_ETHERNET = 1
|
|
DEVICE_TYPE_802_11_WIRELESS = 2
|
|
|
|
NM_DEVICE_CAP_NONE = 0x00000000
|
|
NM_DEVICE_CAP_NM_SUPPORTED = 0x00000001
|
|
NM_DEVICE_CAP_CARRIER_DETECT = 0x00000002
|
|
NM_DEVICE_CAP_WIRELESS_SCAN = 0x00000004
|
|
|
|
|
|
sys_bus = dbus.SystemBus()
|
|
|
|
|
|
class Network(gobject.GObject):
|
|
__gsignals__ = {
|
|
'init-failed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([]))
|
|
}
|
|
|
|
def __init__(self, op):
|
|
gobject.GObject.__init__(self)
|
|
self._op = op
|
|
self._ssid = None
|
|
self._mode = None
|
|
self._strength = 0
|
|
self._valid = False
|
|
|
|
obj = sys_bus.get_object(NM_SERVICE, self._op)
|
|
net = dbus.Interface(obj, NM_IFACE_DEVICES)
|
|
net.getProperties(reply_handler=self._update_reply_cb,
|
|
error_handler=self._update_error_cb)
|
|
|
|
def _update_reply_cb(self, *props):
|
|
self._ssid = props[1]
|
|
self._strength = props[3]
|
|
self._mode = props[6]
|
|
self._valid = True
|
|
logging.debug("Net(%s): ssid '%s', mode %d, strength %d" % (self._op,
|
|
self._ssid, self._mode, self._strength))
|
|
|
|
def _update_error_cb(self, err):
|
|
logging.debug("Net(%s): failed to update. (%s)" % (self._op, err))
|
|
self._valid = False
|
|
self.emit('init-failed')
|
|
|
|
def get_ssid(self):
|
|
return self._ssid
|
|
|
|
def get_op(self):
|
|
return self._op
|
|
|
|
def get_strength(self):
|
|
return self._strength
|
|
|
|
def set_strength(self, strength):
|
|
self._strength = strength
|
|
|
|
def is_valid(self):
|
|
return self._valid
|
|
|
|
def add_to_menu(self, menu, callback, dev):
|
|
strength = self._strength
|
|
if strength > 100:
|
|
strength = 100
|
|
elif strength < 0:
|
|
strength = 0
|
|
item = NetworkMenuItem(text=self._ssid, percent=strength)
|
|
item.connect('button-press-event', callback, (dev, self))
|
|
menu.add_item(item)
|
|
|
|
|
|
class Device(gobject.GObject):
|
|
__gsignals__ = {
|
|
'init-failed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])),
|
|
'activated': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])),
|
|
'strength-changed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE,
|
|
([gobject.TYPE_PYOBJECT]))
|
|
}
|
|
|
|
def __init__(self, op):
|
|
gobject.GObject.__init__(self)
|
|
self._op = op
|
|
self._iface = None
|
|
self._type = DEVICE_TYPE_UNKNOWN
|
|
self._udi = None
|
|
self._active = False
|
|
self._strength = 0
|
|
self._link = False
|
|
self._valid = False
|
|
self._networks = {}
|
|
self._active_net = None
|
|
self._caps = 0
|
|
|
|
obj = sys_bus.get_object(NM_SERVICE, self._op)
|
|
dev = dbus.Interface(obj, NM_IFACE_DEVICES)
|
|
dev.getProperties(reply_handler=self._update_reply_cb,
|
|
error_handler=self._update_error_cb)
|
|
|
|
def _update_reply_cb(self, *props):
|
|
self._iface = props[1]
|
|
self._type = props[2]
|
|
self._udi = props[3]
|
|
self._active = props[4]
|
|
if self._active:
|
|
self.emit('activated')
|
|
self._link = props[15]
|
|
self._caps = props[17]
|
|
|
|
if self._type == DEVICE_TYPE_802_11_WIRELESS:
|
|
old_strength = self._strength
|
|
self._strength = props[14]
|
|
if self._strength != old_strength:
|
|
self.emit('strength-changed', self._strength)
|
|
self._update_networks(props[20], props[19])
|
|
|
|
self._valid = True
|
|
|
|
def _update_networks(self, net_ops, active_op):
|
|
for op in net_ops:
|
|
net = Network(op)
|
|
self._networks[op] = net
|
|
net.connect('init-failed', self._net_init_failed)
|
|
if op == active_op:
|
|
self._active_net = op
|
|
|
|
def _update_error_cb(self, err):
|
|
logging.debug("Device(%s): failed to update. (%s)" % (self._op, err))
|
|
self._valid = False
|
|
self.emit('init-failed')
|
|
|
|
def _net_init_failed(self, net):
|
|
net_op = net.get_op()
|
|
if not self._networks.has_key(net_op):
|
|
return
|
|
if net_op == self._active_net:
|
|
self._active_net = None
|
|
del self._networks[net_op]
|
|
|
|
def _add_to_menu_wired(self, menu, callback):
|
|
item = NetworkMenuItem(_("Wired Network"), stylesheet="nm.Bubble.Wired",
|
|
hi_stylesheet="nm.Bubble.Wired.Hi",
|
|
act_stylesheet="nm.Bubble.Wired.Activated")
|
|
item.connect('button-press-event', callback, (self, None))
|
|
menu.add_item(item)
|
|
|
|
def _add_to_menu_wireless(self, menu, callback, active_only):
|
|
act_net = None
|
|
if self._active_net and self._networks.has_key(self._active_net):
|
|
act_net = self._networks[self._active_net]
|
|
|
|
# Only add the active network if active_only == True
|
|
if active_only and act_net:
|
|
act_net.add_to_menu(menu, callback, self)
|
|
return
|
|
|
|
# Otherwise, add all networks _except_ the active one
|
|
for net in self._networks.values():
|
|
if not net.is_valid():
|
|
continue
|
|
if act_net == net:
|
|
continue
|
|
net.add_to_menu(menu, callback, self)
|
|
|
|
def add_to_menu(self, menu, callback, active_only=False):
|
|
if self._type == DEVICE_TYPE_802_3_ETHERNET:
|
|
self._add_to_menu_wired(menu, callback)
|
|
elif self._type == DEVICE_TYPE_802_11_WIRELESS:
|
|
self._add_to_menu_wireless(menu, callback, active_only)
|
|
|
|
def get_op(self):
|
|
return self._op
|
|
|
|
def get_network(self, op):
|
|
if self._networks.has_key(op):
|
|
return self._networks[op]
|
|
return None
|
|
|
|
def get_network_ops(self):
|
|
return self._networks.keys()
|
|
|
|
def get_strength(self):
|
|
return self._strength
|
|
|
|
def set_strength(self, strength):
|
|
if strength == self._strength:
|
|
return False
|
|
|
|
if strength >= 0 and strength <= 100:
|
|
self._strength = strength
|
|
else:
|
|
self._strength = 0
|
|
|
|
self.emit('strength-changed', self._strength)
|
|
|
|
def network_appeared(self, network):
|
|
if self._networks.has_key(network):
|
|
return
|
|
net = Network(network)
|
|
self._networks[network] = net
|
|
net.connect('init-failed', self._net_init_failed)
|
|
|
|
def network_disappeared(self, network):
|
|
if not self._networks.has_key(network):
|
|
return
|
|
if network == self._active_net:
|
|
self._active_net = None
|
|
del self._networks[network]
|
|
|
|
def get_active(self):
|
|
return self._active
|
|
|
|
def set_active(self, active, ssid=None):
|
|
self._active = active
|
|
if self._type == DEVICE_TYPE_802_11_WIRELESS:
|
|
if not ssid:
|
|
self._active_net = None
|
|
else:
|
|
for (op, net) in self._networks.items():
|
|
if net.get_ssid() == ssid:
|
|
self._active_net = op
|
|
|
|
def get_type(self):
|
|
return self._type
|
|
|
|
def is_valid(self):
|
|
return self._valid
|
|
|
|
def set_carrier(self, on):
|
|
self._link = on
|
|
|
|
def get_capabilities(self):
|
|
return self._caps
|
|
|
|
nm_bubble_wireless = {
|
|
'fill-color' : 0x646464FF,
|
|
'stroke-color' : 0x646464FF,
|
|
'progress-color': 0x333333FF,
|
|
'spacing' : style.space_unit,
|
|
'padding' : style.space_unit * 1.5
|
|
}
|
|
|
|
nm_bubble_wireless_hi = {
|
|
'fill-color' : 0x979797FF,
|
|
'stroke-color' : 0x979797FF,
|
|
'progress-color': 0x666666FF,
|
|
'spacing' : style.space_unit,
|
|
'padding' : style.space_unit * 1.5
|
|
}
|
|
|
|
nm_bubble_wireless_activated = {
|
|
'fill-color' : 0xA7A7A7FF,
|
|
'stroke-color' : 0xA7A7A7FF,
|
|
'progress-color': 0x777777FF,
|
|
'spacing' : style.space_unit,
|
|
'padding' : style.space_unit * 1.5
|
|
}
|
|
|
|
nm_bubble_wired = {
|
|
'fill-color' : 0x000000FF,
|
|
'stroke-color' : 0x000000FF,
|
|
'progress-color': 0x000000FF,
|
|
'spacing' : style.space_unit,
|
|
'padding' : style.space_unit * 1.5
|
|
}
|
|
|
|
nm_bubble_wired_hi = {
|
|
'fill-color' : 0x333333FF,
|
|
'stroke-color' : 0x333333FF,
|
|
'progress-color': 0x000000FF,
|
|
'spacing' : style.space_unit,
|
|
'padding' : style.space_unit * 1.5
|
|
}
|
|
|
|
nm_bubble_wired_activated = {
|
|
'fill-color' : 0x444444FF,
|
|
'stroke-color' : 0x444444FF,
|
|
'progress-color': 0x000000FF,
|
|
'spacing' : style.space_unit,
|
|
'padding' : style.space_unit * 1.5
|
|
}
|
|
|
|
nm_menu_item_title = {
|
|
'xalign': hippo.ALIGNMENT_START,
|
|
'padding-left': 5,
|
|
'color' : 0xFFFFFFFF,
|
|
'font' : style.get_font_description('Bold', 1.2)
|
|
}
|
|
|
|
|
|
style.register_stylesheet("nm.Bubble.Wireless", nm_bubble_wireless)
|
|
style.register_stylesheet("nm.Bubble.Wireless.Hi", nm_bubble_wireless_hi)
|
|
style.register_stylesheet("nm.Bubble.Wireless.Activated", nm_bubble_wireless_activated)
|
|
style.register_stylesheet("nm.Bubble.Wired", nm_bubble_wired)
|
|
style.register_stylesheet("nm.Bubble.Wired.Hi", nm_bubble_wired_hi)
|
|
style.register_stylesheet("nm.Bubble.Wired.Activated", nm_bubble_wired_activated)
|
|
style.register_stylesheet("nm.MenuItem.Title", nm_menu_item_title)
|
|
|
|
class NetworkMenuItem(Bubble):
|
|
def __init__(self, text, percent=0, stylesheet="nm.Bubble.Wireless",
|
|
hi_stylesheet="nm.Bubble.Wireless.Hi",
|
|
act_stylesheet="nm.Bubble.Wireless.Activated"):
|
|
Bubble.__init__(self, percent=percent)
|
|
self._hover = False
|
|
self._default_stylesheet = stylesheet
|
|
self._hi_stylesheet = hi_stylesheet
|
|
self._act_stylesheet = act_stylesheet
|
|
style.apply_stylesheet(self, stylesheet)
|
|
|
|
text_item = hippo.CanvasText(text=text)
|
|
style.apply_stylesheet(text_item, 'nm.MenuItem.Title')
|
|
self.append(text_item)
|
|
|
|
self.connect('motion-notify-event', self._motion_notify_event_cb)
|
|
# Disable active hilight for now...
|
|
#self.connect('button-press-event', self._button_press_event_cb)
|
|
|
|
def _motion_notify_event_cb(self, widget, event):
|
|
if event.detail == hippo.MOTION_DETAIL_ENTER:
|
|
if not self._hover:
|
|
self._hover = True
|
|
style.apply_stylesheet(self, self._hi_stylesheet)
|
|
elif event.detail == hippo.MOTION_DETAIL_LEAVE:
|
|
if self._hover:
|
|
self._hover = False
|
|
style.apply_stylesheet(self, self._default_stylesheet)
|
|
return True
|
|
|
|
def _button_press_event_cb(self, widget, event):
|
|
style.apply_stylesheet(self, self._act_stylesheet)
|
|
return False
|
|
|
|
class NetworkMenu(gtk.Window):
|
|
__gsignals__ = {
|
|
'action': (gobject.SIGNAL_RUN_FIRST,
|
|
gobject.TYPE_NONE, ([int])),
|
|
}
|
|
|
|
def __init__(self):
|
|
gtk.Window.__init__(self, gtk.WINDOW_POPUP)
|
|
|
|
canvas = hippo.Canvas()
|
|
self.add(canvas)
|
|
canvas.show()
|
|
|
|
self._root = hippo.CanvasBox()
|
|
style.apply_stylesheet(self._root, 'menu')
|
|
canvas.set_root(self._root)
|
|
|
|
def add_separator(self):
|
|
separator = hippo.CanvasBox()
|
|
style.apply_stylesheet(separator, 'menu.Separator')
|
|
self._root.append(separator)
|
|
|
|
def add_item(self, item):
|
|
self._root.append(item)
|
|
|
|
|
|
|
|
NM_STATE_UNKNOWN = 0
|
|
NM_STATE_ASLEEP = 1
|
|
NM_STATE_CONNECTING = 2
|
|
NM_STATE_CONNECTED = 3
|
|
NM_STATE_DISCONNECTED = 4
|
|
|
|
ICON_WIRED = "stock-net-wired"
|
|
ICON_WIRELESS_00 = "stock-net-wireless-00"
|
|
ICON_WIRELESS_01_20 = "stock-net-wireless-01-20"
|
|
ICON_WIRELESS_21_40 = "stock-net-wireless-21-40"
|
|
ICON_WIRELESS_41_60 = "stock-net-wireless-41-60"
|
|
ICON_WIRELESS_61_80 = "stock-net-wireless-61-80"
|
|
ICON_WIRELESS_81_100 = "stock-net-wireless-81-100"
|
|
|
|
class NMClientApp:
|
|
def __init__(self):
|
|
self.nminfo = None
|
|
self._nm_present = False
|
|
self._nm_state = NM_STATE_UNKNOWN
|
|
self._icon_theme = gtk.icon_theme_get_default()
|
|
self._update_timer = 0
|
|
self._active_device = None
|
|
self._devices = {}
|
|
|
|
self._menu = None
|
|
self._hover_menu = False
|
|
self._timeline = Timeline(self)
|
|
self._timeline.add_tag('popup', 6, 6)
|
|
self._timeline.add_tag('before_popdown', 7, 7)
|
|
self._timeline.add_tag('popdown', 8, 8)
|
|
|
|
self._icons = {}
|
|
self._cur_icon = None
|
|
|
|
try:
|
|
self.nminfo = nminfo.NMInfo()
|
|
except RuntimeError:
|
|
pass
|
|
self._setup_dbus()
|
|
if self._nm_present:
|
|
self._get_nm_state()
|
|
self._get_initial_devices()
|
|
|
|
try:
|
|
self._icons = self._load_icons()
|
|
except RuntimeError:
|
|
logging.debug("Couldn't find required icon resources, will exit.")
|
|
os._exit(1)
|
|
self._setup_trayicon()
|
|
|
|
def _get_one_icon_pixbuf(self, name):
|
|
info = self._icon_theme.lookup_icon(name, 75, 0)
|
|
if not info or not info.get_filename():
|
|
raise RuntimeError
|
|
return gtk.gdk.pixbuf_new_from_file(info.get_filename())
|
|
|
|
def _load_icons(self):
|
|
icons = {}
|
|
icons[ICON_WIRED] = self._get_one_icon_pixbuf(ICON_WIRED)
|
|
icons[ICON_WIRELESS_00] = self._get_one_icon_pixbuf(ICON_WIRELESS_00)
|
|
icons[ICON_WIRELESS_01_20] = self._get_one_icon_pixbuf(ICON_WIRELESS_01_20)
|
|
icons[ICON_WIRELESS_21_40] = self._get_one_icon_pixbuf(ICON_WIRELESS_21_40)
|
|
icons[ICON_WIRELESS_41_60] = self._get_one_icon_pixbuf(ICON_WIRELESS_41_60)
|
|
icons[ICON_WIRELESS_61_80] = self._get_one_icon_pixbuf(ICON_WIRELESS_61_80)
|
|
icons[ICON_WIRELESS_81_100] = self._get_one_icon_pixbuf(ICON_WIRELESS_81_100)
|
|
return icons
|
|
|
|
def _get_nm_state(self):
|
|
# Grab NM's state
|
|
nm_obj = sys_bus.get_object(NM_SERVICE, NM_PATH)
|
|
nm = dbus.Interface(nm_obj, NM_IFACE)
|
|
nm.state(reply_handler=self._get_state_reply_cb, \
|
|
error_handler=self._get_state_error_cb)
|
|
|
|
def _get_state_reply_cb(self, state):
|
|
if self._nm_state != state:
|
|
self._schedule_icon_update(immediate=True)
|
|
self._nm_state = state
|
|
|
|
def _get_state_error_cb(self, err):
|
|
logging.debug("Failed to get NetworkManager state! %s" % err)
|
|
|
|
def _get_icon(self):
|
|
act_dev = None
|
|
if self._active_device and self._devices.has_key(self._active_device):
|
|
act_dev = self._devices[self._active_device]
|
|
|
|
pixbuf = None
|
|
if not self._nm_present \
|
|
or not act_dev \
|
|
or self._nm_state == NM_STATE_UNKNOWN \
|
|
or self._nm_state == NM_STATE_ASLEEP \
|
|
or self._nm_state == NM_STATE_DISCONNECTED:
|
|
pixbuf = self._icons[ICON_WIRELESS_00]
|
|
elif act_dev.get_type() == DEVICE_TYPE_802_3_ETHERNET:
|
|
pixbuf = self._icons[ICON_WIRED]
|
|
elif act_dev.get_type() == DEVICE_TYPE_802_11_WIRELESS:
|
|
strength = act_dev.get_strength()
|
|
if strength <= 0:
|
|
pixbuf = self._icons[ICON_WIRELESS_00]
|
|
elif strength >= 1 and strength <= 20:
|
|
pixbuf = self._icons[ICON_WIRELESS_01_20]
|
|
elif strength >= 21 and strength <= 40:
|
|
pixbuf = self._icons[ICON_WIRELESS_21_40]
|
|
elif strength >= 41 and strength <= 60:
|
|
pixbuf = self._icons[ICON_WIRELESS_41_60]
|
|
elif strength >= 61 and strength <= 80:
|
|
pixbuf = self._icons[ICON_WIRELESS_61_80]
|
|
elif strength >= 81 and strength:
|
|
pixbuf = self._icons[ICON_WIRELESS_81_100]
|
|
|
|
if not pixbuf:
|
|
pixbuf = self._icons[ICON_WIRELESS_00]
|
|
return pixbuf
|
|
|
|
def _setup_trayicon(self):
|
|
pixbuf = self._get_icon()
|
|
self._trayicon = gtk.status_icon_new_from_pixbuf(pixbuf)
|
|
self._trayicon.connect("popup_menu", self._status_icon_clicked)
|
|
self._trayicon.connect("activate", self._status_icon_clicked)
|
|
self._schedule_icon_update()
|
|
|
|
def _status_icon_clicked(self, button=0, time=None):
|
|
self._timeline.play(None, 'popup')
|
|
|
|
def _get_menu_position(self, menu, item):
|
|
(screen, rect, orientation) = item.get_geometry()
|
|
[item_x, item_y, item_w, item_h] = rect
|
|
[menu_w, menu_h] = menu.size_request()
|
|
|
|
x = item_x + item_w - menu_w
|
|
y = item_y + item_h
|
|
|
|
x = min(x, screen.get_width() - menu_w)
|
|
x = max(0, x)
|
|
|
|
y = min(y, screen.get_height() - menu_h)
|
|
y = max(0, y)
|
|
|
|
return (x, y)
|
|
|
|
def do_popup(self, current, n_frames):
|
|
if self._menu:
|
|
return
|
|
|
|
self._menu = self._create_menu()
|
|
self._menu.connect('enter-notify-event',
|
|
self._menu_enter_notify_event_cb)
|
|
self._menu.connect('leave-notify-event',
|
|
self._menu_leave_notify_event_cb)
|
|
(x, y) = self._get_menu_position(self._menu, self._trayicon)
|
|
self._menu.move(x, y)
|
|
self._menu.show_all()
|
|
|
|
def do_popdown(self, current, frame):
|
|
if self._menu:
|
|
self._menu.destroy()
|
|
self._menu = None
|
|
|
|
def _popdown(self):
|
|
self._timeline.play('popdown', 'popdown')
|
|
|
|
def _menu_enter_notify_event_cb(self, widget, event):
|
|
self._hover_menu = True
|
|
self._timeline.play('popup', 'popup')
|
|
|
|
def _menu_leave_notify_event_cb(self, widget, event):
|
|
self._hover_menu = False
|
|
self._popdown()
|
|
|
|
def _create_menu(self):
|
|
menu = NetworkMenu()
|
|
|
|
# Active device goes above the separator
|
|
act_dev = None
|
|
if self._active_device and self._devices.has_key(self._active_device):
|
|
act_dev = self._devices[self._active_device]
|
|
|
|
if act_dev:
|
|
act_dev.add_to_menu(menu, self._menu_item_clicked_cb, active_only=True)
|
|
menu.add_separator()
|
|
|
|
# Wired devices first, if they don't support carrier detect
|
|
for dev in self._devices.values():
|
|
if not dev.is_valid():
|
|
continue
|
|
if dev.get_type() != DEVICE_TYPE_802_3_ETHERNET:
|
|
continue
|
|
if dev.get_capabilities() & NM_DEVICE_CAP_CARRIER_DETECT:
|
|
continue
|
|
if dev == act_dev:
|
|
continue
|
|
dev.add_to_menu(menu, self._menu_item_clicked_cb)
|
|
|
|
# Wireless devices second
|
|
for dev in self._devices.values():
|
|
if not dev.is_valid():
|
|
continue
|
|
if dev.get_type() != DEVICE_TYPE_802_11_WIRELESS:
|
|
continue
|
|
dev.add_to_menu(menu, self._menu_item_clicked_cb)
|
|
|
|
return menu
|
|
|
|
def _update_icon(self):
|
|
pixbuf = self._get_icon()
|
|
if self._cur_icon != pixbuf:
|
|
self._trayicon.set_from_pixbuf(pixbuf)
|
|
self._cur_icon = pixbuf
|
|
|
|
blink = False
|
|
if self._nm_state == NM_STATE_CONNECTING:
|
|
blink = True
|
|
self._trayicon.set_blinking(blink)
|
|
|
|
self._update_timer = 0
|
|
return False
|
|
|
|
def _schedule_icon_update(self, immediate=False):
|
|
if immediate and self._update_timer:
|
|
gobject.source_remove(self._update_timer)
|
|
self._update_timer = 0
|
|
|
|
if self._update_timer != 0:
|
|
# There is already an update scheduled
|
|
return
|
|
|
|
if immediate:
|
|
self._update_timer = gobject.idle_add(self._update_icon)
|
|
else:
|
|
self._update_timer = gobject.timeout_add(2000, self._update_icon)
|
|
|
|
def _get_initial_devices_reply_cb(self, ops):
|
|
for op in ops:
|
|
self._add_device(op)
|
|
|
|
def _dev_init_failed_cb(self, dev):
|
|
# Device failed to initialize, likely due to dbus errors or something
|
|
op = dev.get_op()
|
|
self._remove_device(op)
|
|
|
|
def _get_initial_devices_error_cb(self, err):
|
|
logging.debug("Error updating devices (%s)" % err)
|
|
|
|
def _get_initial_devices(self):
|
|
nm_obj = sys_bus.get_object(NM_SERVICE, NM_PATH)
|
|
nm = dbus.Interface(nm_obj, NM_IFACE)
|
|
nm.getDevices(reply_handler=self._get_initial_devices_reply_cb, \
|
|
error_handler=self._get_initial_devices_error_cb)
|
|
|
|
def _add_device(self, dev_op):
|
|
if self._devices.has_key(dev_op):
|
|
return
|
|
dev = Device(dev_op)
|
|
self._devices[dev_op] = dev
|
|
dev.connect('init-failed', self._dev_init_failed_cb)
|
|
dev.connect('activated', self._dev_activated_cb)
|
|
dev.connect('strength-changed', self._dev_strength_changed_cb)
|
|
|
|
def _remove_device(self, dev_op):
|
|
if not self._devices.has_key(dev_op):
|
|
return
|
|
if self._active_device == dev_op:
|
|
self._active_device = None
|
|
dev = self._devices[dev_op]
|
|
dev.disconnect('activated')
|
|
dev.disconnect('init-failed')
|
|
dev.disconnect('strength-changed')
|
|
del self._devices[dev_op]
|
|
self._schedule_icon_update(immediate=True)
|
|
|
|
def _dev_activated_cb(self, dev):
|
|
op = dev.get_op()
|
|
if not self._devices.has_key(op):
|
|
return
|
|
if not dev.get_active():
|
|
return
|
|
self._active_device = op
|
|
self._schedule_icon_update(immediate=True)
|
|
|
|
def _dev_strength_changed_cb(self, dev, strength):
|
|
op = dev.get_op()
|
|
if not self._devices.has_key(op):
|
|
return
|
|
if not dev.get_active():
|
|
return
|
|
self._schedule_icon_update()
|
|
|
|
def _setup_dbus(self):
|
|
self._sig_handlers = {
|
|
'StateChange': self.state_change_sig_handler,
|
|
'DeviceAdded': self.device_added_sig_handler,
|
|
'DeviceRemoved': self.device_removed_sig_handler,
|
|
'DeviceActivationStage': self.device_activation_stage_sig_handler,
|
|
'DeviceActivating': self.device_activating_sig_handler,
|
|
'DeviceNowActive': self.device_now_active_sig_handler,
|
|
'DeviceNoLongerActive': self.device_no_longer_active_sig_handler,
|
|
'DeviceCarrierOn': self.device_carrier_on_sig_handler,
|
|
'DeviceCarrierOff': self.device_carrier_off_sig_handler,
|
|
'DeviceStrengthChanged': self.wireless_device_strength_changed_sig_handler,
|
|
'WirelessNetworkAppeared': self.wireless_network_appeared_sig_handler,
|
|
'WirelessNetworkDisappeared': self.wireless_network_disappeared_sig_handler,
|
|
'WirelessNetworkStrengthChanged': self.wireless_network_strength_changed_sig_handler
|
|
}
|
|
|
|
self.nm_proxy = sys_bus.get_object(NM_SERVICE, NM_PATH)
|
|
|
|
sys_bus.add_signal_receiver(self.name_owner_changed_sig_handler,
|
|
signal_name="NameOwnerChanged",
|
|
dbus_interface="org.freedesktop.DBus")
|
|
|
|
sys_bus.add_signal_receiver(self.catchall_signal_handler,
|
|
dbus_interface=NM_IFACE)
|
|
|
|
sys_bus.add_signal_receiver(self.catchall_signal_handler,
|
|
dbus_interface=NM_IFACE + 'Devices')
|
|
|
|
for (signal, handler) in self._sig_handlers.items():
|
|
sys_bus.add_signal_receiver(handler, signal_name=signal, dbus_interface=NM_IFACE)
|
|
|
|
# Find out whether or not NM is running
|
|
try:
|
|
bus_object = sys_bus.get_object('org.freedesktop.DBus', '/org/freedesktop/DBus')
|
|
name = bus_object.GetNameOwner("org.freedesktop.NetworkManagerInfo", \
|
|
dbus_interface='org.freedesktop.DBus')
|
|
if name:
|
|
self._nm_present = True
|
|
except dbus.DBusException:
|
|
pass
|
|
|
|
@dbus.decorators.explicitly_pass_message
|
|
def catchall_signal_handler(self, *args, **keywords):
|
|
dbus_message = keywords['dbus_message']
|
|
mem = dbus_message.get_member()
|
|
iface = dbus_message.get_interface()
|
|
|
|
if iface == NM_IFACE and mem in self._sig_handlers.keys():
|
|
return
|
|
|
|
logging.debug('Caught signal %s.%s' % (dbus_message.get_interface(), mem))
|
|
for arg in args:
|
|
logging.debug(' ' + str(arg))
|
|
|
|
def _menu_item_clicked_cb(self, widget, event, dev_data):
|
|
(device, network) = dev_data
|
|
net_op = ""
|
|
if network:
|
|
net_op = network.get_op()
|
|
logging.debug("clicked dev %s, net %s" % (device.get_op(), net_op))
|
|
|
|
def device_activation_stage_sig_handler(self, device, stage):
|
|
print 'Network Manager Device Stage "%s" for device %s'%(NM_DEVICE_STAGE_STRINGS[stage], device)
|
|
|
|
def state_change_sig_handler(self, state):
|
|
self._nm_state = state
|
|
self._schedule_icon_update(immediate=True)
|
|
|
|
def device_activating_sig_handler(self, device):
|
|
self._active_device = device
|
|
|
|
def device_now_active_sig_handler(self, device, ssid=None):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
self._active_device = device
|
|
self._devices[device].set_active(True, ssid)
|
|
self._schedule_icon_update(immediate=True)
|
|
|
|
def device_no_longer_active_sig_handler(self, device):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
if self._active_device == device:
|
|
self._active_device = None
|
|
self._devices[device].set_active(False)
|
|
self._schedule_icon_update(immediate=True)
|
|
|
|
def name_owner_changed_sig_handler(self, name, old, new):
|
|
if name != NM_SERVICE:
|
|
return
|
|
if (old and len(old)) and (not new and not len(new)):
|
|
# NM went away
|
|
self._nm_present = False
|
|
self._schedule_update_timer()
|
|
for op in self._devices.keys():
|
|
del self._devices[op]
|
|
self._devices = {}
|
|
self._active_device = None
|
|
self._nm_state = NM_STATE_UNKNOWN
|
|
elif (not old and not len(old)) and (new and len(new)):
|
|
# NM started up
|
|
self._nm_present = True
|
|
self._get_nm_state()
|
|
self._get_initial_devices()
|
|
|
|
def device_added_sig_handler(self, device):
|
|
self._add_device(device)
|
|
|
|
def device_removed_sig_handler(self, device):
|
|
self._remove_device(device)
|
|
|
|
def wireless_network_appeared_sig_handler(self, device, network):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
self._devices[device].network_appeared(network)
|
|
|
|
def wireless_network_disappeared_sig_handler(self, device, network):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
self._devices[device].network_disappeared(network)
|
|
|
|
def wireless_device_strength_changed_sig_handler(self, device, strength):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
self._devices[device].set_strength(strength)
|
|
|
|
def wireless_network_strength_changed_sig_handler(self, device, network, strength):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
net = self._devices[device].get_network(network)
|
|
if net:
|
|
net.set_strength(strength)
|
|
|
|
def device_carrier_on_sig_handler(self, device):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
self._devices[device].set_carrier(True)
|
|
|
|
def device_carrier_off_sig_handler(self, device):
|
|
if not self._devices.has_key(device):
|
|
return
|
|
self._devices[device].set_carrier(False)
|
|
|
|
def run(self):
|
|
loop = gobject.MainLoop()
|
|
try:
|
|
loop.run()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|