|
1 | 1 | import django.db.models
|
| 2 | +import django.shortcuts |
2 | 3 | import django.utils.timezone
|
3 | 4 | import rest_framework.generics
|
4 | 5 | import rest_framework.permissions
|
5 | 6 | import rest_framework.response
|
6 | 7 | import rest_framework.status
|
| 8 | +import rest_framework.views |
7 | 9 | import rest_framework_simplejwt.tokens
|
8 | 10 | import rest_framework_simplejwt.views
|
9 | 11 |
|
10 | 12 | import business.constants
|
11 | 13 | import business.models
|
| 14 | +import user.models |
12 | 15 | import user.pagination
|
13 | 16 | import user.serializers
|
14 | 17 |
|
@@ -224,3 +227,52 @@ def list(self, request, *args, **kwargs):
|
224 | 227 | self.validated_query_params = query_serializer.validated_data
|
225 | 228 |
|
226 | 229 | return super().list(request, *args, **kwargs)
|
| 230 | + |
| 231 | + |
| 232 | +class UserPromoLikeView(rest_framework.views.APIView): |
| 233 | + permission_classes = [rest_framework.permissions.IsAuthenticated] |
| 234 | + |
| 235 | + def get_promo_object(self, promo_id): |
| 236 | + return django.shortcuts.get_object_or_404( |
| 237 | + business.models.Promo, |
| 238 | + id=promo_id, |
| 239 | + ) |
| 240 | + |
| 241 | + def post(self, request, id): |
| 242 | + """Add a like to the promo code.""" |
| 243 | + promo = self.get_promo_object(id) |
| 244 | + |
| 245 | + created = user.models.PromoLike.objects.get_or_create( |
| 246 | + user=request.user, |
| 247 | + promo=promo, |
| 248 | + ) |
| 249 | + |
| 250 | + if created: |
| 251 | + promo.like_count = django.db.models.F('like_count') + 1 |
| 252 | + promo.save(update_fields=['like_count']) |
| 253 | + |
| 254 | + return rest_framework.response.Response( |
| 255 | + {'status': 'ok'}, |
| 256 | + status=rest_framework.status.HTTP_200_OK, |
| 257 | + ) |
| 258 | + |
| 259 | + def delete(self, request, id): |
| 260 | + """Remove a like from the promo code.""" |
| 261 | + promo = self.get_promo_object(id) |
| 262 | + |
| 263 | + # Idempotency: if the like doesn't exist, |
| 264 | + # do nothing and still return 200 OK. |
| 265 | + like_instance = user.models.PromoLike.objects.filter( |
| 266 | + user=request.user, |
| 267 | + promo=promo, |
| 268 | + ).first() |
| 269 | + |
| 270 | + if like_instance: |
| 271 | + like_instance.delete() |
| 272 | + promo.like_count = django.db.models.F('like_count') - 1 |
| 273 | + promo.save(update_fields=['like_count']) |
| 274 | + |
| 275 | + return rest_framework.response.Response( |
| 276 | + {'status': 'ok'}, |
| 277 | + status=rest_framework.status.HTTP_200_OK, |
| 278 | + ) |
0 commit comments