import json from django.db import models from django.utils import timezone from django.contrib.auth.models import User class Client(models.Model): RESPONSE_TYPE_CHOICES = [ ('code', 'code (Authorization Code Flow)'), ('id_token', 'id_token (Implicit Flow)'), ('id_token token', 'id_token token (Implicit Flow)'), ] name = models.CharField(max_length=100, default='') client_id = models.CharField(max_length=255, unique=True) client_secret = models.CharField(max_length=255, unique=True) response_type = models.CharField(max_length=30, choices=RESPONSE_TYPE_CHOICES) _redirect_uris = models.TextField(default='') def redirect_uris(): def fget(self): return self._redirect_uris.splitlines() def fset(self, value): self._redirect_uris = '\n'.join(value) return locals() redirect_uris = property(**redirect_uris()) @property def default_redirect_uri(self): return self.redirect_uris[0] if self.redirect_uris else '' class Code(models.Model): user = models.ForeignKey(User) client = models.ForeignKey(Client) code = models.CharField(max_length=255, unique=True) expires_at = models.DateTimeField() _scope = models.TextField(default='') def scope(): def fget(self): return self._scope.split() def fset(self, value): self._scope = ' '.join(value) return locals() scope = property(**scope()) def has_expired(self): return timezone.now() >= self.expires_at class Token(models.Model): user = models.ForeignKey(User) client = models.ForeignKey(Client) access_token = models.CharField(max_length=255, unique=True) expires_at = models.DateTimeField() _scope = models.TextField(default='') def scope(): def fget(self): return self._scope.split() def fset(self, value): self._scope = ' '.join(value) return locals() scope = property(**scope()) _id_token = models.TextField() def id_token(): def fget(self): return json.loads(self._id_token) def fset(self, value): self._id_token = json.dumps(value) return locals() id_token = property(**id_token()) class UserInfo(models.Model): user = models.OneToOneField(User, primary_key=True) given_name = models.CharField(max_length=255, default='') family_name = models.CharField(max_length=255, default='') middle_name = models.CharField(max_length=255, default='') nickname = models.CharField(max_length=255, default='') preferred_username = models.CharField(max_length=255, default='') profile = models.URLField(default='') picture = models.URLField(default='') website = models.URLField(default='') email_verified = models.BooleanField(default=False) gender = models.CharField(max_length=100, default='') birthdate = models.DateField(null=True) zoneinfo = models.CharField(max_length=100, default='') locale = models.CharField(max_length=100, default='') phone_number = models.CharField(max_length=255, default='') phone_number_verified = models.BooleanField(default=False) address_formatted = models.CharField(max_length=255, default='') address_street_address = models.CharField(max_length=255, default='') address_locality = models.CharField(max_length=255, default='') address_region = models.CharField(max_length=255, default='') address_postal_code = models.CharField(max_length=255, default='') address_country = models.CharField(max_length=255, default='') updated_at = models.DateTimeField(auto_now=True, null=True) @property def name(self): name = '' if self.given_name: name = self.given_name if self.family_name: name = name + ' ' + self.family_name return name