yggdrasil-api/api/niflheim-api.py

import time
from flask import Flask, render_template
from flask_restful import Resource, Api
import requests
import psycopg2

app = Flask(__name__)
api = Api(app)

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

#will add this as something seperate and pull down to a file rather
#than request it everytime.
def get_nodelist():
    data = requests.get("use the raw view of the github nodelist", timeout=1)
    nodes = [x.split() for x in data.text.split('\n') if x]

    index_table = {}

    for key in nodes:
        index_table[key[0]] = key[1]
    return index_table


def check_nodelist(nodetable, key):
    if nodetable:
        if nodetable.get(key):
            return nodetable.get(key)
        else:
            return key
    else:
        return key


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

#active nodes in the past 4hrs
class nodesCurrent(Resource):
    def get(self):
        dbconn = psycopg2.connect(host=DB_HOST,\
                                    database=DB_NAME,\
                                    user=DB_USER,\
                                    password=DB_PASSWORD)
        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])]

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

        nodelist = {}
        nodelist['yggnodes'] = nodes

        return nodelist


@app.route("/")
def fpage():
    dbconn = psycopg2.connect(host=DB_HOST,\
                                database=DB_NAME,\
                                user=DB_USER,\
                                password=DB_PASSWORD)
    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])]

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

    return render_template('index.html', nodes=str(len(nodes)))


@app.route("/contrib")
def cpage():
    try:
        domain_nodelist = get_nodelist()
        print "list exists"
    except:
        print "failed"
        domain_nodelist = None

    dbconn = psycopg2.connect(host=DB_HOST,\
                                database=DB_NAME,\
                                user=DB_USER,\
                                password=DB_PASSWORD)
    cur = dbconn.cursor()
    cur.execute("select * from contrib")
    nodes = []

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

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

    dnodes = []
    for key in nodes:
        dnodes.append(check_nodelist(domain_nodelist, key))

    dnodes.sort(reverse=True)

    return render_template('contrib.html', contribnodes=dnodes, nocontribs=str(len(dnodes)))


#sort out the api request here for the url
api.add_resource(nodesCurrent, '/current')

if __name__ == '__main__':
    app.run(host='::', port=3000)