22 lines
571 B
Python
22 lines
571 B
Python
from base64 import urlsafe_b64encode
|
|
|
|
import ecdsa
|
|
from django.core.management.base import BaseCommand
|
|
from djangoldp_webpushnotification.models import VAPIDKeyset
|
|
from ecdsa import SigningKey
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Generate VAPID key pair"
|
|
|
|
def handle(self, *args, **options):
|
|
priv_key = SigningKey.generate(curve=ecdsa.NIST256p)
|
|
|
|
VAPIDKeyset.objects.create(
|
|
private_key=urlsafe_b64encode(priv_key.to_string()).strip(b"=")
|
|
)
|
|
|
|
self.stdout.write("VAPID Keyset succesfully generated")
|
|
|
|
exit(0)
|