moviewyrm/bookwyrm/views/goal.py

83 lines
2.8 KiB
Python
Raw Normal View History

2021-03-08 16:49:10 +00:00
""" non-interactive pages """
2021-01-16 19:34:19 +00:00
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
2021-01-16 19:34:19 +00:00
from django.template.response import TemplateResponse
2021-08-04 19:05:24 +00:00
from django.utils import timezone
2021-01-16 19:34:19 +00:00
from django.utils.decorators import method_decorator
from django.views import View
from django.views.decorators.http import require_POST
2021-01-16 19:34:19 +00:00
from bookwyrm import forms, models
2021-01-16 20:48:04 +00:00
from bookwyrm.status import create_generated_note
from .helpers import get_user_from_username
2021-01-16 19:34:19 +00:00
# pylint: disable= no-self-use
2021-03-08 16:49:10 +00:00
@method_decorator(login_required, name="dispatch")
2021-01-16 19:34:19 +00:00
class Goal(View):
2021-04-26 16:15:42 +00:00
"""track books for the year"""
2021-03-08 16:49:10 +00:00
2021-01-16 19:34:19 +00:00
def get(self, request, username, year):
2021-04-26 16:15:42 +00:00
"""reading goal page"""
user = get_user_from_username(request.user, username)
2021-01-16 19:34:19 +00:00
year = int(year)
2021-03-08 16:49:10 +00:00
goal = models.AnnualGoal.objects.filter(year=year, user=user).first()
2021-01-16 19:34:19 +00:00
if not goal and user != request.user:
2021-01-16 21:29:28 +00:00
return HttpResponseNotFound()
2021-01-16 19:34:19 +00:00
2021-08-04 19:05:24 +00:00
current_year = timezone.now().year
if not goal and year != timezone.now().year:
2021-08-04 19:40:15 +00:00
return redirect("user-goal", username, current_year)
2021-08-04 19:05:24 +00:00
2021-09-28 00:52:20 +00:00
if goal:
goal.raise_visible_to_user(request.user)
2021-01-16 19:34:19 +00:00
data = {
2021-03-08 16:49:10 +00:00
"goal_form": forms.GoalForm(instance=goal),
"goal": goal,
"user": user,
"year": year,
"is_self": request.user == user,
2021-01-16 19:34:19 +00:00
}
2021-09-28 23:18:32 +00:00
return TemplateResponse(request, "user/goal.html", data)
2021-01-16 19:34:19 +00:00
def post(self, request, username, year):
2021-04-26 16:15:42 +00:00
"""update or create an annual goal"""
year = int(year)
user = get_user_from_username(request.user, username)
goal = models.AnnualGoal.objects.filter(year=year, user=user).first()
if goal:
goal.raise_not_editable(request.user)
2021-01-16 19:34:19 +00:00
form = forms.GoalForm(request.POST, instance=goal)
if not form.is_valid():
data = {
2021-03-08 16:49:10 +00:00
"goal_form": form,
"goal": goal,
"year": year,
2021-01-16 19:34:19 +00:00
}
2021-09-28 23:18:32 +00:00
return TemplateResponse(request, "user/goal.html", data)
2021-01-16 21:29:28 +00:00
goal = form.save()
2021-01-16 19:34:19 +00:00
2021-03-08 16:49:10 +00:00
if request.POST.get("post-status"):
# create status, if appropriate
2021-03-08 16:49:10 +00:00
template = get_template("snippets/generated_status/goal.html")
2021-02-04 21:21:55 +00:00
create_generated_note(
2021-01-16 20:48:04 +00:00
request.user,
template.render({"goal": goal, "user": user}).strip(),
2021-03-08 16:49:10 +00:00
privacy=goal.privacy,
2021-01-16 20:48:04 +00:00
)
2021-03-08 16:49:10 +00:00
return redirect(request.headers.get("Referer", "/"))
@require_POST
@login_required
def hide_goal(request):
2021-04-26 16:15:42 +00:00
"""don't keep bugging people to set a goal"""
request.user.show_goal = False
request.user.save(broadcast=False, update_fields=["show_goal"])
return redirect(request.headers.get("Referer", "/"))