#!/usr/bin/env python
"""
  fritzbox_connection_uptime - A munin plugin for Linux to monitor AVM Fritzbox connection uptime
  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 os
import sys

from fritzconnection import FritzConnection


def print_values():
    server = os.environ['fritzbox_ip']
    if 'fritzbox_port' in os.environ :
        port = os.environ['fritzbox_port']
    else :
        port = None
    if 'fritzbox_user' in os.environ :
        user = os.environ['fritzbox_user']
    else :
        user = None
    if 'fritzbox_password' in os.environ :
        password = os.environ['fritzbox_password']
    else:
        password = None

    try:
        conn = FritzConnection(address=server, port=port, user=user, password=password)
    except Exception as e:
        sys.exit("Couldn't get connection uptime", e)

    uptime = conn.call_action('WANIPConn', 'GetStatusInfo')['NewUptime']
    print('uptime.value %.2f' % (int(uptime) / 86400.0))


def print_config():
    print("graph_title AVM Fritz!Box Connection Uptime")
    print("graph_args --base 1000 -l 0")
    print('graph_vlabel uptime in days')
    print("graph_scale no'")
    print("graph_category network")
    print("uptime.label uptime")
    print("uptime.draw AREA")
    if os.environ.get('host_name'):
        print("host_name " + os.environ['host_name'])


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 connection uptime")