1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# 2009-12-27 David Goncalves - Version 1.2
# Total rewrite of NUT-Monitor to optimize GUI interaction.
# Added favorites support (saved to user's home)
# Added status icon on the notification area
#
# 2010-02-26 David Goncalves
# Added UPS vars display and the possibility to change values
# when user double-clicks on a RW var.
#
# 2010-05-01 David Goncalves
# Added support for PyNotify (if available)
#
# 2010-05-05 David Goncalves
# Added support for command line options
# -> --start-hidden
# -> --favorite
#
# NUT-Monitor now tries to detect if there is a NUT server
# on localhost and if there is 1 UPS, connects to it.
#
# 2010-10-06 David Goncalves - Version 1.3
# Added localisation support
#
# 2015-02-14 Michal Fincham - Version 1.3.1
# Corrected unsafe permissions on ~/.nut-monitor (Debian #777706)
import gtk, gtk.glade, gobject
import sys
import base64
import os, os.path
import stat
import platform
import time
import threading
import optparse
import ConfigParser
import locale
import gettext
import PyNUT
# Activate threadings on glib
gobject.threads_init()
class interface :
DESIRED_FAVORITES_DIRECTORY_MODE = 0700
__widgets = {}
__callbacks = {}
__favorites = {}
__favorites_file = None
__favorites_path = ""
__fav_menu_items = list()
__window_visible = True
__glade_file = None
__connected = False
__ups_handler = None
__ups_commands = None
__ups_vars = None
__ups_rw_vars = None
__gui_thread = None
__current_ups = None
def __init__( self ) :
# Before anything, parse command line options if any present...
opt_parser = optparse.OptionParser()
opt_parser.add_option( "-H", "--start-hidden", action="store_true", default=False, dest="hidden", help="Start iconified in tray" )
opt_parser.add_option( "-F", "--favorite", dest="favorite", help="Load the specified favorite and connect to UPS" )
( cmd_opts, args ) = opt_parser.parse_args()
self.__glade_file = '/usr/share/nut-monitor/gui-1.3.glade'
self.__widgets["interface"] = gtk.glade.XML( self.__glade_file, "window1", APP )
self.__widgets["main_window"] = self.__widgets["interface"].get_widget("window1")
self.__widgets["status_bar"] = self.__widgets["interface"].get_widget("statusbar2")
self.__widgets["ups_host_entry"] = self.__widgets["interface"].get_widget("entry1")
self.__widgets["ups_port_entry"] = self.__widgets["interface"].get_widget("spinbutton1")
self.__widgets["ups_refresh_button"] = self.__widgets["interface"].get_widget("button1")
self.__widgets["ups_authentication_check"] = self.__widgets["interface"].get_widget("checkbutton1")
self.__widgets["ups_authentication_frame"] = self.__widgets["interface"].get_widget("hbox1")
self.__widgets["ups_authentication_login"] = self.__widgets["interface"].get_widget("entry2")
self.__widgets["ups_authentication_password"] = self.__widgets["interface"].get_widget("entry3")
self.__widgets["ups_list_combo"] = self.__widgets["interface"].get_widget("combobox1")
self.__widgets["ups_commands_button"] = self.__widgets["interface"].get_widget("button8")
self.__widgets["ups_connect"] = self.__widgets["interface"].get_widget("button2")
self.__widgets["ups_disconnect"] = self.__widgets["interface"].get_widget("button7")
self.__widgets["ups_params_box"] = self.__widgets["interface"].get_widget("vbox6")
self.__widgets["ups_infos"] = self.__widgets["interface"].get_widget("notebook1")
self.__widgets["ups_vars_tree"] = self.__widgets["interface"].get_widget("treeview1")
self.__widgets["ups_vars_refresh"] = self.__widgets["interface"].get_widget("button9")
self.__widgets["ups_status_image"] = self.__widgets["interface"].get_widget("image1")
self.__widgets["ups_status_left"] = self.__widgets["interface"].get_widget("label10")
self.__widgets["ups_status_right"] = self.__widgets["interface"].get_widget("label11")
self.__widgets["ups_status_time"] = self.__widgets["interface"].get_widget("label15")
self.__widgets["menu_favorites_root"] = self.__widgets["interface"].get_widget("menuitem3")
self.__widgets["menu_favorites"] = self.__widgets["interface"].get_widget("menu2")
self.__widgets["menu_favorites_add"] = self.__widgets["interface"].get_widget("menuitem4")
self.__widgets["menu_favorites_del"] = self.__widgets["interface"].get_widget("menuitem5")
self.__widgets["progress_battery_charge"] = self.__widgets["interface"].get_widget("progressbar1")
self.__widgets["progress_battery_load"] = self.__widgets["interface"].get_widget("progressbar2")
# Create the tray icon and connect it to the show/hide method...
self.__widgets["status_icon"] = gtk.StatusIcon()
self.__widgets["status_icon"].set_from_file( "/usr/share/nut-monitor/pixmaps/on_line.png" )
self.__widgets["status_icon"].set_visible( True )
self.__widgets["status_icon"].connect( "activate", self.tray_activated )
self.__widgets["ups_status_image"].set_from_file( "/usr/share/nut-monitor/pixmaps/on_line.png" )
# Define interface callbacks actions
self.__callbacks = { "on_window1_destroy" : self.quit,
"on_imagemenuitem1_activate" : self.gui_about_dialog,
"on_imagemenuitem5_activate" : self.quit,
"on_entry1_changed" : self.__check_gui_fields,
"on_entry2_changed" : self.__check_gui_fields,
"on_entry3_changed" : self.__check_gui_fields,
"on_checkbutton1_toggled" : self.__check_gui_fields,
"on_spinbutton1_value_changed" : self.__check_gui_fields,
"on_button1_clicked" : self.__update_ups_list,
"on_button2_clicked" : self.connect_to_ups,
"on_button7_clicked" : self.disconnect_from_ups,
"on_button9_clicked" : self.__gui_update_ups_vars_view,
"on_menuitem4_activate" : self.__gui_add_favorite,
"on_menuitem5_activate" : self.__gui_delete_favorite,
"on_treeview1_button_press_event" : self.__gui_ups_vars_selected
}
# Connect the callbacks
self.__widgets["interface"].signal_autoconnect( self.__callbacks )
# Remove the dummy combobox entry on UPS List and Commands
self.__widgets["ups_list_combo"].remove_text( 0 )
# Set UPS vars treeview properties -----------------------------
store = gtk.ListStore( gtk.gdk.Pixbuf, gobject.TYPE_STRING, gobject.TYPE_STRING )
self.__widgets["ups_vars_tree"].set_model( store )
self.__widgets["ups_vars_tree"].set_headers_visible( True )
# Column 0
cr = gtk.CellRendererPixbuf()
column = gtk.TreeViewColumn( '', cr )
column.add_attribute( cr, 'pixbuf', 0 )
self.__widgets["ups_vars_tree"].append_column( column )
# Column 1
cr = gtk.CellRendererText()
cr.set_property( 'editable', False )
column = gtk.TreeViewColumn( _('Var name'), cr )
column.set_sort_column_id( 1 )
column.add_attribute( cr, 'text', 1 )
self.__widgets["ups_vars_tree"].append_column( column )
# Column 2
cr = gtk.CellRendererText()
cr.set_property( 'editable', False )
column = gtk.TreeViewColumn( _('Value'), cr )
column.add_attribute( cr, 'text', 2 )
self.__widgets["ups_vars_tree"].append_column( column )
self.__widgets["ups_vars_tree"].get_model().set_sort_column_id( 1, gtk.SORT_ASCENDING )
self.__widgets["ups_vars_tree_store"] = store
self.__widgets["ups_vars_tree"].set_size_request( -1, 50 )
#---------------------------------------------------------------
# UPS Commands combo box creation ------------------------------
container = self.__widgets["ups_commands_button"].get_parent()
self.__widgets["ups_commands_button"].destroy()
self.__widgets["ups_commands_combo"] = gtk.ComboBox()
list_store = gtk.ListStore( gobject.TYPE_STRING )
self.__widgets["ups_commands_combo"].set_model( list_store )
cell_renderer = gtk.CellRendererText()
cell_renderer.set_property( "xalign", 0 )
self.__widgets["ups_commands_combo"].pack_start( cell_renderer, True )
self.__widgets["ups_commands_combo"].add_attribute( cell_renderer, "markup", 0 )
container.pack_start( self.__widgets["ups_commands_combo"], True )
self.__widgets["ups_commands_combo"].set_active( 0 )
self.__widgets["ups_commands_combo"].show_all()
self.__widgets["ups_commands_button"] = gtk.Button( stock=gtk.STOCK_EXECUTE )
container.pack_start( self.__widgets["ups_commands_button"], True )
self.__widgets["ups_commands_button"].show()
self.__widgets["ups_commands_button"].connect( "clicked", self.__gui_send_ups_command )
self.__widgets["ups_commands_combo_store"] = list_store
#---------------------------------------------------------------
if ( cmd_opts.hidden != True ) :
self.__widgets["main_window"].show()
# Define favorites path and load favorites
if ( platform.system() == "Linux" ) :
self.__favorites_path = os.path.join( os.environ.get("HOME"), ".nut-monitor" )
elif ( platform.system() == "Windows" ) :
self.__favorites_path = os.path.join( os.environ.get("USERPROFILE"), "Application Data", "NUT-Monitor" )
self.__favorites_file = os.path.join( self.__favorites_path, "favorites.ini" )
self.__parse_favorites()
self.gui_status_message( _("Welcome to NUT Monitor") )
if ( cmd_opts.favorite != None ) :
if ( self.__favorites.has_key( cmd_opts.favorite ) ) :
self.__gui_load_favorite( fav_name=cmd_opts.favorite )
self.connect_to_ups()
else :
# Try to scan localhost for available ups and connect to it if there is only one
self.__widgets["ups_host_entry"].set_text( "localhost" )
self.__update_ups_list()
if ( len( self.__widgets["ups_list_combo"].get_model() ) == 1 ) :
self.connect_to_ups()
# Check if correct fields are filled to enable connection to the UPS
def __check_gui_fields( self, widget=None ) :
# If UPS list contains something, clear it
if self.__widgets["ups_list_combo"].get_active() != -1 :
self.__widgets["ups_list_combo"].get_model().clear()
self.__widgets["ups_connect"].set_sensitive( False )
self.__widgets["menu_favorites_add"].set_sensitive( False )
# Host/Port selection
if len( self.__widgets["ups_host_entry"].get_text() ) > 0 :
sensitive = True
# If authentication is selected, check that we have a login and password
if self.__widgets["ups_authentication_check"].get_active() :
if len( self.__widgets["ups_authentication_login"].get_text() ) == 0 :
sensitive = False
if len( self.__widgets["ups_authentication_password"].get_text() ) == 0 :
sensitive = False
self.__widgets["ups_refresh_button"].set_sensitive( sensitive )
if not sensitive :
self.__widgets["ups_connect"].set_sensitive( False )
self.__widgets["menu_favorites_add"].set_sensitive( False )
else :
self.__widgets["ups_refresh_button"].set_sensitive( False )
self.__widgets["ups_connect"].set_sensitive( False )
self.__widgets["menu_favorites_add"].set_sensitive( False )
# Use authentication fields...
if self.__widgets["ups_authentication_check"].get_active() :
self.__widgets["ups_authentication_frame"].set_sensitive( True )
else :
self.__widgets["ups_authentication_frame"].set_sensitive( False )
self.gui_status_message()
#-------------------------------------------------------------------
# This method is used to show/hide the main window when user clicks on the tray icon
def tray_activated( self, widget=None, data=None ) :
if self.__window_visible :
self.__widgets["main_window"].hide()
else :
self.__widgets["main_window"].show()
self.__window_visible = not self.__window_visible
#-------------------------------------------------------------------
# Change the status icon and tray icon
def change_status_icon( self, icon="on_line", blink=False ) :
self.__widgets["status_icon"].set_from_file( "/usr/share/nut-monitor/pixmaps/%s.png" % icon )
self.__widgets["ups_status_image"].set_from_file( "/usr/share/nut-monitor/pixmaps/%s.png" % icon )
self.__widgets["status_icon"].set_blinking( blink )
#-------------------------------------------------------------------
# This method connects to the NUT server and retrieve availables UPSes
# using connection parameters (host, port, login, pass...)
def __update_ups_list( self, widget=None ) :
host = self.__widgets["ups_host_entry"].get_text()
port = int( self.__widgets["ups_port_entry"].get_value() )
login = None
password = None
if self.__widgets["ups_authentication_check"].get_active() :
login = self.__widgets["ups_authentication_login"].get_text()
password = self.__widgets["ups_authentication_password"].get_text()
try :
nut_handler = PyNUT.PyNUTClient( host=host, port=port, login=login, password=password )
upses = nut_handler.GetUPSList()
ups_list = upses.keys()
ups_list.sort()
# If UPS list contains something, clear it
self.__widgets["ups_list_combo"].get_model().clear()
for current in ups_list :
self.__widgets["ups_list_combo"].append_text( current )
self.__widgets["ups_list_combo"].set_active( 0 )
self.__widgets["ups_connect"].set_sensitive( True )
self.__widgets["menu_favorites_add"].set_sensitive( True )
self.gui_status_message( _("Found {0} devices on {1}").format( len( ups_list ), host ) )
except :
error_msg = _("Error connecting to '{0}' ({1})").format( host, sys.exc_info()[1] )
self.gui_status_message( error_msg )
#-------------------------------------------------------------------
# Quit program
def quit( self, widget=None ) :
# If we are connected to an UPS, disconnect first...
if self.__connected :
self.gui_status_message( _("Disconnecting from device") )
self.disconnect_from_ups()
gtk.main_quit()
#-------------------------------------------------------------------
# Method called when user wants to add a new favorite entry. It
# displays a dialog to enable user to select the name of the favorite
def __gui_add_favorite( self, widget=None ) :
dialog_interface = gtk.glade.XML( self.__glade_file, "dialog1" )
dialog = dialog_interface.get_widget( "dialog1" )
self.__widgets["favorites_dialog_button_add"] = dialog_interface.get_widget("button3")
# Define interface callbacks actions
callbacks = { "on_entry4_changed" : self.__gui_add_favorite_check_gui_fields }
dialog_interface.signal_autoconnect( callbacks )
self.__widgets["main_window"].set_sensitive( False )
rc = dialog.run()
if rc == 1 :
fav_data = {}
fav_data["host"] = self.__widgets["ups_host_entry"].get_text()
fav_data["port"] = "%d" % self.__widgets["ups_port_entry"].get_value()
fav_data["ups"] = self.__widgets["ups_list_combo"].get_active_text()
fav_data["auth"] = self.__widgets["ups_authentication_check"].get_active()
if fav_data["auth"] :
fav_data["login"] = self.__widgets["ups_authentication_login"].get_text()
fav_data["password"] = base64.b64encode( self.__widgets["ups_authentication_password"].get_text() )
fav_name = dialog_interface.get_widget("entry4").get_text()
self.__favorites[ fav_name ] = fav_data
self.__gui_refresh_favorites_menu()
# Save all favorites
self.__save_favorites()
dialog.destroy()
self.__widgets["main_window"].set_sensitive( True )
#-------------------------------------------------------------------
# Method called when user wants to delete an entry from favorites
def __gui_delete_favorite( self, widget=None ) :
dialog_interface = gtk.glade.XML( self.__glade_file, "dialog2" )
dialog = dialog_interface.get_widget( "dialog2" )
# Remove the dummy combobox entry on list
dialog_interface.get_widget("combobox2").remove_text( 0 )
favs = self.__favorites.keys()
favs.sort()
for current in favs :
dialog_interface.get_widget("combobox2").append_text( current )
dialog_interface.get_widget("combobox2").set_active( 0 )
self.__widgets["main_window"].set_sensitive( False )
rc = dialog.run()
fav_name = dialog_interface.get_widget("combobox2").get_active_text()
dialog.destroy()
self.__widgets["main_window"].set_sensitive( True )
if ( rc == 1 ) :
# Remove entry, show confirmation dialog
md = gtk.MessageDialog( None, gtk.DIALOG_MODAL, gtk.MESSAGE_QUESTION, gtk.BUTTONS_YES_NO, _("Are you sure that you want to remove this favorite ?") )
resp = md.run()
md.destroy()
if ( resp == gtk.RESPONSE_YES ) :
del self.__favorites[ fav_name ]
self.__gui_refresh_favorites_menu()
self.__save_favorites()
self.gui_status_message( _("Removed favorite '%s'") % fav_name )
#-------------------------------------------------------------------
# Method called when user selects a favorite from the favorites menu
def __gui_load_favorite( self, fav_name="" ) :
if ( self.__favorites.has_key( fav_name ) ) :
# If auth is activated, process it before other fields to avoir weird
# reactions with the 'check_gui_fields' function.
if ( self.__favorites[fav_name].get("auth", False ) ) :
self.__widgets["ups_authentication_check"].set_active( True )
self.__widgets["ups_authentication_login"].set_text( self.__favorites[fav_name].get("login","") )
self.__widgets["ups_authentication_password"].set_text( self.__favorites[fav_name].get("password","") )
self.__widgets["ups_host_entry"].set_text( self.__favorites[fav_name].get("host","") )
self.__widgets["ups_port_entry"].set_value( float(self.__favorites[fav_name].get("port",3493.0)) )
# Clear UPS list and add current UPS name
self.__widgets["ups_list_combo"].get_model().clear()
self.__widgets["ups_list_combo"].append_text( self.__favorites[fav_name].get("ups","") )
self.__widgets["ups_list_combo"].set_active( 0 )
# Activate the connect button
self.__widgets["ups_connect"].set_sensitive( True )
self.gui_status_message( _("Loaded '%s'") % fav_name )
#-------------------------------------------------------------------
# Send the selected command to the UPS
def __gui_send_ups_command( self, widget=None ) :
offset = self.__widgets["ups_commands_combo"].get_active()
cmd = self.__ups_commands[ offset ]
md = gtk.MessageDialog( None, gtk.DIALOG_MODAL, gtk.MESSAGE_QUESTION, gtk.BUTTONS_YES_NO, _("Are you sure that you want to send\n'%s' to the device ?") % cmd )
self.__widgets["main_window"].set_sensitive( False )
resp = md.run()
md.destroy()
self.__widgets["main_window"].set_sensitive( True )
if ( resp == gtk.RESPONSE_YES ) :
try :
self.__ups_handler.RunUPSCommand( self.__current_ups, cmd )
self.gui_status_message( _("Sent '{0}' command to {1}").format( cmd, self.__current_ups ) )
except :
self.gui_status_message( _("Failed to send '{0}' ({1})").format( cmd, sys.exc_info()[1] ) )
#-------------------------------------------------------------------
# Method called when user clicks on the UPS vars treeview. If the user
# performs a double click on a RW var, the GUI shows the update var dialog.
def __gui_ups_vars_selected( self, widget, event ) :
# Check if it's a double click...
if ( (event.button == 1) and (event.type == gtk.gdk._2BUTTON_PRESS) ) :
treeselection = self.__widgets["ups_vars_tree"].get_selection()
(model,iter) = treeselection.get_selected()
try :
ups_var = model.get_value( iter, 1 )
if ( ups_var in self.__ups_rw_vars ) :
# The selected var is RW, then we can show the update dialog
dialog_interface = gtk.glade.XML( self.__glade_file, "dialog3" )
dialog = dialog_interface.get_widget( "dialog3" )
lab = dialog_interface.get_widget( "label9" )
lab.set_markup( _("Enter a new value for the variable.\n\n{0} = {1} <span color=\"#606060\"><i>(current value)</i></span>").format( ups_var, self.__ups_rw_vars.get(ups_var)) )
str = dialog_interface.get_widget( "entry5" )
str.set_text( self.__ups_rw_vars.get(ups_var) )
self.__widgets["main_window"].set_sensitive( False )
rc = dialog.run()
new_val = str.get_text()
dialog.destroy()
self.__widgets["main_window"].set_sensitive( True )
if ( rc == 1 ) :
try :
self.__ups_handler.SetRWVar( ups=self.__current_ups, var=ups_var, value=new_val )
self.gui_status_message( _("Updated variable on %s") % self.__current_ups )
# Change the value on the local dict to update the GUI
self.__ups_vars[ups_var] = new_val
self.__ups_rw_vars[ups_var] = new_val
self.__gui_update_ups_vars_view()
except :
error_msg = _("Error updating variable on '{0}' ({1})").format( self.__current_ups, sys.exc_info()[1] )
self.gui_status_message( error_msg )
else :
# User cancelled modification...
error_msg = _("No variable modified on %s - User cancelled") % self.__current_ups
self.gui_status_message( error_msg )
except :
# Failed to get informations from the treeview... skip action
pass
#-------------------------------------------------------------------
# Refresh the content of the favorites menu according to the defined favorites
def __gui_refresh_favorites_menu( self ) :
for current in self.__fav_menu_items :
current.destroy()
self.__fav_menu_items = list()
items = self.__favorites.keys()
items.sort()
for current in items :
menu_item = gtk.MenuItem( current )
menu_item.show()
self.__fav_menu_items.append( menu_item )
self.__widgets["menu_favorites"].append( menu_item )
menu_item.connect_object( "activate", self.__gui_load_favorite, current )
if len( items ) > 0 :
self.__widgets["menu_favorites_del"].set_sensitive( True )
else :
self.__widgets["menu_favorites_del"].set_sensitive( False )
#-------------------------------------------------------------------
# In 'add favorites' dialog, this method compares the content of the
# text widget representing the name of the new favorite with existing
# ones. If they match, the 'add' button will be set to non sensitive
# to avoid creating entries with the same name.
def __gui_add_favorite_check_gui_fields( self, widget=None ) :
fav_name = widget.get_text()
if ( len( fav_name ) > 0 ) and ( fav_name not in self.__favorites.keys() ) :
self.__widgets["favorites_dialog_button_add"].set_sensitive( True )
else :
self.__widgets["favorites_dialog_button_add"].set_sensitive( False )
#-------------------------------------------------------------------
# Load and parse favorites
def __parse_favorites( self ) :
if ( not os.path.exists( self.__favorites_file ) ) :
# There is no favorites files, do nothing
return
try :
if ( not stat.S_IMODE( os.stat( self.__favorites_path ).st_mode ) == self.DESIRED_FAVORITES_DIRECTORY_MODE ) : # unsafe pre-1.2 directory found
os.chmod( self.__favorites_path, self.DESIRED_FAVORITES_DIRECTORY_MODE )
conf = ConfigParser.ConfigParser()
conf.read( self.__favorites_file )
for current in conf.sections() :
# Check if mandatory fields are present
if ( conf.has_option( current, "host" ) and conf.has_option( current, "ups" ) ) :
# Valid entry found, add it to the list
fav_data = {}
fav_data["host"] = conf.get( current, "host" )
fav_data["ups"] = conf.get( current, "ups" )
if ( conf.has_option( current, "port" ) ) :
fav_data["port"] = conf.get( current, "port" )
else :
fav_data["port"] = "3493"
# If auth is defined the section must have login and pass defined
if ( conf.has_option( current, "auth" ) ) :
if( conf.has_option( current, "login" ) and conf.has_option( current, "password" ) ) :
# Add the entry
fav_data["auth"] = conf.getboolean( current, "auth" )
fav_data["login"] = conf.get( current, "login" )
try :
fav_data["password"] = base64.decodestring( conf.get( current, "password" ) )
except :
# If the password is not in base64, let the field empty
print( _("Error parsing favorites, password for '%s' is not in base64\nSkipping password for this entry") % current )
fav_data["password"] = ""
else :
fav_data["auth"] = False
self.__favorites[current] = fav_data
self.__gui_refresh_favorites_menu()
except :
self.gui_status_message( _("Error while parsing favorites file (%s)") % sys.exc_info()[1] )
#-------------------------------------------------------------------
# Save favorites to the defined favorites file using ini format
def __save_favorites( self ) :
# If path does not exists, try to create it
if ( not os.path.exists( self.__favorites_file ) ) :
try :
os.makedirs( self.__favorites_path, mode=self.DESIRED_FAVORITES_DIRECTORY_MODE )
except :
self.gui_status_message( _("Error while creating configuration folder (%s)") % sys.exc_info()[1] )
save_conf = ConfigParser.ConfigParser()
for current in self.__favorites.keys() :
save_conf.add_section( current )
for k, v in self.__favorites[ current ].iteritems() :
save_conf.set( current, k, v )
try :
fh = open( self.__favorites_file, "w" )
save_conf.write( fh )
fh.close()
self.gui_status_message( _("Saved favorites...") )
except :
self.gui_status_message( _("Error while saving favorites (%s)") % sys.exc_info()[1] )
#-------------------------------------------------------------------
# Display the about dialog
def gui_about_dialog( self, widget=None ) :
dialog_interface = gtk.glade.XML( self.__glade_file, "aboutdialog1" )
dialog = dialog_interface.get_widget( "aboutdialog1" )
self.__widgets["main_window"].set_sensitive( False )
dialog.run()
dialog.destroy()
self.__widgets["main_window"].set_sensitive( True )
#-------------------------------------------------------------------
# Display a message on the status bar. The message is also set as
# tooltip to enable users to see long messages.
def gui_status_message( self, msg="" ) :
context_id = self.__widgets["status_bar"].get_context_id("Infos")
self.__widgets["status_bar"].pop( context_id )
if ( platform.system() == "Windows" ) :
text = msg.decode("cp1250").encode("utf8")
else :
text = msg
message_id = self.__widgets["status_bar"].push( context_id, text.replace("\n", "") )
self.__widgets["status_bar"].set_tooltip_text( text )
#-------------------------------------------------------------------
# Display a notification using PyNotify with an optional icon
def gui_status_notification( self, message="", icon_file="" ) :
# Try to init pynotify
try :
import pynotify
pynotify.init( "NUT Monitor" )
if ( icon_file != "" ) :
icon = "file://%s" % os.path.abspath( os.path.join( os.path.dirname( sys.argv[0] ), "pixmaps", icon_file ) )
else :
icon = None
notif = pynotify.Notification( "NUT Monitor", message, icon )
notif.show()
except :
pass
#-------------------------------------------------------------------
# Let GTK refresh GUI :)
def refresh_gui( self ) :
while gtk.events_pending() :
gtk.main_iteration( False )
return( True )
#-------------------------------------------------------------------
# Connect to the selected UPS using parameters (host,port,login,pass)
def connect_to_ups( self, widget=None ) :
host = self.__widgets["ups_host_entry"].get_text()
port = int( self.__widgets["ups_port_entry"].get_value() )
login = None
password = None
if self.__widgets["ups_authentication_check"].get_active() :
login = self.__widgets["ups_authentication_login"].get_text()
password = self.__widgets["ups_authentication_password"].get_text()
try :
self.__ups_handler = PyNUT.PyNUTClient( host=host, port=port, login=login, password=password )
except :
self.gui_status_message( _("Error connecting to '{0}' ({1})").format( host, sys.exc_info()[1] ) )
self.gui_status_notification( _("Error connecting to '{0}'\n{1}").format( host, sys.exc_info()[1] ), "warning.png" )
return
# Check if selected UPS exists on server...
srv_upses = self.__ups_handler.GetUPSList()
self.__current_ups = self.__widgets["ups_list_combo"].get_active_text()
if not srv_upses.has_key( self.__current_ups ) :
self.gui_status_message( _("Device '%s' not found on server") % self.__current_ups )
self.gui_status_notification( _("Device '%s' not found on server") % self.__current_ups, "warning.png" )
return
self.__connected = True
self.__widgets["ups_connect"].hide()
self.__widgets["ups_disconnect"].show()
self.__widgets["ups_infos"].show()
self.__widgets["ups_params_box"].set_sensitive( False )
self.__widgets["menu_favorites_root"].set_sensitive( False )
self.__widgets["ups_params_box"].hide()
commands = self.__ups_handler.GetUPSCommands( self.__current_ups )
self.__ups_commands = commands.keys()
self.__ups_commands.sort()
# Refresh UPS commands combo box
self.__widgets["ups_commands_combo_store"].clear()
for desc in self.__ups_commands :
self.__widgets["ups_commands_combo_store"].append( [ "%s\n<span color=\"#707070\">%s</span>" % ( desc, commands[desc] ) ] )
self.__widgets["ups_commands_combo"].set_active( 0 )
# Update UPS vars manually before the thread
self.__ups_vars = self.__ups_handler.GetUPSVars( self.__current_ups )
self.__ups_rw_vars = self.__ups_handler.GetRWVars( self.__current_ups )
self.__gui_update_ups_vars_view()
# Try to resize the main window...
self.__widgets["main_window"].resize( 1, 1 )
# Start the GUI updater thread
self.__gui_thread = gui_updater( self )
self.__gui_thread.start()
self.gui_status_message( _("Connected to '{0}' on {1}").format( self.__current_ups, host ) )
#-------------------------------------------------------------------
# Refresh UPS vars in the treeview
def __gui_update_ups_vars_view( self, widget=None ) :
if self.__ups_handler :
vars = self.__ups_vars
rwvars = self.__ups_rw_vars
self.__widgets["ups_vars_tree_store"].clear()
for k,v in vars.iteritems() :
if ( rwvars.has_key( k ) ) :
icon_file = "/usr/share/nut-monitor/pixmaps/var-rw.png"
else :
icon_file = "/usr/share/nut-monitor/pixmaps/var-ro.png"
icon = gtk.gdk.pixbuf_new_from_file( icon_file )
self.__widgets["ups_vars_tree_store"].append( [ icon, k, v ] )
#-------------------------------------------------------------------
# Disconnect from the UPS
def disconnect_from_ups( self, widget=None ) :
self.__connected = False
self.__widgets["ups_connect"].show()
self.__widgets["ups_disconnect"].hide()
self.__widgets["ups_infos"].hide()
self.__widgets["ups_params_box"].set_sensitive( True )
self.__widgets["menu_favorites_root"].set_sensitive( True )
self.__widgets["status_icon"].set_tooltip_markup( _("<i>Not connected</i>") )
self.__widgets["ups_params_box"].show()
# Try to resize the main window...
self.__widgets["main_window"].resize( 1, 1 )
# Stop the GUI updater thread
self.__gui_thread.stop_thread()
del self.__ups_handler
self.gui_status_message( _("Disconnected from '%s'") % self.__current_ups )
self.change_status_icon( "on_line", blink=False )
self.__current_ups = None
#-----------------------------------------------------------------------
# GUI Updater class
# This class updates the main gui with data from connected UPS
class gui_updater( threading.Thread ) :
__parent_class = None
__stop_thread = False
def __init__( self, parent_class ) :
threading.Thread.__init__( self )
self.__parent_class = parent_class
def run( self ) :
ups = self.__parent_class._interface__current_ups
was_online = True
# Define a dict containing different UPS status
status_mapper = { "LB" : "<span color=\"#BB0000\"><b>%s</b></span>" % _("Low batteries"),
"RB" : "<span color=\"#FF0000\"><b>%s</b></span>" % _("Replace batteries !"),
"BYPASS" : "<span color=\"#BB0000\">Bypass</span> <i>%s</i>" % _("(no battery protection)"),
"CAL" : _("Performing runtime calibration"),
"OFF" : "<span color=\"#000090\">%s</span> <i>(%s)</i>" % ( _("Offline"), _("not providing power to the load") ),
"OVER" : "<span color=\"#BB0000\">%s</span> <i>(%s)</i>" % ( _("Overloaded !"), _("there is too much load for device") ),
"TRIM" : _("Triming <i>(UPS is triming incoming voltage)</i>"),
"BOOST" : _("Boost <i>(UPS is boosting incoming voltage)</i>")
}
while not self.__stop_thread :
try :
vars = self.__parent_class._interface__ups_handler.GetUPSVars( ups )
self.__parent_class._interface__ups_vars = vars
# Text displayed on the status frame
text_left = ""
text_right = ""
status_text = ""
text_left += "<b>%s</b>\n" % _("Device status :")
if ( vars.get("ups.status").find("OL") != -1 ) :
text_right += "<span color=\"#009000\"><b>%s</b></span>" % _("Online")
if not was_online :
self.__parent_class.change_status_icon( "on_line", blink=False )
was_online = True
if ( vars.get("ups.status").find("OB") != -1 ) :
text_right += "<span color=\"#900000\"><b>%s</b></span>" % _("On batteries")
if was_online :
self.__parent_class.change_status_icon( "on_battery", blink=True )
self.__parent_class.gui_status_notification( _("Device is running on batteries"), "on_battery.png" )
was_online = False
# Check for additionnal informations
for k,v in status_mapper.iteritems() :
if vars.get("ups.status").find(k) != -1 :
if ( text_right != "" ) :
text_right += " - %s" % v
else :
text_right += "%s" % v
# CHRG and DISCHRG cannot be trated with the previous loop ;)
if ( vars.get("ups.status").find("DISCHRG") != -1 ) :
text_right += " - <i>%s</i>" % _("discharging")
elif ( vars.get("ups.status").find("CHRG") != -1 ) :
text_right += " - <i>%s</i>" % _("charging")
status_text += text_right
text_right += "\n"
if ( vars.has_key( "ups.mfr" ) ) :
text_left += "<b>%s</b>\n\n" % _("Model :")
text_right += "%s\n%s\n" % ( vars.get("ups.mfr",""), vars.get("ups.model","") )
if ( vars.has_key( "ups.temperature" ) ) :
text_left += "<b>%s</b>\n" % _("Temperature :")
text_right += "%s\n" % int( float( vars.get( "ups.temperature", 0 ) ) )
if ( vars.has_key( "battery.voltage" ) ) :
text_left += "<b>%s</b>\n" % _("Battery voltage :")
text_right += "%sv\n" % vars.get( "battery.voltage", 0 )
self.__parent_class._interface__widgets["ups_status_left"].set_markup( text_left[:-1] )
self.__parent_class._interface__widgets["ups_status_right"].set_markup( text_right[:-1] )
# UPS load and battery charge progress bars
if ( vars.has_key( "battery.charge" ) ) :
charge = vars.get( "battery.charge", "0" )
self.__parent_class._interface__widgets["progress_battery_charge"].set_fraction( float( charge ) / 100.0 )
self.__parent_class._interface__widgets["progress_battery_charge"].set_text( "%s %%" % int( float( charge ) ) )
status_text += "\n%s %s%%" % ( _("Battery charge :"), int( float( charge ) ) )
else :
self.__parent_class._interface__widgets["progress_battery_charge"].set_fraction( 0.0 )
self.__parent_class._interface__widgets["progress_battery_charge"].set_text( _("Not available") )
if ( vars.has_key( "ups.load" ) ) :
load = vars.get( "ups.load", "0" )
self.__parent_class._interface__widgets["progress_battery_load"].set_fraction( float( load ) / 100.0 )
self.__parent_class._interface__widgets["progress_battery_load"].set_text( "%s %%" % int( float( load ) ) )
status_text += "\n%s %s%%" % ( _("UPS load :"), int( float( load ) ) )
else :
self.__parent_class._interface__widgets["progress_battery_load"].set_fraction( 0.0 )
self.__parent_class._interface__widgets["progress_battery_load"].set_text( _("Not available") )
if ( vars.has_key( "battery.runtime" ) ) :
autonomy = int( float( vars.get( "battery.runtime", 0 ) ) )
if ( autonomy >= 3600 ) :
info = time.strftime( _("<b>%H hours %M minutes %S seconds</b>"), time.gmtime( autonomy ) )
elif ( autonomy > 300 ) :
info = time.strftime( _("<b>%M minutes %S seconds</b>"), time.gmtime( autonomy ) )
else :
info = time.strftime( _("<b><span color=\"#DD0000\">%M minutes %S seconds</span></b>"), time.gmtime( autonomy ) )
else :
info = _("Not available")
self.__parent_class._interface__widgets["ups_status_time"].set_markup( info )
# Display UPS status as tooltip for tray icon
self.__parent_class._interface__widgets["status_icon"].set_tooltip_markup( status_text )
except :
self.__parent_class.gui_status_message( _("Error from '{0}' ({1})").format( ups, sys.exc_info()[1] ) )
self.__parent_class.gui_status_notification( _("Error from '{0}'\n{1}").format( ups, sys.exc_info()[1] ), "warning.png" )
time.sleep( 1 )
def stop_thread( self ) :
self.__stop_thread = True
#-----------------------------------------------------------------------
# The main program starts here :-)
if __name__ == "__main__" :
# Init the localisation
APP = "NUT-Monitor"
DIR = "locale"
gettext.bindtextdomain( APP, DIR )
gettext.textdomain( APP )
_ = gettext.gettext
for module in ( gettext, gtk.glade ) :
module.bindtextdomain( APP, DIR )
module.textdomain( APP )
gui = interface()
gtk.main()
|
