clubhaus-schornbach/clubhaus/homepage/views.py

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

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


def index(request: HttpRequest) -> django.http.HttpResponse:
    return render(request, 'homepage/index.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")

        return render(request, 'homepage/events.html', {'next_event': next_event, "dates": dates, 'votes': votes})
    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"):
        return HttpResponseForbidden()

    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 request.method != "POST" and rate_cache.get(cache_key) > django_settings.IP_RATE_LIMIT_COUNT:
        return HttpResponseRedirect(reverse("events"))

    modify_key = request.POST["modifyKey"] or ""
    if name := request.POST["name"]:
        request.session["name"] = name
        request.session["modifyKey"] = modify_key
        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:
            date_in_request = date.date.isoformat() in request.POST
            EventDateVotes.objects.update_or_create(
                voter=user, date=date,
                defaults={"voter": user, "date": date, "available": date_in_request})

    return HttpResponseRedirect(reverse("events"))