Advertisement
Guest User

Untitled

a guest
Jul 4th, 2015
209
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.64 KB | None | 0 0
  1. class EditProjectForm(ModelForm):
  2. def __init__(self, action, *args, **kwargs):
  3. super(EditProjectForm, self).__init__(*args, **kwargs)
  4.  
  5. now = datetime.datetime.now()
  6. if action == 'edit':
  7. project_year = kwargs['project_year']
  8. self.fields['year'].choices = [(project_year, project_year)]
  9. else:
  10. self.fields['year'].choices = [(now.year, now.year), (now.year + 1, now.year + 1)]
  11.  
  12. year = forms.TypedChoiceField(coerce=int)
  13. ...
  14.  
  15. form_params = {
  16. 'project_year': datetime.datetime.now().year,
  17. }
  18. form = EditProjectForm('new', form_params)
  19. self.assertTrue(form.is_valid())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement