From 80392ae7ba8c4da49dae8ff5ce47d87ae3b77359 Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Thu, 15 Mar 2018 09:37:03 +0300
Subject: [PATCH] Update NotificationEditView

---
 apps/user/views.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/apps/user/views.py b/apps/user/views.py
index 9f647065..3cc75cc3 100644
--- a/apps/user/views.py
+++ b/apps/user/views.py
@@ -25,6 +25,7 @@ from apps.course.models import Course
 from apps.notification.utils import send_email
 from apps.school.models import SchoolSchedule
 from apps.payment.models import AuthorBalance, CoursePayment, SchoolPayment
+from apps.user.models import SubscriptionCategory
 
 from .forms import AuthorRequesForm, UserEditForm, WithdrawalForm
 from .models import AuthorRequest
@@ -93,6 +94,18 @@ class NotificationEditView(TemplateView):
     def get(self, request, pk=None):
         return super().get(request)
 
+    def post(self, request, pk=None):
+        categories = request.POST.getlist('category', [])
+        request.user.email_subscription.categories.set(
+            SubscriptionCategory.objects.filter(id__in=categories)
+        )
+        return redirect('user-edit-notifications', request.user.id)
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['subscription_categories'] = SubscriptionCategory.objects.all()
+        return context
+
 
 @method_decorator(login_required, name='dispatch')
 class PaymentHistoryView(FormView):