bars.py

import json
from math import sqrt
from pathlib import Path
from decimal import Decimal, DecimalException


def load_data(filepath):
    with open(filepath, 'r', encoding='cp1251') as file:
        json_data_list = json.loads(file.read())

    return json_data_list


def get_smallest_bar(json_data):
    smallest_bar_ = min(json_data, key=lambda key: key['SeatsCount'])

    return smallest_bar_


def get_biggest_bar(json_data):
    biggest_bar_ = max(json_data, key=lambda key: key['SeatsCount'])

    return biggest_bar_


def get_closest_bar(bars_list_, longitude_, latitude_):
    my_gps_coords = {'latitude': Decimal(latitude_), 'longitude': Decimal(longitude_)}
    for bar_ in bars_list_:
        bar_gps_coords = {'latitude': Decimal(bar_['Latitude_WGS84']), 'longitude': Decimal(bar_['Longitude_WGS84'])}
        bar_['RelativeDistance'] = sqrt((my_gps_coords['latitude'] - bar_gps_coords['latitude']) ** 2
                                        + (my_gps_coords['longitude'] - bar_gps_coords['longitude']) ** 2)

    return min(bars_list_, key=lambda key: key['RelativeDistance'])


def is_decimal(str_input):
    try:
        Decimal(str_input)
    except (DecimalException, TypeError):
        return False

    return True


def is_latitude(str_input):
    if is_decimal(str_input) and -90 <= Decimal(str_input) <= 90:
        return True
    else:
        print('Широта может быть в диапазоне -90 ... 90')
        return False


def is_longitude(str_input):
    if is_decimal(str_input) and -180 <= Decimal(str_input) <= 180:
        return True
    else:
        print('Долгота может быть в диапазоне -180 ... 180')
        return False


def is_good_json(str_input):
    if not str_input:
        return False
    elif Path(str_input).is_file() and str_input.lower().endswith('.json'):
        return True
    else:
        print('Проверьте правильность пути и имени файла')
        return False

if __name__ == '__main__':
    json_path, latitude, longitude = None, None, None

    while not is_good_json(json_path):
        json_path = input('Введите путь к JSON-файлу: ')
    while not is_latitude(latitude):
        latitude = input('Введите вашу широту: ')  # E.g.: 55.60
    while not is_longitude(longitude):
        longitude = input('Введите вашу долготу: ')  # E.g.: 37.74

    bars_list = load_data(json_path)
    biggest_bar = get_biggest_bar(bars_list)
    smallest_bar = get_smallest_bar(bars_list)
    closest_bar = get_closest_bar(bars_list, longitude, latitude)

    print('\nСамый большой бар'.upper())
    print('    {}: {} мест'.format(biggest_bar['Name'], biggest_bar['SeatsCount']))

    print('\nСамый маленький бар'.upper())
    print('    {}: {} мест'.format(smallest_bar['Name'], smallest_bar['SeatsCount']))

    print('\nСамый близкий бар'.upper())
    print('    {}, {}, {}'.format(closest_bar['Name'], closest_bar['District'], closest_bar['Address']))