mirror of
https://github.com/gryf/wicd.git
synced 2025-12-20 12:58:07 +01:00
219 lines
8.8 KiB
Python
Executable File
219 lines
8.8 KiB
Python
Executable File
########
|
|
## DO NOT RUN THIS FILE DIRECTLY
|
|
## USE TRAY.PY INSTEAD
|
|
## nothing bad will happen if you do
|
|
## but that is not the preferred method
|
|
## tray.py automatically picks the correct version
|
|
## of the tray icon to use
|
|
########
|
|
|
|
########
|
|
##thanks to Arne Brix for programming most of this
|
|
##released under the GNU Public License
|
|
##see http://www.gnu.org/copyleft/gpl.html for details
|
|
##this will only work in Edgy and above because of gtk requirements
|
|
##to run the tray icon
|
|
########
|
|
import os,sys
|
|
if __name__ == '__main__':
|
|
os.chdir(os.path.dirname(os.path.normpath(os.path.join(os.getcwd(),sys.argv[0]))))
|
|
import gtk, gobject, dbus, dbus.service, os, sys, locale, gettext, signal, time
|
|
if getattr(dbus, 'version', (0,0,0)) >= (0,41,0):
|
|
import dbus.glib
|
|
|
|
#############
|
|
#declare the connections to our daemon.
|
|
#without them nothing useful will happen
|
|
#the daemon should be running as root
|
|
#some connections aren't used so they are commented
|
|
bus = dbus.SystemBus()
|
|
try:
|
|
print 'attempting to connect daemon...'
|
|
proxy_obj = bus.get_object('org.wicd.daemon', '/org/wicd/daemon')
|
|
print 'success'
|
|
except:
|
|
print 'daemon not running, running gksudo ./daemon.py...'
|
|
import misc
|
|
misc.PromptToStartDaemon()
|
|
time.sleep(5)
|
|
try:
|
|
proxy_obj = bus.get_object('org.wicd.daemon', '/org/wicd/daemon')
|
|
except:
|
|
print 'daemon still not running, aborting.'
|
|
daemon = dbus.Interface(proxy_obj, 'org.wicd.daemon') # Had to uncomment it
|
|
wireless = dbus.Interface(proxy_obj, 'org.wicd.daemon.wireless')
|
|
wired = dbus.Interface(proxy_obj, 'org.wicd.daemon.wired')
|
|
config = dbus.Interface(proxy_obj, 'org.wicd.daemon.config')
|
|
#############
|
|
|
|
local_path = os.path.realpath(os.path.dirname(sys.argv[0])) + '/translations'
|
|
langs = []
|
|
lc, encoding = locale.getdefaultlocale()
|
|
if (lc):
|
|
langs = [lc]
|
|
osLanguage = os.environ.get('LANGUAGE', None)
|
|
if (osLanguage):
|
|
langs += osLanguage.split(":")
|
|
langs += ["en_US"]
|
|
lang = gettext.translation('wicd', local_path, languages=langs, fallback = True)
|
|
_ = lang.gettext
|
|
language = {}
|
|
language['connected_to_wireless'] = _('Connected to $A at $B% (IP: $C)')
|
|
language['connected_to_wired'] = _('Connected to wired network (IP: $A)')
|
|
language['not_connected'] = _('Not connected')
|
|
|
|
tr=None
|
|
tooltip = gtk.Tooltips()
|
|
pic=None
|
|
lastWinId = 0
|
|
|
|
def open_wicd_gui():
|
|
global lastWinId
|
|
ret = 0
|
|
if lastWinId != 0:
|
|
os.kill(lastWinId,signal.SIGQUIT)
|
|
ret = os.waitpid(lastWinId,0)[1]
|
|
lastWinId = 0
|
|
if ret == 0:
|
|
lastWinId = os.spawnlpe(os.P_NOWAIT, './gui.py', os.environ)
|
|
|
|
def set_signal_image():
|
|
global LastStrength
|
|
global stillWired #keeps us from resetting the wired info over and over
|
|
global network #declared as global so it gets initialized before initial use
|
|
|
|
# Disable logging if debugging isn't on to prevent log spam
|
|
if not daemon.GetDebugMode():
|
|
config.DisableLogging()
|
|
|
|
wired_ip = wired.GetWiredIP()
|
|
if wired.CheckPluggedIn() == True and wired_ip:
|
|
if stillWired == False: # Only set image/tooltip if it hasn't been set already
|
|
tr.set_from_file("images/wired.png")
|
|
tr.set_tooltip(language['connected_to_wired'].replace('$A',wired_ip))
|
|
stillWired = True
|
|
lock = ''
|
|
else:
|
|
stillWired = False
|
|
wireless_ip = wireless.GetWirelessIP()
|
|
#If ip returns as None, we are probably returning from hibernation and need to force signal to 0 to avoid crashing
|
|
if wireless_ip != None:
|
|
signal = int(wireless.GetCurrentSignalStrength())
|
|
else:
|
|
signal = 0
|
|
|
|
#only update if the signal strength has changed because doing I/O calls is expensive,
|
|
#and the icon flickers
|
|
if (signal != LastStrength or network != wireless.GetCurrentNetwork()) and wireless_ip != None:
|
|
LastStrength = signal
|
|
lock = '' #set the string to '' so that when it is put in "high-signal" + lock + ".png", there will be nothing
|
|
curNetID = wireless.GetCurrentNetworkID() #the network ID needs to be checked because a negative value here will break the tray
|
|
if signal > 0 and curNetID > -1 and wireless.GetWirelessProperty(curNetID,"encryption"):
|
|
lock = '-lock' #set the string to '-lock' so that it will display the lock picture
|
|
network = str(wireless.GetCurrentNetwork())
|
|
tr.set_tooltip(language['connected_to_wireless'].replace('$A',network).replace('$B',str(signal)).replace('$C',str(wireless_ip)))
|
|
if signal > 75:
|
|
tr.set_from_file("images/high-signal" + lock + ".png")
|
|
elif signal > 50:
|
|
tr.set_from_file("images/good-signal" + lock + ".png")
|
|
elif signal > 25:
|
|
tr.set_from_file("images/low-signal" + lock + ".png")
|
|
elif signal > 0:
|
|
tr.set_from_file("images/bad-signal" + lock + ".png")
|
|
elif signal == 0:
|
|
tr.set_from_file("images/no-signal.png")
|
|
#Auto-reconnect code - not sure how well this works. I know that without the ForcedDisconnect check it reconnects you when
|
|
#a disconnect is forced. People who have disconnection problems need to test it to determine if it actually works.
|
|
#First it will attempt to reconnect to the last known wireless network, and if that fails it should run a scan and try to
|
|
#connect to any network set to autoconnect.
|
|
if wireless.GetAutoReconnect() == True and wireless.CheckIfWirelessConnecting() == False and wireless.GetForcedDisconnect() == False:
|
|
curNetID = wireless.GetCurrentNetworkID()
|
|
if curNetID > -1:
|
|
wireless.ConnectWireless(wireless.GetCurrentNetworkID())
|
|
print 'Trying to autoreconnect'
|
|
while wireless.CheckIfWirelessConnecting() == True:
|
|
time.sleep(1)
|
|
if wireless.GetCurrentSignalStrength() != 0:
|
|
print "Successfully autoreconnected."
|
|
else:
|
|
print "Couldn't reconnect to last used network, scanning for an autoconnect network..."
|
|
print wireless.AutoConnect(True)
|
|
|
|
elif wireless_ip == None:
|
|
tr.set_from_file("images/no-signal.png")
|
|
tr.set_tooltip(language['not_connected'])
|
|
|
|
if not daemon.GetDebugMode():
|
|
config.EnableLogging()
|
|
|
|
return True
|
|
|
|
class TrackerStatusIcon(gtk.StatusIcon):
|
|
def __init__(self):
|
|
gtk.StatusIcon.__init__(self)
|
|
menu = '''
|
|
<ui>
|
|
<menubar name="Menubar">
|
|
<menu action="Menu">
|
|
<menuitem action="Connect"/>
|
|
<separator/>
|
|
<menuitem action="About"/>
|
|
<menuitem action="Quit"/>
|
|
</menu>
|
|
</menubar>
|
|
</ui>
|
|
'''
|
|
actions = [
|
|
('Menu', None, 'Menu'),
|
|
('Connect', gtk.STOCK_CONNECT, '_Connect...', None, 'Connect to network', self.on_preferences),
|
|
('About', gtk.STOCK_ABOUT, '_About...', None, 'About wicd-tray-icon', self.on_about),
|
|
('Quit',gtk.STOCK_QUIT,'_Quit',None,'Quit wicd-tray-icon', self.on_quit),
|
|
|
|
]
|
|
ag = gtk.ActionGroup('Actions')
|
|
ag.add_actions(actions)
|
|
self.manager = gtk.UIManager()
|
|
self.manager.insert_action_group(ag, 0)
|
|
self.manager.add_ui_from_string(menu)
|
|
self.menu = self.manager.get_widget('/Menubar/Menu/About').props.parent
|
|
self.current_icon_path = ''
|
|
self.set_from_file("images/no-signal.png")
|
|
self.set_visible(True)
|
|
self.connect('activate', self.on_activate)
|
|
self.connect('popup-menu', self.on_popup_menu)
|
|
|
|
def on_activate(self, data):
|
|
open_wicd_gui()
|
|
|
|
def on_quit(self,widget):
|
|
sys.exit()
|
|
|
|
def on_popup_menu(self, status, button, time):
|
|
self.menu.popup(None, None, None, button, time)
|
|
|
|
def on_preferences(self, data):
|
|
open_wicd_gui()
|
|
|
|
def on_about(self, data):
|
|
dialog = gtk.AboutDialog()
|
|
dialog.set_name('wicd tray icon')
|
|
dialog.set_version('0.2')
|
|
dialog.set_comments('an icon that shows your network connectivity')
|
|
dialog.set_website('http://wicd.sourceforge.net')
|
|
dialog.run()
|
|
dialog.destroy()
|
|
|
|
def set_from_file(self,path):
|
|
if path != self.current_icon_path:
|
|
self.current_icon_path = path
|
|
gtk.StatusIcon.set_from_file(self,path)
|
|
|
|
LastStrength = -2
|
|
stillWired = False
|
|
network = ''
|
|
tr=None
|
|
|
|
tr=TrackerStatusIcon()
|
|
gobject.timeout_add(3000,set_signal_image)
|
|
gtk.main()
|