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})
