#!/usr/bin/env python3

import json
import urllib.request
import argparse
import collections
from operator import itemgetter

parser = argparse.ArgumentParser(description="API Call to collect data")
parser.add_argument("tenant", type=str, help="Tenant Name")
parser.add_argument("token", type=str, help="Tenat API Token")
parser.add_argument("-t", "--timeperiod", type=int, default='604800', help="Timeperiod (default: 604800)")

try:
	args = parser.parse_args()
	tenant = args.tenant
	token = args.token
	timeperiod = args.timeperiod

except argparse.ArgumentError as e:
	print(str(e))

def print_dict(dict):
	for key, value in sorted(dict.items(), key = itemgetter(1), reverse = True):
		print ("{:<35s}{:5d}".format(key, value))

base_url = "https://{}.goskope.com/api/v1/events?token={}&type=page&timeperiod={}".format(tenant, token, timeperiod)

req = urllib.request.Request(base_url)
with urllib.request.urlopen(req) as response:
	content = response.read()
json_content = json.loads(content)

domains = collections.Counter()
categories = collections.Counter()

for i in range (0, len (json_content['data'])):
	domain = json_content["data"][i]["domain"]
	ccl = json_content["data"][i]["ccl"]
	category = json_content["data"][i]["category"]
	domains[domain] += 1
	categories[category][count] += 1


#print ("===== Domains =====")
#print_dict(domains)


print ("\n===== Categories =====")
print_dict(categories)