django-oidc-provider/oidc_provider/models.py

135 lines
4.5 KiB
Python
Raw Normal View History

import json
2014-12-19 15:27:43 +00:00
from django.db import models
from django.utils import timezone
from django.contrib.auth.models import User
2014-12-19 15:27:43 +00:00
2015-03-20 17:38:28 +00:00
def scope_property():
def fget(self):
return self._scope.split()
def fset(self, value):
self._scope = ' '.join(value)
return locals()
2014-12-19 15:27:43 +00:00
class Client(models.Model):
RESPONSE_TYPE_CHOICES = [
2015-01-08 20:55:24 +00:00
('code', 'code (Authorization Code Flow)'),
('id_token', 'id_token (Implicit Flow)'),
('id_token token', 'id_token token (Implicit Flow)'),
2014-12-19 15:27:43 +00:00
]
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)
2015-02-06 17:26:34 +00:00
response_type = models.CharField(max_length=30,
choices=RESPONSE_TYPE_CHOICES)
2014-12-19 15:27:43 +00:00
_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())
2014-12-19 15:27:43 +00:00
@property
def default_redirect_uri(self):
return self.redirect_uris[0] if self.redirect_uris else ''
2014-12-19 15:27:43 +00:00
2014-12-19 15:27:43 +00:00
class Code(models.Model):
user = models.ForeignKey(User)
client = models.ForeignKey(Client)
code = models.CharField(max_length=255, unique=True)
expires_at = models.DateTimeField()
2015-02-06 17:26:34 +00:00
_scope = models.TextField(default='')
2015-03-20 17:38:28 +00:00
scope = property(**scope_property())
2014-12-19 15:27:43 +00:00
def has_expired(self):
return timezone.now() >= self.expires_at
2014-12-19 15:27:43 +00:00
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()
2015-02-06 17:26:34 +00:00
_scope = models.TextField(default='')
2015-03-20 17:38:28 +00:00
scope = property(**scope_property())
2014-12-19 15:27:43 +00:00
_id_token = models.TextField()
2014-12-19 15:27:43 +00:00
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())
2014-12-19 15:27:43 +00:00
class UserInfo(models.Model):
2015-03-27 17:42:41 +00:00
GENDER_CHOICES = [
('F', 'Female'),
('M', 'Male'),
]
2014-12-19 15:27:43 +00:00
user = models.OneToOneField(User, primary_key=True)
2015-02-18 19:23:00 +00:00
given_name = models.CharField(max_length=255, blank=True, null=True)
family_name = models.CharField(max_length=255, blank=True, null=True)
middle_name = models.CharField(max_length=255, blank=True, null=True)
nickname = models.CharField(max_length=255, blank=True, null=True)
2015-03-27 17:42:41 +00:00
gender = models.CharField(max_length=100, choices=GENDER_CHOICES, null=True)
birthdate = models.DateField(null=True)
zoneinfo = models.CharField(max_length=100, default='', blank=True,
null=True)
locale = models.CharField(max_length=100, default='', blank=True, null=True)
2015-02-18 19:23:00 +00:00
preferred_username = models.CharField(max_length=255, blank=True, null=True)
profile = models.URLField(default='', null=True, blank=True)
picture = models.URLField(default='', null=True, blank=True)
website = models.URLField(default='', null=True, blank=True)
email_verified = models.NullBooleanField(default=False)
locale = models.CharField(max_length=100, blank=True, null=True)
phone_number = models.CharField(max_length=255, blank=True, null=True)
phone_number_verified = models.NullBooleanField(default=False)
address_street_address = models.CharField(max_length=255, blank=True,
null=True)
address_locality = models.CharField(max_length=255, blank=True, null=True)
address_region = models.CharField(max_length=255, blank=True, null=True)
address_postal_code = models.CharField(max_length=255, blank=True,
null=True)
address_country = models.CharField(max_length=255, blank=True, null=True)
updated_at = models.DateTimeField(auto_now=True, null=True)
2014-12-19 15:27:43 +00:00
@property
def name(self):
name = ''
if self.given_name:
name = self.given_name
if self.family_name:
name = name + ' ' + self.family_name
2015-02-06 17:26:34 +00:00
return name
@property
def address_formatted(self):
formatted = ', '.join([
self.address_street_address,
self.address_locality,
self.address_country])
if formatted.startswith(', '):
formatted = formatted[2:]
if formatted.endswith(', '):
formatted = formatted[:-2]