Create goal with privacy

bookwyrm/views/__init__.py

@@ -7,6 +7,7 @@ from .direct_message import DirectMessage
 from .error import not_found_page, server_error_page
 from .follow import follow, unfollow
 from .follow import accept_follow_request, delete_follow_request, handle_accept
+from .goal import Goal
 from .import_data import Import, ImportStatus
 from .interaction import Favorite, Unfavorite, Boost, Unboost
 from .invite import ManageInvites, Invite

bookwyrm/views/goal.py

@@ -0,0 +1,63 @@
+''' non-interactive pages '''
+from django.contrib.auth.decorators import login_required
+from django.http import HttpResponseNotFound
+from django.shortcuts import redirect
+from django.template.response import TemplateResponse
+from django.utils.decorators import method_decorator
+from django.views import View
+
+from bookwyrm import forms, models
+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(username)
+        year = int(year)
+        goal = models.AnnualGoal.objects.filter(
+            year=year, user=user
+        ).first()
+        if not goal and user != request.user:
+            return redirect('/')
+
+        if goal and user != request.user:
+            if goal.privacy == 'direct' or \
+                    (goal.privacy == 'followers' and not follower):
+                return HttpResponseNotFound()
+
+        data = {
+            'title': '%s\'s %d Reading' % (user.display_name, year),
+            'goal_form': forms.GoalForm(instance=goal),
+            'goal': goal,
+            'user': user,
+            'year': year,
+        }
+        return TemplateResponse(request, 'goal.html', data)
+
+
+    def post(self, request, username, year):
+        ''' update or create an annual goal '''
+        user = get_user_from_username(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 = {
+                'title': '%s\'s %d Reading' % (goal.user.display_name, year),
+                'goal_form': form,
+                'goal': goal,
+                'year': year,
+            }
+            return TemplateResponse(request, 'goal.html', data)
+        form.save()
+
+        return redirect(request.headers.get('Referer', '/'))