from rest_framework.response import Response
from rest_framework import permissions
from django.core.cache import cache
import json
from service.utils import raschet as raschet_alg
import pandas as pd
from io import BytesIO
from django.http import HttpResponse
from rest_framework.generics import GenericAPIView

def rename_result_dataset(dataframe, rename_dict, reverse=False):
    if reverse:
        return dataframe.rename(columns={v: k for k, v in rename_dict.items()})
    return dataframe.rename(columns=rename_dict)

class ao_and_rayons(GenericAPIView):
    permission_classes = [permissions.AllowAny]

    def get(self, request, format=None):
        d = cache.get('ao_and_rayons')
        if d is None:
            data = json.loads(open('ao_and_rayons.json', 'r').read())
            cache.set('ao_and_rayons', data, 60 * 60 * 24)
            d = data
        return Response(d)


class raschet(GenericAPIView):
    permission_classes = [permissions.AllowAny]

    def post(self, request, format=None):
        POINTS_RENAME_DICT = {
            "point_id": "ID точки (для всех наборов одинаковые)",
            "category": "тип объекта",
            "msk_ao": "Адм. округ",
            "msk_rayon": "Адм. район",
            "people": "Численность населения в 2021 г.",
            "people2025": "Численность населения в 2025 г. (прогноз)",
            "stops_ot": "Остановки общественного транспорта",
            "routes_ot": "Маршруты общетвенного транспорта",
            "in_metro": "Входы в ближайшее метро в месяц",
            "out_metro": "Выходы из ближайшего метро в месяц",
            "tc": "Тогровые центры",
            "empls": "Рабочие места",
            "walkers": "Трафик населения",
            "schools": "Школы и детские сады",
            "parking": "Парковочные места",
            "pvz": "Пункты выдачи заказов",
            "gov_place": "Рекомендованные пункты размещения постаматов",
            "bike_park": "Городская аренда велосипедов, шт.",
            "products": "Продовольственные магазины",
            "nonprod": "Непродовольственные магазины",
            "service": "Пункты оказания бытовых услуг",
            "vuz": "ВУЗы и техникумы"
        }
        NET_RENAME_DICT = {
            "cell_id": "ID ячейки (для всех трёх сеток - разные, нумерация сквозная)",
            "msk_ao": "Адм. округ",
            "msk_rayon": "Адм. район",
            "people": "Численность населения в 2021 г.",
            "people2025": "Численность населения в 2025 г. (прогноз)",
            "stops_ot": "Остановки общественного транспорта",
            "routes_ot": "Маршруты общетвенного транспорта",
            "in_metro": "Входы в ближайшее метро в месяц",
            "out_metro": "Выходы из ближайшего метро в месяц",
            "tc": "Тогровые центры",
            "empls": "Рабочие места",
            "walkers": "Трафик населения",
            "schools": "Школы и детские сады",
            "parking": "Парковочные места",
            "pvz": "Пункты выдачи заказов",
            "gov_place": "Рекомендованные пункты размещения постаматов",
            "bike_park": "Городская аренда велосипедов, шт.",
            "products": "Продовольственные магазины",
            "nonprod": "Непродовольственные магазины",
            "service": "Пункты оказания бытовых услуг",
            "vuz": "ВУЗы и техникумы"
        }
        df_points_, df_nets_ = raschet_alg(**request.data)
        df_points = rename_result_dataset(df_points_, POINTS_RENAME_DICT)
        df_nets = rename_result_dataset(df_nets_, NET_RENAME_DICT)
        with BytesIO() as b:
            # Use the StringIO object as the filehandle.
            writer = pd.ExcelWriter(b, engine='xlsxwriter')
            if df_points is not None:
                df_points.to_excel(writer, sheet_name='Точки', index=False)
            if df_nets is not None:
                df_nets.to_excel(writer, sheet_name='Полигоны', index=False)
            writer.save()
            # Set up the Http response.
            filename = f'Выгрузка.xlsx'
            response = HttpResponse(
                b.getvalue(),
                content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
            )
            response['Content-Disposition'] = 'attachment; filename=%s' % filename
            return response