Cleanup the PrimaryCharacterForm to be more "django standard"

This commit is contained in:
2011-07-08 11:14:11 +01:00
parent e201f0153e
commit 99a2c14fe6
2 changed files with 11 additions and 11 deletions

View File

@@ -125,12 +125,14 @@ class EmailChangeForm(forms.Form):
return email2
def CreatePrimaryCharacterForm(user):
""" Generate a Primary Character form populated with the user's characters """
class PrimaryCharacterForm(forms.Form):
chars = EVEPlayerCharacter.objects.filter(eveaccount__user=user)
character = forms.ModelChoiceField(queryset=None, required=True, empty_label=None)
class PrimaryCharacter(forms.Form):
character = forms.ModelChoiceField(queryset=chars, required=True, empty_label=None)
def __init__(self, *args, **kwargs):
self.user = kwargs.pop('user', None)
super(PrimaryCharacterForm, self).__init__(*args, **kwargs)
if self.user:
self.fields['character'].queryset = EVEPlayerCharacter.objects.filter(eveaccount__user=self.user)
return PrimaryCharacter