当创建一个客户端的时候,自动创建一个API Token
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from django.db import models
|
||||
from django.utils import timezone
|
||||
import secrets
|
||||
|
||||
# Status choices for tasks
|
||||
STATUS_CHOICES = [
|
||||
@@ -20,6 +21,13 @@ class Client(models.Model):
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
# Generate a unique token if this is a new client
|
||||
if not self.pk and not self.token:
|
||||
# Use secrets module to generate a secure random token
|
||||
self.token = secrets.token_urlsafe(64) # 64 bytes -> ~86 characters
|
||||
super().save(*args, **kwargs)
|
||||
|
||||
class Meta:
|
||||
verbose_name = '客户端'
|
||||
|
||||
Reference in New Issue
Block a user