36 lines
1.4 KiB
Python
36 lines
1.4 KiB
Python
from django.http.response import HttpResponse
|
|
from django.views.decorators.http import require_GET
|
|
from django.http.response import JsonResponse, HttpResponse
|
|
from django.views.decorators.http import require_GET, require_POST
|
|
from django.shortcuts import render, get_object_or_404
|
|
from django.contrib.auth.models import User
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
from webpush import send_user_notification
|
|
from django.conf import settings
|
|
import json
|
|
|
|
@require_GET
|
|
def home(request):
|
|
webpush_settings = getattr(settings, 'WEBPUSH_SETTINGS', {})
|
|
vapid_key = webpush_settings.get('VAPID_PUBLIC_KEY')
|
|
user = request.user
|
|
return render(request, 'home.html', {user: user, 'vapid_key': vapid_key})
|
|
|
|
@require_POST
|
|
@csrf_exempt
|
|
def send_push(request):
|
|
try:
|
|
body = request.body
|
|
data = json.loads(body)
|
|
|
|
if 'head' not in data or 'body' not in data or 'id' not in data:
|
|
return JsonResponse(status=400, data={"message": "Invalid data format"})
|
|
|
|
user_id = data['id']
|
|
user = get_object_or_404(User, pk=user_id)
|
|
payload = {'head': data['head'], 'body': data['body']}
|
|
send_user_notification(user=user, payload=payload, ttl=1000)
|
|
|
|
return JsonResponse(status=200, data={"message": "Web push successful"})
|
|
except TypeError:
|
|
return JsonResponse(status=500, data={"message": "An error occurred"}) |