skillbox/server/core/models.py

33 lines
1.0 KiB
Python

from datetime import datetime
from django.db import models
from django.contrib.auth import get_user_model
from core.managers import AdminDataManager
class AdminData(models.Model):
hep_admin_token = models.CharField(max_length=100, blank=False, null=False)
updated_at = models.DateTimeField(blank=False, null=True, auto_now=True)
objects = AdminDataManager()
# https://docs.authlib.org/en/latest/client/frameworks.html#frameworks-clients
class OAuth2Token(models.Model):
name = models.CharField(max_length=40)
token_type = models.CharField(max_length=40)
access_token = models.CharField(max_length=200)
refresh_token = models.CharField(max_length=200)
expires_at = models.PositiveIntegerField()
user = models.ForeignKey(get_user_model(), on_delete=models.CASCADE)
def to_token(self):
return dict(
access_token=self.access_token,
token_type=self.token_type,
refresh_token=self.refresh_token,
expires_at=self.expires_at,
)