56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
# 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 main.forms import SimpleCreateUserForm
|
||
from main.models import Order
|
||
from main.models import Transaction
|
||
|
||
|
||
def create_user(request):
|
||
if request.method == "POST":
|
||
form = SimpleCreateUserForm(data=request.POST)
|
||
if form.is_valid():
|
||
user = form.save()
|
||
messages.success(request, "Hurra! {} {} 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
|