#!/usr/bin/env python

#max/min for the day with nodes

import psycopg2
import time

#run every hour

DB_PASS = "password"
DB_USER = "yggindex"
DB_NAME = "yggindex"
DB_HOST = "localhost"

# count peer alive if it was available not more that amount of seconds ago
# I'm using 1 hour beause of running crawler every 15 minutes
ALIVE_SECONDS = 3600 # 1 hour


def age_calc(ustamp):
	if (time.time() - ustamp) <= ALIVE_SECONDS :
		return True
	else:
		return False

def get_nodes_for_count():
	dbconn = psycopg2.connect(host=DB_HOST,database=DB_NAME, user=DB_USER, password=DB_PASS)
	cur = dbconn.cursor()
	nodes = {}
	cur.execute("select * from yggindex")

	for i in cur.fetchall():
		if age_calc(int(i[2])):
			nodes[i[0]] = [i[1],int(i[2])]

	cur.close()
	dbconn.close()

	return str(len(nodes))

def add_to_db():
	dbconn = psycopg2.connect(host=DB_HOST,database=DB_NAME, user=DB_USER, password=DB_PASS)
	cur = dbconn.cursor()

	cur.execute('''INSERT INTO timeseries(max, unixtstamp) VALUES(''' + "'" + get_nodes_for_count() + "'," + str(int(time.time())) + ''')''')

	dbconn.commit()
	cur.close()
	dbconn.close()

add_to_db()