Feature: Add token creation for auth

This commit is contained in:
winkidney
2022-02-18 20:27:19 -08:00
parent bd7251f8d1
commit 2892ebb333
3 changed files with 21 additions and 0 deletions

View File

@@ -1,6 +1,8 @@
import hashlib
from django.contrib.auth.models import User as BaseUser
from django.db.models.signals import post_save
from django.dispatch import receiver
class User(BaseUser):
@@ -10,3 +12,16 @@ class User(BaseUser):
class Meta:
proxy = True
def create_token_if_necessary(self):
from rest_framework.authtoken.models import Token
token = Token.objects.filter(user=self).first()
if token is not None:
return token
else:
return Token.objects.create(user=self)
@receiver(post_save, sender=User)
def create_profile(sender, instance: User, **kwargs):
instance.create_token_if_necessary()