Сделал индикатор для NUTа, может, кому пригодится поделие

[Роман (rvb)]

На входе - список хостов, на которых опрашивать UPSы. Если ничего не передано - ищет UPS только на локальном хосте.

Висит в области индикаторов, меняет иконку в зависимости от состояния, в разворачивающемся меню показывает состояние найденных UPSов, по клику - показывает выбранный детальнее. Поскольку UPSы у меня в окрестности простенькие, отображаю там мало что, но как список отображаемого (и отслеживаемого на предмет критичной ситуации) расширить - из кода, надеюсь, очевидно.

(Нажмите, чтобы показать/скрыть)

Код: [Выделить]

#!/usr/bin/env python

import sys
import gtk
import appindicator

import imaplib
import re

import os
import pynotify

PING_FREQUENCY = 10 # seconds

def exec_command(command):
  stream = os.popen(command)
  out = stream.read()
  out = re.split('\n',out);
  res = []
  for newline in out:
    if len(newline)==0:
      continue;
    elif newline.startswith("Error:"):
      res=[];
      break;
    else:
      res.append(newline)
  return res

class CheckNUT:

    def __init__(self):
        self.ind = appindicator.Indicator("nut-indicator",
                                          "gpm-ups-100-charging",                                       
                                  appindicator.CATEGORY_APPLICATION_STATUS)
        self.ind.set_status(appindicator.STATUS_ACTIVE)
        self.ind.set_attention_icon("gpm-ups-000")

hosts = sys.argv[1:]

if len(hosts)==0:
          hosts = {'localhost'}

        self.upses=[]
        for host in hosts :
          command = "upsc -l {0}".format(host)
          local_upses=exec_command(command)
          for ups in local_upses:
            self.upses.append('{0}@{1}'.format(ups,host))

        print self.upses
        self.menu_setup()
        self.ind.set_menu(self.menu)

        if not pynotify.init ("nut-indicator"):
          self.disable_notify=1
        else:
          self.disable_notify=0

        self.ups_dictionary = {}
        self.ups_dictionary['battery charge'] = 'Battery charge %'
        self.ups_dictionary['input voltage'] = 'Input voltage'
        self.ups_dictionary['output voltage'] = 'Output voltage'
        self.ups_dictionary['ups load'] = 'UPS load %'
        self.ups_dictionary['ups status'] = 'UPS status'

        self.ups_value_dictionary = {}
        self.ups_value_dictionary['ups status'] = {}
        self.ups_value_dictionary['ups status']['OL'] = {1,'online'}
        self.ups_value_dictionary['ups status']['OB'] = {0,'on battery'}
        self.ups_value_dictionary['ups status']['LB'] = {-1,'low batter'}

    def menu_setup(self):
        self.menu = gtk.Menu()

        self.menuitems = {}
        for ups in self.upses:
          print ups;
          menu_item = gtk.MenuItem(ups)
          menu_item.connect("activate",self.call_check_local_ups,ups)
          menu_item.show()
          self.menu.append(menu_item)
          self.menuitems[ups]=menu_item

        separator = gtk.SeparatorMenuItem()
        separator.show()
        self.menu.append(separator)

        self.quit_item = gtk.MenuItem("Quit")
        self.quit_item.connect("activate", self.quit)
        self.quit_item.show()
        self.menu.append(self.quit_item)

    def main(self):
        self.check_ups()
        gtk.timeout_add(PING_FREQUENCY * 1000, self.check_ups)
        gtk.main()

    def quit(self, widget):
        sys.exit(0)

    def call_check_local_ups(self,item,ups):
        return self.check_local_ups(ups,1)

    def check_local_ups(self,ups,notify):
        command = "upsc {0}".format(ups)
        res = exec_command(command)
        status="unknown"
        battery="unknown"
        text=""
        minstatus=1
        for newline in res:
          paramname,paramvalue = newline.split(': ',1)
          paramname = paramname.replace('.',' ')
          try:
            status,paramvalue = self.ups_value_dictionary[paramname][paramvalue]
            if status < minstatus:
              status=minstatus
          except:
            pass
          try:
            printname = self.ups_dictionary[paramname]
            text="{0}\n{1} = {2}".format(text,printname,paramvalue)
          except:
            pass
           
          if paramname=="battery charge":
            battery=paramvalue
          elif paramname=="ups status":
            ups_status=paramvalue

        self.menuitems[ups].get_child().set_text('{0}: {1}, {2}%'.format(ups,ups_status,battery))
        if (notify == 1 or minstatus < 0) and self.disable_notify==0:
          n = pynotify.Notification(ups,text)
          helper = gtk.Button()
          icon = helper.render_icon(gtk.STOCK_DIALOG_INFO, gtk.ICON_SIZE_DIALOG)
          n.set_icon_from_pixbuf(icon)
          n.show()
        return minstatus

    def check_ups(self):

        min_status=1;

        for ups in self.upses:
          status=self.check_local_ups(ups,0)
          if status < min_status:
            min_status=status

        if status < 0:
            self.ind.set_status(appindicator.STATUS_ATTENTION)
            self.ind.set_attention_icon("gpm-ups-000")
        elif status==0:
            self.ind.set_status(appindicator.STATUS_ATTENTION)
            self.ind.set_attention_icon("gpm-ups-060")
        else:
            self.ind.set_status(appindicator.STATUS_ACTIVE)
        return True

if __name__ == "__main__":
    indicator = CheckNUT()
    indicator.main()


[tremor]

У меня вот нет никаких упсов , видимо self.upses пустая и переменная status не создается. поэтому при запуске ошибки:

./nut.py
sh: upsc: not found
[]
Traceback (most recent call last):
  File "./nut.py", line 162, in <module>
    indicator.main()
  File "./nut.py", line 95, in main
    self.check_ups()
  File "./nut.py", line 150, in check_ups
    if status < 0:
UnboundLocalError: local variable 'status' referenced before assignment


Команды upsc у меня тоже нет и вообще exec_command это нехорошо, надо было средствами питона, или как то предусматривать ошибку...

Хотя наверное задача довести до продакшена и не ставилась )) Скажи, где брал мануал по appindictor ?

[Роман (rvb)]

Мануал нигде не брал, смотрел исходники hardware sensor indicator.

Если nut стоит - upsc тоже есть, если nut не стоит, то какой нафиг nut-indicator?

На всякий случай - достаточно обрамить try: ... except: pass вызов upsc - этого будет достаточно, чтобы программа не ругалась. Лезть на уровне сокетов в протокол NUT не хочу по идейным соображениям .