#!/usr/bin/env python3

# -*- coding: utf-8 -*-

import argparse

import time

try:

import networkx as nx

except ImportError:

print('This script requires NetworkX library to be installed.')

exit(1)

import graph.io as io

import graph.stats as stats

import graph.predicates as predicates

import utils.print as gprint

def gen_username(first_name, last_name, uid):

return '{} {} ({})'.format(first_name, last_name, uid)

def add_rows(table_data, table):

for username in table_data:

table_data[username].insert(0, username)

table.add_row(table_data[username])

def top_of_table(table_str, num_newlines):

return table_str[:cur_index + 1] + first_line

def append_central_nodes(graph, table_data):

table_data[node_name][-1] = 'True'

def append_periphery_nodes(graph, table_data):

table_data[node_name][-1] = 'True'

def append_degree(graph, table_data):

table_data[node_name] = [node_degree]

def append_num_friends(graph, table_data):

table_data[node_name] = [num_friends]

def append_num_followers(graph, table_data):

table_data[node_name] = [num_friends]

def append_pagerank(graph, table_data):

table_data[node_name] = [pagerank]

def append_media_activist(graph, table_data):

table_data[node_name] = [is_activist]

# list of implemented fields

impl_fields = [

]

DESCRIPTION = 'Print characteristics of specified NetworkX graph'

parser = argparse.ArgumentParser(description=DESCRIPTION)

parser.add_argument('path', metavar='PATH', type=str,

help='path to YAML file which contains graph data')

parser.add_argument('-i', '--info',

help='print general information about graph',

action='store_true')

parser.add_argument('-r', '--radius', help='print graph radius',

action='store_true')

parser.add_argument('-d', '--diameter', help='print graph diameter',

action='store_true')

parser.add_argument('--avg-friends', help='print average number of friends',

action='store_true')

parser.add_argument('--avg-followers', action='store_true',

help='print average number of followers')

impl_headers = ','.join([field['header'] for field in impl_fields])

parser.add_argument('-f', '--fields', metavar='FIELDS', type=str,

help='print specified fields of each node:\n'

'{}'.format(impl_headers))

parser.add_argument('-s', '--sort', metavar='FIELD', type=str,

help='sort by FIELD from specified FIELDS')

parser.add_argument('-t', '--top', metavar='NUM_USERS', type=int,

help='print only top NUM_USERS')

args = parser.parse_args()

try:

start_time = time.time()

G = io.read_graph(args.path)

if args.info:

print(nx.info(G), '\n')

if args.radius:

print('Graph radius: ', nx.radius(G))

if args.diameter:

print('Graph diameter: ', nx.diameter(G))

if args.avg_friends:

print('Average number of friends: ', stats.avg_num_friends(G))

if args.avg_followers:

print('Average number of followers: ', stats.avg_num_followers(G))

if args.fields:

try:

from prettytable import PrettyTable

except ImportError:

print('This script requires PrettyTable library to be installed.')

exit(1)

args_fields = args.fields.split(',')

for i, field in enumerate(args_fields):

args_fields[i] = field.lower().strip()

table_headers = ['Username']

table_align = {'Username': 'l'}

table_data = {}

for field in impl_fields:

if field['header'] in args_fields:

col_header = field['header'].capitalize()

table_headers.append(col_header)

col_function = field['function']

col_function(G, table_data)

table_align[col_header] = field['align']

table = PrettyTable(table_headers)

add_rows(table_data, table)

# set align

for col_header in table_align:

table.align[col_header] = table_align[col_header]

table.float_format = '5.5'

if args.sort:

args_sort = args.sort.capitalize()

if args_sort in table_headers:

table.sortby = args_sort

table.reversesort = True

else:

print('Please, specify correct field to sort on:\n',

', '.join(table_headers))

exit(1)

table_str = str(table)

if args.top:

table_str = top_of_table(table_str, args.top)

print(table_str)

except FileNotFoundError:

print('No such file or directory! Quitting...')

except IOError:

print('IOError happened! Quitting...')

else:

gprint.print_elapsed_time(time.time() - start_time)