from calendar import month_name, day_abbr from calendar import Calendar from membership.models import Calendar as CalendarModel import datetime now = datetime.datetime.now() class CustomCalendar(Calendar): def __init__(self): super(CustomCalendar, self).__init__(firstweekday=0) def itermonthdays(self, year, month): """ Like itermonthdates(), but will yield day numbers. For days outside the specified month the day number. """ for date in self.itermonthdates(year, month): yield date.day def itermonthdays2(self, year, month): """ Like itermonthdates(), but will yield (day number, weekday number) tuples. For days outside the specified month the day number is 0. """ for date in self.itermonthdates(year, month): yield (date.day, date.weekday(), date.month) class CustomHTMLCalendar(CustomCalendar): """ This calendar returns complete HTML pages. """ # CSS classes for the day s def __init__(self, requested_month): self.requested_month = requested_month super(CustomHTMLCalendar, self).__init__() def formatday(self, day, weekday, month=None, year=None): """ Return a day as a table cell. """ booked = CalendarModel.objects.filter(user_id=self.user.id) is_booked = booked.filter(datebooked=datetime.date(day=day, month=month, year=year)) if month < int(self.requested_month): return '%d' % ("selected" if is_booked else "", day) elif month > int(self.requested_month): return '%d' % ("selected" if is_booked else "", day) else: return '%d' % ("selected" if is_booked else "", day) def formatweek(self, theweek, year): """ Return a complete week as a table row. """ s = ''.join(self.formatday(d, wd, month, year) for (d, wd, month) in theweek) return '%s' % s def formatmonthname(self, theyear, themonth, withyear=True): """ Return a month name as a table row. """ if withyear: s = '%s %s' % (month_name[themonth], theyear) else: s = '%s' % month_name[themonth] return '%s' % s def add_before(self): return '' def add_after(self): return '' def formatmonth(self, theyear, themonth, withyear=True): """ Return a formatted month as a table. """ v = [] a = v.append a('') a('\n') a(self.add_before()) a('\n') a(self.formatmonthname(theyear, themonth, withyear=withyear)) a('\n') a(self.add_after()) a('\n') a(self.formatweekheader()) a('\n') for week in self.monthdays2calendar(theyear, themonth): a(self.formatweek(week, theyear)) a('\n') a('
') a('\n') a('') return ''.join(v) class BookCalendar(CustomHTMLCalendar): def __init__(self, user, requested_month): self.user = user super(BookCalendar, self).__init__(requested_month) def formatmonth(self, year, month): self.year, self.month = year, month return super(BookCalendar, self).formatmonth(year, month) def day_cell(self, cssclass, body): return '%s' % body def formatmonthname(self, theyear, themonth, withyear): """ Return a month name as a table row. """ s = '%s' % month_name[themonth] return '%s' % s def formatweekday(self, day): """ Return a weekday name as a table header. """ ret = '%s' % day_abbr[day][0:2] return ret def formatweekheader(self): """ Return a header for a week as a table row. """ s = ''.join(self.formatweekday(i) for i in self.iterweekdays()) return '%s' % s