from collections import namedtuple

import json

import requests

from algo import kmeans

from clean_data import cleaning

institution_map = {'home': 'https://opendata.arcgis.com/datasets/ac6fc684043341f6b1d6298c146a0bcf_1.geojson',

'school': 'https://opendata.arcgis.com/datasets/cccae6f029334927856da6e20a50561f_19.geojson',

'hospital': 'https://opendata.arcgis.com/datasets/a5867b5375544ceb8f06544a5ed349a5_15.geojson',

'libraries': 'https://opendata.arcgis.com/datasets/67a54ea25d944cf7b66750ba57da822c_1.geojson',

'museums': 'https://opendata.arcgis.com/datasets/6728810fb847489985d4b735502205a0_2.geojson',

'golf_courses': 'https://opendata.arcgis.com/datasets/7b46f5eabbe0496fb168860ddc22fa35_14.geojson',

'parks': 'https://opendata.arcgis.com/datasets/4f1b554e743b423f9574e7a3ca814cce_6.geojson',

'community_centres': 'https://opendata.arcgis.com/datasets/272667665de646768db14e9fa1676405_11.geojson',

'tourism_poi': 'https://opendata.arcgis.com/datasets/e09cca59a8584cc28e7879bd6b0ce7f4_0.geojson'}

data_point = namedtuple('data_point', ['institution', 'longitude', 'latitude'])

def get_data(institution_type):

print(f'Getting data points for institution {institution_type}')

try:

data = json.load(open(f'cached_raw_data/{institution_type}'))

print('Successfully retrieved cached data')

except:

print('Failed to retrieve from cache')

data = requests.request(method='get', url=institution_map[institution_type]).json()

json.dump(data, open(f'cached_raw_data/{institution_type}', 'w'))

print(f'Cached data')

data_points = []

for x in data['features']:

x = x['properties']

data_points.append(data_point(institution_type,

float(x['LONGITUDE']),

float(x['LATITUDE'])))

return data_points

def process_query(k_value: int, poi: dict):

data_points = []

cleaning(poi)

for institution_type in poi:

if institution_type in institution_map:

data_points.extend(get_data(institution_type))

return kmeans(k_value, poi, data_points)