52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from django.contrib import messages
|
||
from django.contrib.auth.decorators import login_required
|
||
from django.db import connection
|
||
from django.db.models import Count, Sum
|
||
from django.shortcuts import redirect, render
|
||
|
||
from main.forms import SimpleCreateUserForm
|
||
from main.models import Order, 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
|