import csv
import os
from pdb import set_trace as debugger

MAIN_CODE_DIR = os.environ.get('MAIN_CODE_DIR', '')

def read_centrality_score(filepath):
    scores = dict()
    with open(filepath) as f:
        csv_reader = csv.reader(f)
        try:
            for row in csv_reader:
                if len(row) != 2:
                    print row
                scores[row[0]] = row[1]
        except Exception as e:
            print e
            return 0

    return scores

def sort_value_of_dictionary(dictionary):
    l = list()
    for k, v in dictionary.iteritems():
        l.append((k, v))

    l.sort(key=lambda tup: tup[1])
    return l


if __name__ == '__main__':
    filepath_prefixes = ['/output/networkx_', '/output/weight_basic_', '/output/heuristic_']
    types = ['edge_list']
    extension = '.csv'

    for t in types:
        # TODO: any other way to handle the relative path???
        filepaths = [MAIN_CODE_DIR + p + t + extension for p in filepath_prefixes]

        nx_scores = read_centrality_score(filepaths[0])
        weight_basic_scores = read_centrality_score(filepaths[1])
        heuristic_scores = read_centrality_score(filepaths[2])

        sorted_nx_scores = sort_value_of_dictionary(nx_scores)

        # write to file
        # FORMAT: ip_address    networkx_centrality     boost_centrality
        print "Compare"
        with open(MAIN_CODE_DIR + '/output/score_summary_%s.txt' % t, 'w') as output:
            for key, nx in sorted_nx_scores:
                weight_basic = weight_basic_scores[key]
                heuristic = heuristic_scores[key]
                output.write('%s\t%s\t%s\t%s\n' % (key, nx, weight_basic, heuristic))