clubhaus-schornbach/clubhaus/homepage/views.py

import django.utils.timezone
from django.core.cache import cache
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse

import clubhaus.settings as django_settings
from .exceptions import RateLimitHit, ProxyUsageDetected
from .models import Tobacco, ClubhausEvent, EventDate, EventDateVotes, VotingUser


def bad_request(request, exception, template_name="homepage/400.html"):
    return render(request, template_name, context={"exception": exception}, status=400)


def index(request: HttpRequest) -> django.http.HttpResponse:
    return render(request, 'homepage/index.html', {})


def impressum(request: HttpRequest) -> django.http.HttpResponse:
    return render(request, 'homepage/impressum.html')


def tobacco(request: HttpRequest) -> django.http.HttpResponse:
    context = {'tobaccos': Tobacco.objects.all().order_by('-in_stock')}
    return render(request, 'homepage/tobacco.html', context)


def events(request: HttpRequest) -> django.http.HttpResponse:
    next_events = ClubhausEvent.objects.filter(active=True).order_by('-id')
    if len(next_events) == 1:
        next_event = next_events[0]
        dates = EventDate.objects.filter(event=next_event).order_by("date")
        votes = EventDateVotes.objects.filter(date__event=next_event).order_by("voter_id", "date")

        unique_voters = list(set(map(lambda v: v.voter, votes)))
        vote_map = {}
        for voter in unique_voters:
            vote_map[voter] = {date.date.isoformat(): None for date in dates}

        for vote in votes:
            voter = vote.voter
            v_date = vote.date.date.isoformat()
            if voter in vote_map and v_date in vote_map[voter]:
                vote_map[voter][v_date] = vote.available

        return render(request, 'homepage/events.html', {'next_event': next_event, "dates": dates, 'votes': vote_map})
    else:
        return HttpResponseRedirect(reverse("index"))


def voting(request: HttpRequest) -> django.http.HttpResponse:
    request.session.clear_expired()

    # Proxy use is forbidden
    if request.META.get("X-Forwarded-For"):
        raise ProxyUsageDetected()

    if request.method != "POST":
        return HttpResponseRedirect(reverse("events"))

    # Check rate limit
    ip = request.META.get("REMOTE_ADDR")
    cache_key = f"voting_block_{ip}"
    rate_cache: django.core.cache.BaseCache = cache
    if ip not in rate_cache:
        rate_cache.add(cache_key, 0, django_settings.IP_RATE_LIMIT_TIME)
    rate_cache.incr(cache_key)

    if rate_cache.get(cache_key) > django_settings.IP_RATE_LIMIT_COUNT:
        raise RateLimitHit()

    # check params
    if not request.POST["name"]:
        return HttpResponseRedirect(reverse("events"))

    name = request.POST["name"]
    modify_key = request.POST["modifyKey"] or ""

    request.session["name"] = name
    request.session["modifyKey"] = modify_key

    # handle delete
    if request.POST["deleteClicked"] == "1":
        VotingUser.objects.filter(name=name, modify_key=modify_key).delete()
        return HttpResponseRedirect(reverse("events"))

    # handle normal voting
    user, _ = VotingUser.objects.get_or_create(name=name, modify_key=modify_key)
    event_dates = EventDate.objects.filter(event__active=True).order_by("date")

    for date in event_dates:
        available = date.date.isoformat() in request.POST
        EventDateVotes.objects.update_or_create(voter=user, date=date, defaults={"available": available})

    return HttpResponseRedirect(reverse("events"))