moviewyrm/bookwyrm/views/goal.py
2021-04-11 09:31:18 -07:00

77 lines
2.6 KiB
Python

""" non-interactive pages """
from django.contrib.auth.decorators import login_required
from django.http import HttpResponseNotFound
from django.shortcuts import redirect
from django.template.loader import get_template
from django.template.response import TemplateResponse
from django.utils.decorators import method_decorator
from django.views import View
from django.views.decorators.http import require_POST
from bookwyrm import forms, models
from bookwyrm.status import create_generated_note
from .helpers import get_user_from_username
# pylint: disable= no-self-use
@method_decorator(login_required, name="dispatch")
class Goal(View):
""" track books for the year """
def get(self, request, username, year):
""" reading goal page """
user = get_user_from_username(request.user, username)
year = int(year)
goal = models.AnnualGoal.objects.filter(year=year, user=user).first()
if not goal and user != request.user:
return HttpResponseNotFound()
if goal and not goal.visible_to_user(request.user):
return HttpResponseNotFound()
data = {
"goal_form": forms.GoalForm(instance=goal),
"goal": goal,
"user": user,
"year": year,
"is_self": request.user == user,
}
return TemplateResponse(request, "goal.html", data)
def post(self, request, username, year):
""" update or create an annual goal """
user = get_user_from_username(request.user, username)
if user != request.user:
return HttpResponseNotFound()
year = int(year)
goal = models.AnnualGoal.objects.filter(year=year, user=request.user).first()
form = forms.GoalForm(request.POST, instance=goal)
if not form.is_valid():
data = {
"goal_form": form,
"goal": goal,
"year": year,
}
return TemplateResponse(request, "goal.html", data)
goal = form.save()
if request.POST.get("post-status"):
# create status, if appropraite
template = get_template("snippets/generated_status/goal.html")
create_generated_note(
request.user,
template.render({"goal": goal, "user": request.user}).strip(),
privacy=goal.privacy,
)
return redirect(request.headers.get("Referer", "/"))
@require_POST
@login_required
def hide_goal(request):
""" don't keep bugging people to set a goal """
request.user.show_goal = False
request.user.save(broadcast=False)
return redirect(request.headers.get("Referer", "/"))