import json
from collections import OrderedDict

from django.http import JsonResponse, HttpResponse
from django.shortcuts import render
from django.template import loader

from .models import Tea, TeaType, TeaCategory, Pic
# Create your views here.
from .utils import get_extension


def list_of_teas():
    dic = OrderedDict()
    for cat in TeaCategory.objects.all():
        dic[cat.name] = OrderedDict()

    for tt in TeaType.objects.all():
        if tt.preferred is not None:
            dic[tt.category.name][tt.name] = OrderedDict()
            dic[tt.category.name][tt.name]['preferred'] = tt.preferred.name
            dic[tt.category.name][tt.name]['price'] = int(tt.preferred.price / 100)
            dic[tt.category.name][tt.name]['pic'] = str(tt.preferred.pic.id) + '.' + get_extension(
                tt.preferred.pic.href)

    return {'categories': dic, 'ms': [0, 50, 100, 150, 200, 250, 300, 350, 400, 450, 500]}


def home(request):
    context = list_of_teas()
    template = loader.get_template('choose.html')
    return HttpResponse(template.render(context, request))


def confirm_choose(request):
    choice = json.loads(request.body.decode(encoding='UTF-8'))
    return JsonResponse({'response': 'ok'})