#!/usr/bin/python

import os
import sys
import string
import time
from datetime import datetime,timedelta

from monitor import database
from monitor import parser as parsermodule
from monitor import config
from monitor.database.info.model import HistorySiteRecord, HistoryNodeRecord, session
from monitor.wrapper import plc, plccache
from monitor.const import MINUP

from monitor.common import *
from nodequery import verify,query_to_dict,node_select
from monitor.model import *

api = plc.getAuthAPI()
def main():
	main2(config)

def main2(config):

	l_nodes = plccache.l_nodes
	l_plcsites = plccache.l_sites

	if config.site:
		l_sites = [config.site]
	elif config.sitelist:
		site_list = config.sitelist.split(',')
		l_sites = site_list
	else:
		l_sites = [site['login_base'] for site in l_plcsites]
	
	checkAndRecordState(l_sites, l_plcsites)

def getnodesup(nodelist):
	up = 0
	for node in nodelist:
		try:
			nodehist = HistoryNodeRecord.findby_or_create(hostname=node['hostname'])
			if nodehist is not None and nodehist.status == "good":
				up = up + 1
		except:
			import traceback
			print traceback.print_exc()
	return up

def check_site_state(rec, sitehist):

	if sitehist.new and sitehist.status != 'new':
		sitehist.status = 'new'
		sitehist.last_changed = datetime.now()

	if not sitehist.new:

		if sitehist.nodes_up >= MINUP:

			if sitehist.status != 'online' and sitehist.status != 'good':
				sitehist.last_changed = datetime.now()

			if changed_lessthan(sitehist.last_changed, 0.5) and sitehist.status != 'online':
				print "changed status from %s to online" % sitehist.status
				sitehist.status = 'online'

			if changed_greaterthan(sitehist.last_changed, 0.5) and sitehist.status != 'good':
				print "changed status from %s to good" % sitehist.status
				sitehist.status = 'good'
	
		else: # sitehist.nodes_up < MINUP:

			if sitehist.status != 'offline' and sitehist.status != 'down':
				sitehist.last_changed = datetime.now()

			if changed_lessthan(sitehist.last_changed, 0.5) and sitehist.status != 'offline':
				print "changed status from %s to offline" % sitehist.status
				sitehist.status = 'offline'

			if changed_greaterthan(sitehist.last_changed, 0.5) and sitehist.status != 'down':
				print "changed status from %s to down" % sitehist.status
				sitehist.status = 'down'

def checkAndRecordState(l_sites, l_plcsites):
	count = 0
	lb2hn = plccache.plcdb_lb2hn
	for sitename in l_sites:
		d_site = None
		for site in l_plcsites:
			if site['login_base'] == sitename:
				d_site = site
				break
		if not d_site:
			continue

		if sitename in lb2hn:
			sitehist = HistorySiteRecord.findby_or_create(loginbase=sitename,
												if_new_set={'status' : 'unknown', 
															'last_changed' : datetime.now(),
															'message_id': 0,
															'penalty_level' : 0})
			sitehist.last_checked = datetime.now()

			sitehist.slices_total = d_site['max_slices']
			sitehist.slices_used = len(d_site['slice_ids'])
			sitehist.nodes_total = len(lb2hn[sitename])
			if sitehist.message_id != 0:
				rtstatus = mailer.getTicketStatus(sitehist.message_id)
				sitehist.message_status = rtstatus['Status']
				sitehist.message_queue = rtstatus['Queue']
				sitehist.message_created = datetime.fromtimestamp(rtstatus['Created'])

			sitehist.nodes_up = getnodesup(lb2hn[sitename])
			sitehist.new = changed_lessthan(datetime.fromtimestamp(d_site['date_created']), 30) # created < 30 days ago
			sitehist.enabled = d_site['enabled']

			check_site_state(d_site, sitehist)

			count += 1
			print "%d %15s slices(%2s) nodes(%2s) up(%2s) %s" % (count, sitename, sitehist.slices_used, 
											sitehist.nodes_total, sitehist.nodes_up, sitehist.status)
			sitehist.flush()

	print HistorySiteRecord.query.count()
	session.flush()

	return True

if __name__ == '__main__':
	from monitor import parser as parsermodule

	parser = parsermodule.getParser()
	parser.set_defaults(filename=None, node=None, site=None, 
						nodeselect=False, nodegroup=None, cachenodes=False)

	parser.add_option("", "--site", dest="site", metavar="login_base", 
						help="Provide a single site to operate on")
	parser.add_option("", "--sitelist", dest="sitelist", 
						help="Provide a list of sites separated by ','")

	config = parsermodule.parse_args(parser)

	try:
		main2(config)
	except Exception, err:
		import traceback
		print traceback.print_exc()
		print "Exception: %s" % err
		sys.exit(0)