From 6fd60e2fd1d41332a19f4e37e2874c51b21f13df Mon Sep 17 00:00:00 2001 From: Radek Czajka Date: Sat, 30 Mar 2019 23:33:45 +0100 Subject: [PATCH] Auto-generate keys for services. --- src/services/models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/services/models.py b/src/services/models.py index 593eff2..c7a5d97 100644 --- a/src/services/models.py +++ b/src/services/models.py @@ -1,3 +1,4 @@ +import secrets from django.conf import settings from django.contrib.auth.models import User from django.db import models @@ -20,6 +21,11 @@ class Service(models.Model): def __str__(self): return self.name + def save(self, *args, **kwargs): + if not self.key: + self.key = secrets.token_urlsafe() + return super().save(*args, **kwargs) + def all_users(self): return User.objects.filter( models.Q(service=self) | -- 2.20.1