#!/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)