Munin contributed plugins

#!/usr/bin/env python

# -*- coding: utf-8 -*-

"""

: << =cut

=head1 NAME

transmission - Munin plugin to monitor Transmission bittorrent daemon

=head1 DESCRIPTION

This plugin implements the multigraph protocol and provides the following graphs

  transmission_throughput - monitor traffic volumes of Transmission torrents

  transmission_activity - plugin to monitor traffic speed of Transmission torrents

This plugin requires python and the transmissionrpc python module.

See http://pypi.python.org/pypi/transmissionrpc/

=head1 CONFIGURATION

    [transmission]

        env.host 10.0.0.1

        env.port 9093

        env.user transmission

        env.pass secret

    [transmission_*]

        env.host 10.0.0.1

        env.port 9093

        env.user transmission

        env.pass secret

=head1 AUTHOR

Thomas Léveil

=head1 LICENSE

Permission to use, copy, and modify this software with or without fee

is hereby granted, provided that this entire notice is included in

all source code copies of any software which is or includes a copy or

modification of this software.

THIS SOFTWARE IS BEING PROVIDED "AS IS", WITHOUT ANY EXPRESS OR

IMPLIED WARRANTY. IN PARTICULAR, NONE OF THE AUTHORS MAKES ANY

REPRESENTATION OR WARRANTY OF ANY KIND CONCERNING THE

MERCHANTABILITY OF THIS SOFTWARE OR ITS FITNESS FOR ANY PARTICULAR

PURPOSE.

=head1 CONTRIBUTE

find this plugin on github at http://github.com/VolatileMesh/munin-plugins

=head1 MAGIC MARKERS

 #%# family=auto contrib

 #%# capabilities=autoconf

=head1 VERSION

    1.1

=head1 CHANGELOG

=head2 1.0 - 2010/11/12

    first release

=head2 1.1 - 2011/05/29

    fix transmission error handling

=cut

"""

__version__ = '1.1'

import os, sys

from string import Template

plugin_name=list(os.path.split(sys.argv[0]))[1]

host = os.getenv('host','localhost')

port = os.getenv('port',9091)

user = os.getenv('user')

passwd = os.getenv('pass')

def config():

    conf = Template("""multigraph ${plugin_name}_throughput

graph_title Transmission throughput for ${host}

graph_vlabel bytes/${graph_period} in (-) / out (+)

graph_args --base 1000

graph_category network

graph_info This graph shows the throughput for Transmission torrents

down.label throughput

down.type COUNTER

down.draw LINE1

down.min 0

down.graph no

up.label sent

up.negative down

up.type COUNTER

up.draw LINE1

up.min 0

multigraph ${plugin_name}_activity

graph_title Transmission activity for ${host}

graph_vlabel torrents

graph_args --base 1000

graph_category network

graph_info This graph shows the number of Transmission torrents

total.label total

total.draw AREA

total.min 0

total.colour AFE3FF

active.label active

active.draw AREA

active.min 0

active.colour 77FF6F

paused.label paused

paused.draw LINE1

paused.min 0

paused.colour 8F8F8F

""")

    print conf.safe_substitute(plugin_name=plugin_name, host=host)

    sys.exit(0)

def autoconf():

    try:

        import transmissionrpc

        print('yes')

    except ImportError:

        print 'no python module \'transmissionrpc\' missing'

def fetch():

    import transmissionrpc

    try:

        client = transmissionrpc.Client(host, port=port, user=user, password=passwd)

    except transmissionrpc.TransmissionError, err:

        print err

        sys.exit(1)

    stats = client.session_stats(10)

    print_values_throughput(stats)

    print_values_activity(stats)

def print_values_activity(stats):

    print "multigraph {plugin_name}_activity".format(plugin_name=plugin_name)

    try:

        print "total.value %s" % stats.torrentCount

    except:

        print "total.value U"

    try:

        print "active.value %s" % stats.activeTorrentCount

    except:

        print "active.value U"

    try:

        print "paused.value %s" % stats.pausedTorrentCount

    except:

        print "paused.value U"

def print_values_throughput(stats):

    print "multigraph {plugin_name}_throughput".format(plugin_name=plugin_name)

    try:

        print "down.value %s" % stats.cumulative_stats['downloadedBytes']

    except:

        print "down.value U"

    try:

        print "up.value %s" % stats.cumulative_stats['uploadedBytes']

    except:

        print "up.value U"

def dumpstats():

    import transmissionrpc

    try:

        client = transmissionrpc.Client(host, port=port, user=user, password=passwd)

    except transmissionrpc.transmission.TransmissionError, err:

        print err

        sys.exit(1)

    stats = client.session_stats(10)

    print stats

if __name__ == '__main__':

    if len(sys.argv)>1 :

        if sys.argv[1]=="dumpstats" :

			dumpstats()

        elif sys.argv[1]=="config" :

            config()

        elif sys.argv[1]=="autoconf" :

            autoconf()

        elif sys.argv[1]!="":

            raise ValueError, "unknown parameter '%s'" % sys.argv[1]

fetch()

sys.exit(0)