dynamicweb/membership/calendar/calendar.py

137 lines
4.3 KiB
Python
Raw Permalink Normal View History

2016-04-23 17:00:20 +00:00
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 <td>s
def __init__(self, requested_month):
self.requested_month = requested_month
super(CustomHTMLCalendar, self).__init__()
def formatday(self, day, weekday, month=None, year=None):
2016-04-23 17:00:20 +00:00
"""
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))
2016-04-23 17:00:20 +00:00
if month < int(self.requested_month):
return '<td class="prev-month %s">%d</td>' % ("selected" if is_booked else "", day)
2016-04-23 17:00:20 +00:00
elif month > int(self.requested_month):
return '<td class="next-month %s">%d</td>' % ("selected" if is_booked else "", day)
2016-04-23 17:00:20 +00:00
else:
return '<td class="%s">%d</td>' % ("selected" if is_booked else "", day)
2016-04-23 17:00:20 +00:00
def formatweek(self, theweek, year):
2016-04-23 17:00:20 +00:00
"""
Return a complete week as a table row.
"""
s = ''.join(self.formatday(d, wd, month, year) for (d, wd, month) in theweek)
2016-04-23 17:00:20 +00:00
return '<tr>%s</tr>' % 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 '<tr><th colspan="7" class="month">%s</th></tr>' % s
def add_before(self):
2016-04-28 17:42:01 +00:00
return '<a class="btn-prev fa fa-angle-left" href="#"></a>'
2016-04-23 17:00:20 +00:00
def add_after(self):
2016-04-28 17:42:01 +00:00
return '<a class="btn-next fa fa-angle-right" href="#"></a>'
2016-04-23 17:00:20 +00:00
def formatmonth(self, theyear, themonth, withyear=True):
"""
Return a formatted month as a table.
"""
v = []
a = v.append
a('<table border="0" cellpadding="0" cellspacing="0" class="month">')
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))
2016-04-23 17:00:20 +00:00
a('\n')
a('</table>')
a('\n')
a('<span id="datesbooked"></span>')
return ''.join(v)
class BookCalendar(CustomHTMLCalendar):
def __init__(self, user, requested_month):
self.user = user
2016-04-23 17:00:20 +00:00
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 '<td>%s</td>' % body
2016-04-23 17:00:20 +00:00
def formatmonthname(self, theyear, themonth, withyear):
"""
Return a month name as a table row.
"""
s = '%s' % month_name[themonth]
return '<span id="monthtitle">%s</span>' % s
def formatweekday(self, day):
"""
Return a weekday name as a table header.
"""
ret = '<td>%s</td>' % 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 '<thead>%s</thead>' % s