mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 06:42:16 +00:00
Andrew Williams
84f625a900
Due to the way the validation worked, all cleanup() functions are ran before the errors are returned. If the username didn't exist the existing user lookup failed.
86 lines
2.9 KiB
Python
86 lines
2.9 KiB
Python
import re
|
|
|
|
from django import forms
|
|
from django.contrib.auth.models import User
|
|
|
|
from eve_api.models.api_player import EVEAccount
|
|
from sso.models import ServiceAccount, Service
|
|
from reddit.models import RedditAccount
|
|
|
|
class EveAPIForm(forms.Form):
|
|
""" EVE API input form """
|
|
|
|
user_id = forms.IntegerField(label = u'User ID')
|
|
api_key = forms.CharField(label = u'API Key', max_length=64)
|
|
description = forms.CharField(max_length=100, required=False)
|
|
|
|
def clean(self):
|
|
if not len(self.cleaned_data['api_key']) == 64:
|
|
raise forms.ValidationError("API Key provided is invalid (Not 64 characters long)")
|
|
|
|
try:
|
|
eaccount = EVEAccount.objects.get(api_user_id=self.cleaned_data['user_id'])
|
|
except EVEAccount.DoesNotExist:
|
|
return self.cleaned_data
|
|
else:
|
|
raise forms.ValidationError("This API User ID is already registered")
|
|
|
|
def UserServiceAccountForm(user):
|
|
""" Generate a Service Account form based on the user's permissions """
|
|
|
|
services = Service.objects.filter(groups__in=user.groups.all())
|
|
|
|
class ServiceAccountForm(forms.Form):
|
|
""" Service Account Form """
|
|
|
|
service = forms.ModelChoiceField(queryset=services)
|
|
username = forms.CharField(min_length=4,max_length=50)
|
|
password = forms.CharField(label = u'Password',widget = forms.PasswordInput(render_value=False))
|
|
|
|
def clean_username(self):
|
|
field = self.cleaned_data.get('username', '')
|
|
|
|
# Checks that usernames consist of letters and numbers only
|
|
if not re.match("^[A-Za-z0-9_-]*$", field):
|
|
raise forms.ValidationError("Invalid character in username, use letters and numbers only")
|
|
|
|
return field
|
|
|
|
|
|
def clean(self):
|
|
try:
|
|
acc = ServiceAccount.objects.get(service_uid=self.cleaned_data['username'],service=self.cleaned_data['service'])
|
|
except (ServiceAccount.DoesNotExist, KeyError):
|
|
pass
|
|
else:
|
|
raise forms.ValidationError("That username is already taken")
|
|
return self.cleaned_data
|
|
|
|
|
|
return ServiceAccountForm
|
|
|
|
class RedditAccountForm(forms.Form):
|
|
""" Reddit Account Form """
|
|
|
|
username = forms.CharField(label = u'User ID', max_length=64)
|
|
|
|
def clean(self):
|
|
try:
|
|
eaccount = RedditAccount.objects.get(username=self.cleaned_data['username'])
|
|
except RedditAccount.DoesNotExist:
|
|
return self.cleaned_data
|
|
else:
|
|
raise forms.ValidationError("This User ID is already registered")
|
|
|
|
class UserLookupForm(forms.Form):
|
|
""" User Lookup Form """
|
|
|
|
username = forms.CharField(label = u'User ID', max_length=64)
|
|
|
|
def clean(self):
|
|
try:
|
|
acc = User.objects.get(username=self.cleaned_data['username'])
|
|
except User.DoesNotExist:
|
|
raise forms.ValidationError("User doesn't exist")
|
|
return self.cleaned_data
|