X-Git-Url: https://git.mdrn.pl/django-cas-provider.git/blobdiff_plain/087a24803f4e6749a5c32a88c62fcb922290635d..f9c70e42b4ba7725813b8bc351b6784540d62f04:/cas_provider/models.py?ds=sidebyside diff --git a/cas_provider/models.py b/cas_provider/models.py index ec4b695..cb14c35 100644 --- a/cas_provider/models.py +++ b/cas_provider/models.py @@ -6,10 +6,14 @@ import string import urllib import urlparse +if hasattr(urlparse, 'parse_qs'): + parse_qs = urlparse.parse_qs +else: + # Python <2.6 compatibility + from cgi import parse_qs __all__ = ['ServiceTicket', 'LoginTicket'] - class BaseTicket(models.Model): ticket = models.CharField(_('ticket'), max_length=32) created = models.DateTimeField(_('created'), auto_now=True) @@ -42,19 +46,55 @@ class ServiceTicket(BaseTicket): def get_redirect_url(self): parsed = urlparse.urlparse(self.service) - query = urlparse.parse_qs(parsed.query) + query = parse_qs(parsed.query) query['ticket'] = [self.ticket] - query = [ ((k, v) if len(v) > 1 else (k, v[0])) for k, v in query.iteritems()] + query = [((k, v) if len(v) > 1 else (k, v[0])) for k, v in query.iteritems()] parsed = urlparse.ParseResult(parsed.scheme, parsed.netloc, - parsed.path, parsed.params, - urllib.urlencode(query), parsed.fragment) + parsed.path, parsed.params, + urllib.urlencode(query), parsed.fragment) return parsed.geturl() class LoginTicket(BaseTicket): - prefix = 'LT' class Meta: verbose_name = _('Login Ticket') verbose_name_plural = _('Login Tickets') + + +class ProxyGrantingTicket(BaseTicket): + serviceTicket = models.ForeignKey(ServiceTicket, null=True) + pgtiou = models.CharField(max_length=256, verbose_name=_('PGTiou')) + prefix = 'PGT' + + def __init__(self, *args, **kwargs): + if 'pgtiou' not in kwargs: + kwargs['pgtiou'] = u"PGTIOU-%s" % (''.join(Random().sample(string.ascii_letters + string.digits, 50))) + super(ProxyGrantingTicket, self).__init__(*args, **kwargs) + + class Meta: + verbose_name = _('Proxy Granting Ticket') + verbose_name_plural = _('Proxy Granting Tickets') + + +class ProxyTicket(ServiceTicket): + proxyGrantingTicket = models.ForeignKey(ProxyGrantingTicket, verbose_name=_('Proxy Granting Ticket')) + + prefix = 'PT' + + class Meta: + verbose_name = _('Proxy Ticket') + verbose_name_plural = _('Proxy Tickets') + + +class ProxyGrantingTicketIOU(BaseTicket): + proxyGrantingTicket = models.ForeignKey(ProxyGrantingTicket, verbose_name=_('Proxy Granting Ticket')) + + prefix = 'PGTIOU' + + class Meta: + verbose_name = _('Proxy Granting Ticket IOU') + verbose_name_plural = _('Proxy Granting Tickets IOU') + +