#!/usr/bin/python

 import dflowlog

 from dflowlog import dflow_fields as fields

 from filter import filter

 import sys

 from datetime import datetime

 from datetime import timedelta

 from optparse import OptionParser

 def get_datetime(row, value):

 	date, time = row[fields[value]].split()

 	year, month, day = date.split("-")

 	hour, minute, second = time.split(":")

 	return datetime(int(year), int(month), int(day), int(hour), int(minute), int(second))

 def time_diff(dt1, dt2):	

 	diff = (max(dt1,dt2) - min(dt1, dt2))

 	diff = diff.seconds / 60 + diff.days * 24 * 60

 	if dt1 < dt2:

 		diff *= -1

 	return diff

 def diff(values, plans):

 	new_plans = []

 	for plan in plans:

 		new_plans.append(plan)

 		v1 = get_datetime(plan, values[0])

 		v2 = get_datetime(plan, values[1])

 		new_plans[-1].append(str(time_diff(v1, v2)))

 	return new_plans

 def column_diff(values, plans):

 	new_plans = []

 	flow_points = []

 	for plan in plans:

 		flow_point = plan[fields["flow_point"]]

 		if flow_point not in flow_points:

 			flow_points.append(flow_point)

 	for fp in flow_points:

 		flow_plans = filter("flow_point", fp, plans)		

 		for plan in flow_plans:

 			min_sep = 999

 			current_plan = plan

 			current_flow_time = get_datetime(current_plan, values[0])			

 			for plan2 in flow_plans:

 				if plan2 is not current_plan:

 					min_sep = min(min_sep, 

 									abs(time_diff(get_datetime(plan2, values[0]),

 									current_flow_time)))				

 			if min_sep < 999:

 				plan.append(str(min_sep))

 			new_plans.append(plan)

 	return new_plans

 if __name__ == "__main__":

 	modes = {"diff": (diff, 3),

 				"cdiff": (column_diff, 2)}

 	arg_parser = OptionParser(usage="usage: %prog [options] diff column1 column2 csv_logs\

 									 %prog [options] cdiff column csv_logs")

 	arg_parser.add_option("-f", "--files",

 				action="store_true", dest="files_mode",

 				help="csv_logs are files")

 	arg_parser.set_defaults(files_mode=False)

 	(options, args) = arg_parser.parse_args()

 	plans = []

 	if options.files_mode:

 		plans =	dflowlog.parse_files(args[modes[args[0]][1]:])

 		options.files_mode = False

 	else:

 		plans = dflowlog.parse_pipe(args[modes[args[0]][1]:])

 	plans = modes[args[0]][0](args[1:], plans)	

 	dflowlog.print_logs(plans)