X-Git-Url: https://git.mdrn.pl/django-cas-provider.git/blobdiff_plain/a9dc703609231521ed596e6ac9fa7cd746aa6fe1..1d2632a20124e8d08a234486e3c2da81b23d5734:/cas_provider/models.py diff --git a/cas_provider/models.py b/cas_provider/models.py index c13aa32..4b75fca 100644 --- a/cas_provider/models.py +++ b/cas_provider/models.py @@ -1,4 +1,4 @@ -from django.contrib.auth.models import User +from django.conf import settings from django.db import models from django.utils.translation import ugettext_lazy as _ from random import Random @@ -29,14 +29,14 @@ class BaseTicket(models.Model): def __unicode__(self): return self.ticket - def _generate_ticket(self, length=29, chars=string.ascii_letters + string.digits): + def _generate_ticket(self, length=ticket.max_length, chars=string.ascii_letters + string.digits): """ Generates a random string of the requested length. Used for creation of tickets. """ - return u"%s-%s" % (self.prefix, ''.join(Random().sample(chars, length))) + return u"%s-%s" % (self.prefix, ''.join(Random().sample(chars, length - (len(self.prefix) + 1)))) class ServiceTicket(BaseTicket): - user = models.ForeignKey(User, verbose_name=_('user')) - service = models.URLField(_('service'), verify_exists=False) + user = models.ForeignKey(settings.AUTH_USER_MODEL, verbose_name=_('user')) + service = models.URLField(_('service')) prefix = 'ST' @@ -66,7 +66,6 @@ class LoginTicket(BaseTicket): class ProxyGrantingTicket(BaseTicket): serviceTicket = models.ForeignKey(ServiceTicket, null=True) pgtiou = models.CharField(max_length=256, verbose_name=_('PGTiou')) - targetService = models.URLField(_('service'), verify_exists=False) prefix = 'PGT' def __init__(self, *args, **kwargs):