77 lines
1.4 KiB
Python
77 lines
1.4 KiB
Python
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:
|
|
|
|
return str(hashlib.sha256(str(self.randomword()).encode('utf-8')).hexdigest())
|
|
|
|
|
|
def token_hash(self, token:str) -> str:
|
|
|
|
salt = settings.SECRET_KEY
|
|
|
|
return str(hashlib.sha256(str(token + 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
|