mirror of
https://github.com/pinry/pinry.git
synced 2026-05-09 00:26:59 +02:00
Feature: Add token creation for auth
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user