Others » GDP infrastructure visualization

#!/usr/bin/env python

import MySQLdb

import base64, string

#mysql database information that will be derived via ingest.sh

HOST =

PORT =

USER =

PASS =

DB =

db = None

cursor = None

#executes the query and returns the result

def execute(query):

    global db

    global cursor

    if db is None:

        db = MySQLdb.connect(host=HOST, port=PORT, user=USER, passwd=PASS, db=DB)

        cursor = db.cursor()

    try:

        cursor.execute(query)

        db.commit()

        ret = cursor.fetchall()

    except:

        db.rollback()

        ret = None

        print "Failed query"

    return ret

#from hex output (that we don't use elsewhere), get a base64 version

def gdp_printable_name(hexVal):

    binary = base64.b16decode(hexVal.upper())

    stdb64 = base64.b64encode(binary)

    # convert to filesystem safe version

    return stdb64.translate(string.maketrans("/+","_-"), "=")

#takes in a single 'item' from the list of primary nodes as input.

#parse the dictionary to get the public IP of primary

def parse_primary(d):

    s = d["NodeID (IP:port)"]

    return s.split(":")[0]

#takes in a single 'item' from the list of secondary nodes as input

#ideally, instead of 'Secondary Nodes', it should be called:

#">bunch of secondary nodes that are behind same public IP<"

#parses the maze, and return a list of pub/priv pairs

def parse_secondary(d):

    pubIP = d["NATIP"]

    nodes = d["Nodes"]

    ret = []

    for n in nodes:

        tmp = n["NodeID (IP:port)"]

        privIP = tmp.split(":")[0]

        ret.append((pubIP, privIP))

    return ret