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')
def draw(self): c.numbers = LuckyNumber.draw() return render('lucky/draw.xml')