#!/usr/bin/env python

# By Arne Schwabe <arne-nagios@rfc2549.org>
# LICENSE: BSD

from subprocess import Popen,PIPE
import sys
import time
import os
import argparse

def main():
    parser = argparse.ArgumentParser(description='Nagios Duplicity status checker')

    parser.add_argument("-w", dest="warninc", default=28, type=int, 
                        help="Number of hours allowed for incremential backup warning level")
    parser.add_argument("-W", dest="warnfull", default=40, type=int, 
                        help="Number of hours allowed for incremential backup critical level")

    parser.add_argument("-c", dest="critinc", default=52, type=int, 
                        help="Number of days allowed for full backup warning level")

    parser.add_argument("-C", dest="critfull", default=60, type=int, 
                        help="Number of days allowed for full backup critical level")

    parser.add_argument("-P", dest="profile", default="backup", type=str, 
                        help="duply profile name")

    args = parser.parse_args()
    
    okay = 0

    output , err = Popen(["sudo", "duply", args.profile, "status"], stdout=PIPE, stderr=PIPE , env={'HOME': '/root', 'PATH': os.environ['PATH']}).communicate()

    #output = Popen(["/usr/local/bin/sudo", "/usr/local/sbin/duply", args.profile, "status"], stdout=PIPE, env={'HOME': '/root', 'PATH': os.environ['PATH']}).communicate()[0]


    lastfull, lastinc = findlastdates(output)


    sincelastfull = time.time() - lastfull 
    sincelastinc  =  time.time() - lastinc 

    msg = "OK: "
    
    if sincelastfull > (args.warnfull * 24 * 3600) or sincelastinc > (args.warninc * 3600):
        okay = 1
        msg = "WARNING: "
    
    if sincelastfull > (args.critfull * 24 * 3600) or sincelastinc > (args.critinc * 3600):
        okay = 2
        msg = "CRITICAL: "

    if not checkoutput(output):
        okay = max(okay,1)
        msg = "WARNING: duplicity output: %s " % repr(output)


    if err:
        okay=2
        msg = "Unexpected output: %s, " % repr(err)

    print msg, "last full %s ago, last incremential %s ago|lastfull=%d, lastinc=%d" % ( formattime(sincelastfull), formattime(sincelastinc), sincelastfull, sincelastinc)
    sys.exit(okay)

def checkoutput(output):
    if output.find("No orphaned or incomplete backup sets found.")==-1:
        return False

    return True

def formattime(seconds):
    days = seconds / (3600 * 24)
    hours = seconds / 3600 % 24

    if days:
        return "%d days %d hours" % (days,hours)
    else:
        return "%d hours" % hours


def findlastdates(output):
    lastfull =0
    lastinc = 0

    for line in output.split("\n"):
        parts = line.split()

        # ['Incremental', 'Sun', 'Oct', '31', '03:00:04', '2010', '1']
        if len (parts) == 7 and parts[0] in ["Full","Incremental"]:
            foo = time.strptime(" ".join(parts[1:6]),"%a %b %d %H:%M:%S %Y")

    
            backuptime =  time.mktime(foo)
    
            if parts[0] == "Incremental" and lastinc < backuptime:
                lastinc = backuptime
            elif parts[0] == "Full" and lastfull < backuptime:
                lastfull = backuptime
        

    # Count a full backup as incremental backup
    lastinc = max(lastfull,lastinc)
    return (lastfull, lastinc)
            
        
	

if __name__=='__main__':
   main()