43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from django.conf import settings
|
|
from django.db import models
|
|
from jose import jwt
|
|
from jsonfield import JSONField
|
|
|
|
|
|
class UserKey(models.Model):
|
|
username = models.CharField(max_length=50)
|
|
properties = JSONField(null=True)
|
|
added_on = models.DateTimeField(auto_now_add=True)
|
|
key_type = models.CharField(max_length=25, default="TOTP")
|
|
enabled = models.BooleanField(default=True)
|
|
expires = models.DateTimeField(null=True, default=None, blank=True)
|
|
last_used = models.DateTimeField(null=True, default=None, blank=True)
|
|
owned_by_enterprise = models.BooleanField(default=None, null=True, blank=True)
|
|
|
|
def save(self, *args, **kwargs):
|
|
if (
|
|
self.key_type == "Trusted Device"
|
|
and self.properties.get("signature", "") == ""
|
|
):
|
|
self.properties["signature"] = jwt.encode(
|
|
{"username": self.username, "key": self.properties["key"]},
|
|
settings.SECRET_KEY,
|
|
)
|
|
super().save(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return "%s -- %s" % (self.username, self.key_type)
|
|
|
|
class Meta:
|
|
app_label = "mfa"
|
|
|
|
|
|
class OTPTracker(models.Model):
|
|
username = models.CharField(max_length=50)
|
|
value = models.CharField(max_length = 6)
|
|
success = models.BooleanField(blank=True)
|
|
done_on = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
app_label = 'mfa'
|
|
indexes = [models.Index(fields=['username'])] |