yggdrasil-api/yggapi.py

from flask import Flask, request
from flask_restful import Resource, Api
from sqlalchemy import create_engine
from json import dumps
#rate limiting support
from flask.ext.jsonpify import jsonify
from flask_limiter import Limiter
from flask_limiter.util import get_remote_address
import time
import sys
import os

#check if a database exists or not
db_path = "vservdb/yggindex.db"
if not os.path.exists(db_path):
    print "could not find a database"
    sys.exit(0)

db_connect = create_engine('sqlite:///' + db_path)
app = Flask(__name__)
api = Api(app)

limiter = Limiter(
    app,
    key_func=get_remote_address,
    default_limits=["500/day", "60/hour"]
)

#quickly figure out which is old or new
def age_calc(ustamp):
    if (time.time() - ustamp) <= 14400 :
        return True
    else:
        return False

#active nodes in the past 4hrs
class nodes_current(Resource):
    def get(self):
        conn = db_connect.connect()
        query = conn.execute("select * from yggindex")
        nodes = {}
        for i in query.cursor.fetchall():
            if age_calc(i[2]):
                nodes[i[0]] = [i[1],i[2]]
        nodelist = {}
        nodelist['yggnodes'] = nodes
        return nodelist

#nodes that may not be active anymore or have been offline for a while such as laptops
#could be used as last seen
class nodes_old(Resource):
    def get(self):
        conn = db_connect.connect()
        query = conn.execute("select * from yggindex")
        nodes = {}
        for i in query.cursor.fetchall():
            if not age_calc(i[2]):
                nodes[i[0]] = [i[1],i[2]]
        nodelist = {}
        nodelist['yggnodes'] = nodes
        return nodelist

#return entire database of nodes regardless of age
class nodes_all(Resource):
    def get(self):
        conn = db_connect.connect()
        query = conn.execute("select * from yggindex")
        nodes = {}
        for i in query.cursor.fetchall():
            nodes[i[0]] = [i[1],i[2]]
        nodelist = {}
        nodelist['yggnodes'] = nodes
        return nodelist

#sort out the api request here for the url
api.add_resource(nodes_current, '/current')
api.add_resource(nodes_old, '/old')
api.add_resource(nodes_all, '/all')

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