# coding: utf-8
from __future__ import unicode_literals

from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.db import connection
from django.db.models import Sum, Count
from django.shortcuts import render, redirect
from django.contrib.auth import logout as auth_logout
from main.forms import SimpleCreateUserForm
from main.models import Order
from main.models import Transaction


def logout(request):
    auth_logout(request)
    return render(request, 'registration/logout.html')


def create_user(request):
    if request.method == "POST":
        form = SimpleCreateUserForm(data=request.POST)
        if form.is_valid():
            user = form.save()
            messages.success(request, "Hurra! {0} {1} er lagt til".format(user.first_name, user.last_name))
            return redirect('create-user')

        else:
            messages.error(request, 'Feil med skjemaet, se under.')
            form = SimpleCreateUserForm(data=request.POST)
    else:
        form = SimpleCreateUserForm()

    return render(request, 'registration/create_user.html', {'form': form})


@login_required
def profile(request):
    transactions = Transaction.objects.filter(user=request.user.pk)
    transaction_totals = {
        'sum': sum([t.amount for t in transactions])
    }
    orders = Order.objects.filter(customer=request.user.pk)
    orders_totals = {
        'sum': sum([o.order_sum for o in orders])
    }

    # order_lines = OrderLine.objects.filter(order__customer__pk=1).order_by("order__created")
    order_report = _orders_monthly(filter_by={'customer': request.user.pk})
    top_months = order_report.order_by('-sum')[:5]

    return render(request, 'profile.html', locals())


def _orders_monthly(filter_by={}, order_by='month'):

    truncate_date = connection.ops.date_trunc_sql('month', 'created')
    qs = Order.objects.filter(**filter_by).extra({'month': truncate_date})
    order_report = qs.values('month').annotate(sum=Sum('order_sum'), num=Count('pk')).order_by(order_by)

    return order_report