""" Generates a calendar along the lines of this one:: < January 2003 > Mo Tu We Th Fr Sa Su 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 It walks through all your entries and marks the dates that have entries so you can click on the date and see entries for that date. It uses the following CSS classes: - blosxomCalendar - for the calendar table - blosxomCalendarHead - for the month year header (January 2003) - blosxomCalendarWeekHeader - for the week header (Su, Mo, Tu, ...) - blosxomCalendarEmpty - for filler days - blosxomCalendarCell - for calendar days that aren't today - blosxomCalendarBlogged - for calendar days that aren't today that have entries - blosxomCalendarSpecificDay - for the specific day we're looking at (if we're looking at a specific day) - blosxomCalendarToday - for today's calendar day To use, place $calendar in your head/foot template. """ __author__ = "Will Guaraldi - willg at bluesock dot org" __version__ = "$Id: pycalendar.py,v 1.21 2004/03/02 15:54:47 willhelm Exp $" from Pyblosxom import tools import time, calendar, string def verify_installation(request): # there's no configuration needed for this plugin. return 1 class PyblCalendar: def __init__(self, request): self._request = request self._cal = None self._today = None self._view = None self._specificday = None self._entries = {} def __str__(self): """ Returns the on-demand generated string. """ if self._cal == None: self.generateCalendar() return self._cal def generateCalendar(self): """ Generates the calendar. We'd like to walk the archives for things that happen in this month and mark the dates accordingly. After doing that we pass it to a formatting method which turns the thing into HTML. """ config = self._request.getConfiguration() data = self._request.getData() entry_list = data["entry_list"] root = config["datadir"] baseurl = config.get("base_url", "") self._today = time.localtime() if len(entry_list) == 0: # if there are no entries, we shouldn't even try to # do something fancy. self._cal = "" return view = list(entry_list[0]["timetuple"]) # this comes in as 2001, 2002, 2003, ... so we can convert it # without an issue temp = data.get("pi_yr", time.strftime("%Y", self._today)) if temp: view[0] = int(temp) # the month is a bit harder since it can come in as "08", "", or # "Aug" (in the example of August). temp = data.get("pi_mo", time.strftime("%m", self._today)) if temp.isdigit(): temp = int(temp) else: if tools.month2num.has_key(temp): temp = int(tools.month2num[temp]) else: temp = view[1] view[1] = temp view = tuple(view) self._view = view # if we're looking at a specific day, we figure out what it is try: if data["pi_yr"] and data["pi_mo"] and data["pi_da"]: if data["pi_mo"].isdigit(): mon = data["pi_mo"] else: mon = tools.month2num[data["pi_mo"]] self._specificday = [data["pi_yr"], mon, data["pi_da"]] self._specificday = tuple([int(mem) for mem in self._specificday]) except: pass archiveList = tools.Walk(self._request, root) yearmonth = {} for mem in archiveList: timetuple = tools.filestat(self._request, mem) # if we already have an entry for this date, we skip to the # next one because we've already done this processing day = str(timetuple[2]).rjust(2) if self._entries.has_key(day): continue # add an entry for yyyymm so we can figure out next/previous year = str(timetuple[0]) dayzfill = string.zfill(timetuple[1], 2) yearmonth[year + dayzfill] = time.strftime("%b", timetuple) # if the entry isn't in the year/month we're looking at with # the calendar, then we skip to the next one if timetuple[0:2] != view[0:2]: continue # mark the entry because it's one we want to show datepiece = time.strftime("%Y/%b/%d", timetuple) self._entries[day] = (baseurl + "/" + datepiece, day) # create the calendar calendar.setfirstweekday(calendar.SUNDAY) cal = calendar.monthcalendar(view[0], view[1]) # insert the days of the week cal.insert(0, ["Su", "Mo", "Tu", "We", "Th", "Fr", "Sa"]) # figure out next and previous links by taking the dict of yyyymm # strings we created, turning it into a list, sorting them, # and then finding "today"'s entry. then the one before it # (index-1) is prev, and the one after (index+1) is next. keys = yearmonth.keys() keys.sort() thismonth = time.strftime("%Y%m", view) # do some quick adjustment to make sure we didn't pick # a yearmonth that's outside the yearmonths of the entries we # know about. if thismonth in keys: index = keys.index(thismonth) elif len(keys) == 0 or keys[0] > thismonth: index = 0 else: index = len(keys) - 1 # build the prev link if index == 0: prev = None else: prev = ("%s/%s/%s" % (baseurl, keys[index-1][:4], yearmonth[keys[index-1]]), "<") # build the next link if index == len(yearmonth)-1: next = None else: next = ("%s/%s/%s" % (baseurl, keys[index+1][:4], yearmonth[keys[index+1]]), ">") # insert the month name and next/previous links cal.insert(0, [prev, time.strftime("%B %Y", view), next]) self._cal = self.formatWithCSS(cal) def _fixlink(self, link): if link: return "%s" % (link[0], link[1]) else: return " " def _fixday(self, day): if day == 0: return " " strday = str(day).rjust(2) if self._entries.has_key(strday): entry = self._entries[strday] link = "%s" % (entry[0], entry[1]) else: link = strday # if it's today if (self._view[0], self._view[1], day) == self._today[0:3]: return "%s" % link if self._specificday: # if it's the day we're viewing if (self._view[0], self._view[1], day) == self._specificday: return "%s" % link # if it's a day that's been blogged if self._entries.has_key(strday): return "%s" % link return "%s" % strday def _fixweek(self, item): return "%s" % item def formatWithCSS(self, cal): """ This formats the calendar using HTML table and CSS. The output can be made to look prettier. """ cal2 = [""] cal2.append("") cal2.append("") cal2.append("") cal2.append("") cal2.append("") cal2.append("%s" % "".join([self._fixweek(m) for m in cal[1]])) for mem in cal[2:]: mem = [self._fixday(m) for m in mem] cal2.append("" + "".join(mem) + "") cal2.append("
" + self._fixlink(cal[0][0]) + "" + cal[0][1] + "" + self._fixlink(cal[0][2]) + "
") return "\n".join(cal2) def cb_prepare(args): request = args["request"] data = request.getData() if data.has_key("entry_list") and data["entry_list"]: data["calendar"] = PyblCalendar(request) # vim: tabstop=4 shiftwidth=4