71
app/api/models/tokens.py
Normal file
71
app/api/models/tokens.py
Normal file
@ -0,0 +1,71 @@
|
||||
import hashlib
|
||||
import random
|
||||
import string
|
||||
|
||||
from django.conf import settings
|
||||
from django.contrib.auth.models import User
|
||||
from django.db import models
|
||||
|
||||
from access.fields import *
|
||||
from access.models import TenancyObject
|
||||
|
||||
|
||||
|
||||
class AuthToken(models.Model):
|
||||
|
||||
|
||||
id = models.AutoField(
|
||||
primary_key=True,
|
||||
unique=True,
|
||||
blank=False
|
||||
)
|
||||
|
||||
note = models.CharField(
|
||||
blank = True,
|
||||
max_length = 50,
|
||||
default = None,
|
||||
null= True,
|
||||
)
|
||||
|
||||
token = models.CharField(
|
||||
verbose_name = 'Auth Token',
|
||||
db_index=True,
|
||||
max_length = 64,
|
||||
null = False,
|
||||
blank = False,
|
||||
unique = True,
|
||||
)
|
||||
|
||||
|
||||
user = models.ForeignKey(
|
||||
settings.AUTH_USER_MODEL,
|
||||
on_delete=models.CASCADE
|
||||
)
|
||||
|
||||
expires = models.DateTimeField(
|
||||
verbose_name = 'Expiry Date',
|
||||
null = False,
|
||||
blank = False
|
||||
)
|
||||
|
||||
|
||||
created = AutoCreatedField()
|
||||
|
||||
modified = AutoLastModifiedField()
|
||||
|
||||
|
||||
def generate(self) -> str:
|
||||
|
||||
salt = settings.SECRET_KEY
|
||||
|
||||
return str(hashlib.sha256(str(self.randomword() + salt).encode('utf-8')).hexdigest())
|
||||
|
||||
|
||||
def randomword(self) -> str:
|
||||
|
||||
return ''.join(random.choice(string.ascii_letters) for i in range(120))
|
||||
|
||||
|
||||
def __str__(self):
|
||||
|
||||
return self.token
|
Reference in New Issue
Block a user