fritzbox_power_consumption.py

#!/usr/bin/env python
"""
  fritzbox_power_consumption - A munin plugin for Linux to monitor AVM Fritzbox
  Copyright (C) 2015 Christian Stade-Schuldt
  Author: Christian Stade-Schuldt
  Like Munin, this plugin is licensed under the GNU GPL v2 license
  http://www.opensource.org/licenses/GPL-2.0
  Add the following section to your munin-node's plugin configuration:

  [fritzbox_*]
  env.fritzbox_ip [ip address of the fritzbox]
  env.fritzbox_password [fritzbox password]
  
  This plugin supports the following munin configuration parameters:
  #%# family=auto contrib
  #%# capabilities=autoconf
"""

import os
import re
import sys
import fritzbox_helper as fh

PAGE = '/system/energy.lua'
pattern = re.compile(".*/(.*act?)\".*=.*\"(.*?)\"")
DEVICE_MAPPING = {'rate_abact': 'ab', 'rate_dspact': 'dsl', 'rate_sumact': 'system',
                  'rate_systemact': 'cpu', 'rate_usbhostact': 'usb', 'rate_wlanact': 'wifi'}


def get_power_consumption():
    """get the current power consumption usage"""

    server = os.environ['fritzbox_ip']
    password = os.environ['fritzbox_password']

    sid = fh.get_sid(server, password)
    data = fh.get_page(server, sid, PAGE)
    matches = re.finditer(pattern, data)
    if matches:
        for m in matches:
            print ( '%s.value %d' % (DEVICE_MAPPING[m.group(1)], int(m.group(2))) )


def print_config():
    print ( "graph_title AVM Fritz!Box Power Consumption" )
    print ( "graph_vlabel %" )
    print ( "graph_category network" )
    print ( "graph_order system cpu wifi dsl ab usb" )
    print ( "system.label system" )
    print ( "system.type GAUGE" )
    print ( "system.graph LINE12" )
    print ( "system.min 0" )
    print ( "system.max 100" )
    print ( "system.info Fritzbox overall power consumption" )
    print ( "cpu.label cpu" )
    print ( "cpu.type GAUGE" )
    print ( "cpu.graph LINE1" )
    print ( "cpu.min 0" )
    print ( "cpu.max 100" )
    print ( "cpu.info Fritzbox central processor power consumption" )
    print ( "wifi.label wifi" )
    print ( "wifi.type GAUGE" )
    print ( "wifi.graph LINE1" )
    print ( "wifi.min 0" )
    print ( "wifi.max 100" )
    print ( "wifi.info Fritzbox wifi power consumption" )
    print ( "dsl.label dsl" )
    print ( "dsl.type GAUGE" )
    print ( "dsl.graph LINE1" )
    print ( "dsl.min 0" )
    print ( "dsl.max 100" )
    print ( "dsl.info Fritzbox dsl power consumption" )
    print ( "ab.label ab" )
    print ( "ab.type GAUGE" )
    print ( "ab.graph LINE1" )
    print ( "ab.min 0" )
    print ( "ab.max 100" )
    print ( "ab.info Fritzbox analog phone ports power consumption" )
    print ( "usb.label usb" )
    print ( "usb.type GAUGE" )
    print ( "usb.graph LINE1" )
    print ( "usb.min 0" )
    print ( "usb.max 100" )
    print ( "usb.info Fritzbox usb devices power consumption" )


if __name__ == '__main__':
    if len(sys.argv) == 2 and sys.argv[1] == 'config':
        print_config()
    elif len(sys.argv) == 2 and sys.argv[1] == 'autoconf':
        print ( 'yes' )
    elif len(sys.argv) == 1 or len(sys.argv) == 2 and sys.argv[1] == 'fetch':
        # Some docs say it'll be called with fetch, some say no arg at all
        try:
            get_power_consumption()
        except:
            sys.exit("Couldn't retrieve fritzbox power consumption")