Merge pull request #455 from wwitzel3/release_3.2.0

Various SAML / auth fixes.
This commit is contained in:
Wayne Witzel III 2017-09-22 16:10:49 -04:00 committed by GitHub
commit 9e9692971f
2 changed files with 6 additions and 3 deletions

View File

@ -236,7 +236,9 @@ class AuthToken(BaseModel):
valid_n_tokens_qs = self.user.auth_tokens.filter(
expires__gt=now,
reason='',
).order_by('-created')[0:settings.AUTH_TOKEN_PER_USER]
).order_by('-created')
if settings.AUTH_TOKEN_PER_USER != -1:
valid_n_tokens_qs = valid_n_tokens_qs[0:settings.AUTH_TOKEN_PER_USER]
valid_n_tokens = valid_n_tokens_qs.values_list('key', flat=True)
return bool(self.key in valid_n_tokens)

View File

@ -136,8 +136,9 @@ class LDAPBackend(BaseLDAPBackend):
def _decorate_enterprise_user(user, provider):
user.set_unusable_password()
user.save()
enterprise_auth = UserEnterpriseAuth(user=user, provider=provider)
enterprise_auth.save()
enterprise_auth, created = UserEnterpriseAuth.objects.get_or_create(user=user, provider=provider)
if created:
enterprise_auth.save()
return enterprise_auth