def __init__(self, *args, **kwargs): weekfields = {} if 'weekfields' in kwargs: weekfields = kwargs.pop('weekfields') gamefields = {} if 'gamefields' in kwargs: gamefields = kwargs.pop('gamefields') super(EditWeekForm, self).__init__(*args, **kwargs) self.initial['pick_deadline'] = get_default_pick_deadline() self.initial['lock_picks'] = weekfields.get('lock_picks') self.fields['lock_picks'] = forms.ChoiceField(widget=forms.RadioSelect, choices=((True, 'Yes'), (False, 'No'))) self.fields['pick_deadline'] = forms.DateTimeField(widget=forms.DateTimeInput) for i in range(1, 11): gamestr = 'game%d_' % i self.initial[gamestr + 'team1'] = gamefields.get(gamestr + 'team1') self.initial[gamestr + 'team2'] = gamefields.get(gamestr + 'team2') if gamefields.get(gamestr + 'favored') is not None: self.initial[gamestr + 'favored'] = 'Team%d' % gamefields[gamestr + 'favored'] self.initial[gamestr + 'spread'] = gamefields.get(gamestr + 'spread') self.initial[gamestr + 'kickoff'] = gamefields.get(gamestr + 'kickoff') self.fields[gamestr + 'team1'] = forms.ChoiceField(choices=tuple((t, t) for t in get_teamlist())) self.fields[gamestr + 'team2'] = forms.ChoiceField(choices=tuple((t, t) for t in get_teamlist())) self.fields[gamestr + 'favored'] = forms.ChoiceField(widget=forms.RadioSelect, choices=tuple(('Team%d' % i, 'Team%d' % i) for i in range(1, 3))) self.fields[gamestr + 'spread'] = forms.DecimalField(decimal_places=1) self.fields[gamestr + 'kickoff'] = forms.DateTimeField(widget=forms.DateTimeInput, required=False)
def post(self, request): form = CreateWeekForm(request.POST) if form.is_valid(): cd = form.cleaned_data year = cd.get('year') week = cd.get('week') add_year(year) w = add_week(year, week, lock_picks=True) w.pick_deadline = get_default_pick_deadline() w.save() return redirect('/pick10/commissioner/editweek/' + str(year) + '/week/' + str(week)) return render(request, 'pick10/create_week_form.html', {'form': form})