def add_week_form(self):
"""Displays a form for adding lucky numbers
for one week (at most)."""
today = datetime.date.today()
weekday = datetime.date.weekday(today)
# If it is Saturday or Sunday today, go to the next Monday
if weekday > 4:
monday = self._closest_working_day(today)
else:
monday = today - datetime.timedelta(weekday)
# Find the first date a lucky number can be drown for
last = LuckyNumber.last()
if last.date < monday:
c.first_date = monday
else:
c.first_date = self._closest_working_day(last.date)
# Find the count of remaining days
c.count = 5 - datetime.date.weekday(c.first_date)
# Draw lucky numbers
c.numbers = LuckyNumber.draw()
# Not enough numbers to fill entire week
if len(c.numbers) < c.count:
c.count = len(c.numbers)
return render('lucky/add_form.xml')
