phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed subscriptions unsubscribe

Browse Source
This commit is contained in:
dormantman 2020-01-22 17:15:49 +03:00
parent 07f4947cc0
commit c78c6a67b1
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/notification/views/common.py
View File

@ -46,14 +46,13 @@ class SubscribeInfoAuthUserView(generics.RetrieveAPIView):
def get_object(self): def get_object(self):
user = self.request.user user = self.request.user
if user.is_authenticated: subscriber = models.Subscriber.objects.filter(user=user).first()
subscriber = models.Subscriber.objects.filter(user=user).first()
if subscriber is None: if subscriber is None:
subscriber = models.Subscriber.objects.make_subscriber( subscriber = models.Subscriber.objects.make_subscriber(
email=user.email, user=user, ip_address=get_user_ip(self.request), email=user.email, user=user, ip_address=get_user_ip(self.request),
country_code=self.request.country_code, locale=self.request.locale country_code=self.request.country_code, locale=self.request.locale
) )
else: else:
return get_object_or_404(models.Subscriber, user=user) return get_object_or_404(models.Subscriber, user=user)
@ -86,7 +85,7 @@ class UnsubscribeAuthUserView(generics.GenericAPIView):
def patch(self, request, *args, **kw): def patch(self, request, *args, **kw):
user = request.user user = request.user
obj = get_object_or_404(models.Subscriber, user=user) obj = models.Subscriber.objects.filter(user=user).first()
obj.unsubscribe(request.query_params) obj.unsubscribe(request.query_params)
serializer = self.get_serializer(instance=obj) serializer = self.get_serializer(instance=obj)
return Response(data=serializer.data) return Response(data=serializer.data)