push notificantion test

backend/account/admin.py

@@ -87,4 +87,12 @@ class AddressAdmin(ModelAdmin, ImportExportModelAdmin):
     }
     def address_display(self, obj):
         return obj.address[0:35] + '...'
-    address_display.short_description = 'ادرس'
+    address_display.short_description = 'ادرس'
+
+
+@admin.register(PushSubscription)
+class PushSubscription(ModelAdmin, ImportExportModelAdmin):
+    import_form_class = ImportForm
+    export_form_class = ExportForm
+    compressed_fields = True
+    warn_unsaved_form = True

backend/account/migrations/0013_pushsubscription.py

@@ -0,0 +1,25 @@
+# Generated by Django 5.1.2 on 2025-02-16 20:03
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('account', '0012_alter_user_video_uploader'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='PushSubscription',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('endpoint', models.TextField()),
+                ('keys', models.JSONField()),
+                ('created_at', models.DateField(auto_now_add=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

backend/account/models.py

@@ -7,7 +7,7 @@ from django.utils import timezone
 from rest_framework_simplejwt.token_blacklist.models import BlacklistedToken, OutstandingToken
 import hashlib
 from django.contrib import admin
-
+from django.conf import settings
 class UserManager(BaseUserManager):
     def create_user(self, phone, password=None):
         if not phone:
@@ -135,4 +135,65 @@ class UserAddressModel(models.Model):
     
     class Meta:
         verbose_name = 'ادرس کاربر'
-        verbose_name_plural = 'ادرس های کاربر'
+        verbose_name_plural = 'ادرس های کاربر'
+
+
+import os
+import json
+from pywebpush import webpush, WebPushException
+
+class PushSubscription(models.Model):
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    endpoint = models.TextField()
+    keys = models.JSONField()
+    created_at = models.DateField(auto_now_add=True)
+
+    def __str__(self):
+        return f'{self.user} push'
+    
+    def send_notif(self, title, body):
+        payload = {
+            "title": title,
+            "body": body,
+            "icon": ''  
+        }
+
+        try:
+            webpush(
+                subscription_info={
+                    "endpoint": self.endpoint,
+                    "keys": self.keys
+                },
+                data=json.dumps(payload),
+                vapid_private_key=settings.VAPID_PRIVATE_KEY,
+                vapid_claims={
+                    "sub": "mailto:admin@example.com"
+                }
+            )
+        except WebPushException as ex:
+            print("Failed to send notification:", ex)
+    
+    @classmethod
+    def send_group_notification(cls, user, title, body):
+        payload = {
+            "title": title,
+            "body": body,
+            "icon": '' 
+        }
+
+        subscriptions = PushSubscription.objects.filter(user=user)
+        for sub in subscriptions:
+            try:
+                webpush(
+                    subscription_info={
+                        "endpoint": sub.endpoint,
+                        "keys": sub.keys
+                    },
+                    data=json.dumps(payload),
+                    vapid_private_key=settings.VAPID_PRIVATE_KEY,
+                    vapid_claims={
+                        "sub": "mailto:admin@example.com"
+                    }
+                )
+            except WebPushException as ex:
+                print(f"Failed to send notification to {sub.user}:", ex)

backend/account/serializers.py

@@ -23,4 +23,10 @@ class UserAddressSerializer(serializers.ModelSerializer):
         user = self.context['request'].user
         if not user.is_authenticated:
             raise serializers.ValidationError("You must be logged in to perform this action.")
-        return data
+        return data
+
+
+class PushSubscriptionSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = PushSubscription
+        fields = ('endpoint', 'keys')

backend/account/urls.py

@@ -12,4 +12,5 @@ urlpatterns = [
     path('address/delete/<int:pk>', views.DeleteAddressView.as_view(), name='delete-address'),
     path('address/list', views.GetUserAddressesView.as_view(), name='list-addresses'),
     path('address/<int:pk>', views.GetIDUserAddressView.as_view(), name='get-ID-address'),
+    path('subscribe', views.SubscribeView.as_view(), name='subscibe')
 ]

backend/account/views.py

@@ -2,7 +2,7 @@ from django.shortcuts import render
 from rest_framework.views import APIView
 from rest_framework import generics, permissions, status
 from rest_framework.response import Response
-from .serializers import ProfileSerializer, UserAddressSerializer, CustomTokenObtainPairSerializer
+from .serializers import *
 from .models import UserAddressModel, User
 from rest_framework.permissions import IsAuthenticated, AllowAny
 from drf_spectacular.utils import extend_schema, OpenApiParameter
@@ -171,4 +171,18 @@ class GetIDUserAddressView(generics.RetrieveAPIView):
     permission_classes = [permissions.IsAuthenticated]
 
     def get_queryset(self):
-        return UserAddressModel.objects.filter(user=self.request.user)
+        return UserAddressModel.objects.filter(user=self.request.user)
+
+
+
+class SubscribeView(APIView):
+    serializer_class = PushSubscriptionSerializer
+    def post(self, request):
+        push_ser = self.serializer_class(data=request.data)
+        if push_ser.is_valid():
+            PushSubscription.objects.update_or_create(
+                user=request.user,
+                defaults=(push_ser.validated_data)
+            )
+            return Response(status=status.HTTP_201_CREATED)
+        return Response(status=status.HTTP_400_BAD_REQUEST)