fritzbox_traffic.py

#!/usr/bin/env python
"""
  fritzbox_traffic - A munin plugin for Linux to monitor AVM Fritzbox WAN traffic
  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
  Like Munin, this plugin is licensed under the GNU GPL v2 license
  http://www.opensource.org/licenses/GPL-2.0
  This plugin requires the fritzconnection plugin. To install it using pip:
  pip install fritzconnection
  This plugin supports the following munin configuration parameters:
  #%# family=auto contrib
  #%# capabilities=autoconf
"""

import sys

from fritzconnection import FritzConnection


def print_values():
    try:
        conn = FritzConnection()
    except Exception as e:
        sys.exit("Couldn't get WAN traffic")

    down_traffic = conn.call_action('WANCommonInterfaceConfig', 'GetTotalBytesReceived')['NewTotalBytesReceived']
    print ('down.value %d' % down_traffic)

    up_traffic = conn.call_action('WANCommonInterfaceConfig', 'GetTotalBytesSent')['NewTotalBytesSent']
    print ('up.value %d' % up_traffic)

    max_down_traffic = conn.call_action('WANCommonInterfaceConfig', 'GetCommonLinkProperties')[
        'NewLayer1DownstreamMaxBitRate']
    print ('maxdown.value %d' % max_down_traffic)

    max_up_traffic = conn.call_action('WANCommonInterfaceConfig', 'GetCommonLinkProperties')[
        'NewLayer1UpstreamMaxBitRate']
    print ('maxup.value %d' % max_up_traffic)


def print_config():
    print ( "graph_title AVM Fritz!Box WAN traffic" )
    print ( "graph_args --base 1000" )
    print ( "graph_vlabel bits in (-) / out (+) per \${graph_period}" )
    print ( "graph_category network" )
    print ( "graph_order down up maxdown maxup" )
    print ( "down.label received" )
    print ( "down.type DERIVE" )
    print ( "down.graph no" )
    print ( "down.cdef down,8,*" )
    print ( "down.min 0" )
    print ( "down.max 1000000000" )
    print ( "up.label bps" )
    print ( "up.type DERIVE" )
    print ( "up.draw AREA" )
    print ( "up.cdef up,8,*" )
    print ( "up.min 0" )
    print ( "up.max 1000000000" )
    print ( "up.negative down" )
    print ( "up.info Traffic of the WAN interface." )
    print ( "maxdown.label received" )
    print ( "maxdown.type GAUGE" )
    print ( "maxdown.graph no" )
    print ( "maxup.label MAX" )
    print ( "maxup.type GAUGE" )
    print ( "maxup.negative maxdown" )
    print ( "maxup.draw LINE1" )
    print ( "maxup.info Maximum speed of the WAN interface." )


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" ) # Some docs say it'll be called with fetch, some say no arg at all
    elif len(sys.argv) == 1 or (len(sys.argv) == 2 and sys.argv[1] == 'fetch'):
        try:
            print_values()
        except:
            sys.exit("Couldn't retrieve fritzbox traffic")