timetracker/src/web/tracker/views.py

from django.shortcuts import render

from .models import Game, Platform, Purchase, Session
from .forms import SessionForm, PurchaseForm, GameForm
from datetime import datetime
from django.db.models import ExpressionWrapper, F, DurationField
import logging


def add_session(request):
    context = {}
    now = datetime.now()
    initial = {"timestamp_start": now, "timestamp_end": now}
    form = SessionForm(request.POST or None, initial=initial)
    if form.is_valid():
        form.save()

    context["form"] = form
    return render(request, "add_session.html", context)


def list_sessions(request, purchase_id=None):
    context = {}

    if purchase_id != None:
        dataset = Session.objects.filter(purchase=purchase_id)
        context["purchase"] = Purchase.objects.get(id=purchase_id)
    else:
        dataset = Session.objects.all()

    dataset = dataset.annotate(
        time_delta=ExpressionWrapper(
            F("timestamp_end") - F("timestamp_start"), output_field=DurationField()
        )
    )
    context["dataset"] = dataset

    return render(request, "list_sessions.html", context)


def add_purchase(request):
    context = {}
    now = datetime.now()
    initial = {"date_purchased": now}
    form = PurchaseForm(request.POST or None, initial=initial)
    if form.is_valid():
        form.save()

    context["form"] = form
    context["title"] = "Add New Purchase"
    return render(request, "add.html", context)


def add_game(request):
    context = {}
    form = GameForm(request.POST or None)
    if form.is_valid():
        form.save()

    context["form"] = form
    context["title"] = "Add New Game"
    return render(request, "add.html", context)