2000-02-02 11:10:15 -04:00
|
|
|
"""Calendar printing functions"""
|
1990-10-13 16:23:40 -03:00
|
|
|
|
2000-06-28 11:48:01 -03:00
|
|
|
# Revision 2: uses functions from built-in time module
|
1990-10-13 16:23:40 -03:00
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
# Import functions and variables from time module
|
1999-05-03 15:07:40 -03:00
|
|
|
from time import localtime, mktime
|
1990-10-13 16:23:40 -03:00
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
# Exception raised for bad input (with string parameter for details)
|
1999-05-03 15:07:40 -03:00
|
|
|
error = ValueError
|
1993-06-17 09:38:10 -03:00
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
# Note when comparing these calendars to the ones printed by cal(1):
|
|
|
|
# My calendars have Monday as the first day of the week, and Sunday as
|
|
|
|
# the last! (I believe this is the European convention.)
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
# Constants for months referenced later
|
|
|
|
January = 1
|
|
|
|
February = 2
|
|
|
|
|
|
|
|
# Number of days per month (except for February in leap years)
|
1992-07-09 08:05:12 -03:00
|
|
|
mdays = [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
# Full and abbreviated names of weekdays
|
2000-02-02 11:10:15 -04:00
|
|
|
day_name = ['Monday', 'Tuesday', 'Wednesday', 'Thursday',
|
|
|
|
'Friday', 'Saturday', 'Sunday']
|
1992-07-09 08:05:12 -03:00
|
|
|
day_abbr = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
|
1990-10-13 16:23:40 -03:00
|
|
|
|
1993-06-23 06:30:50 -03:00
|
|
|
# Full and abbreviated names of months (1-based arrays!!!)
|
2000-02-02 11:10:15 -04:00
|
|
|
month_name = ['', 'January', 'February', 'March', 'April',
|
|
|
|
'May', 'June', 'July', 'August',
|
|
|
|
'September', 'October', 'November', 'December']
|
|
|
|
month_abbr = [' ', 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun',
|
|
|
|
'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
|
1990-10-13 16:23:40 -03:00
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
def isleap(year):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Return 1 for leap years, 0 for non-leap years."""
|
|
|
|
return year % 4 == 0 and (year % 100 <> 0 or year % 400 == 0)
|
1990-10-13 16:23:40 -03:00
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
def leapdays(y1, y2):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Return number of leap years in range [y1, y2).
|
|
|
|
Assume y1 <= y2 and no funny (non-leap century) years."""
|
|
|
|
return (y2+3)/4 - (y1+3)/4
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
def weekday(year, month, day):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Return weekday (0-6 ~ Mon-Sun) for year (1970-...), month (1-12), day (1-31)."""
|
|
|
|
secs = mktime((year, month, day, 0, 0, 0, 0, 0, 0))
|
|
|
|
tuple = localtime(secs)
|
|
|
|
return tuple[6]
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
def monthrange(year, month):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Return weekday (0-6 ~ Mon-Sun) and number of days (28-31) for year, month."""
|
|
|
|
if not 1 <= month <= 12: raise ValueError, 'bad month number'
|
|
|
|
day1 = weekday(year, month, 1)
|
|
|
|
ndays = mdays[month] + (month == February and isleap(year))
|
|
|
|
return day1, ndays
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
def _monthcalendar(year, month):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Return a matrix representing a month's calendar.
|
|
|
|
Each row represents a week; days outside this month are zero."""
|
|
|
|
day1, ndays = monthrange(year, month)
|
|
|
|
rows = []
|
|
|
|
r7 = range(7)
|
|
|
|
day = 1 - day1
|
|
|
|
while day <= ndays:
|
|
|
|
row = [0, 0, 0, 0, 0, 0, 0]
|
|
|
|
for i in r7:
|
|
|
|
if 1 <= day <= ndays: row[i] = day
|
|
|
|
day = day + 1
|
|
|
|
rows.append(row)
|
|
|
|
return rows
|
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
_mc_cache = {}
|
1990-10-13 16:23:40 -03:00
|
|
|
def monthcalendar(year, month):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Caching interface to _monthcalendar."""
|
|
|
|
key = (year, month)
|
|
|
|
if _mc_cache.has_key(key):
|
|
|
|
return _mc_cache[key]
|
|
|
|
else:
|
|
|
|
_mc_cache[key] = ret = _monthcalendar(year, month)
|
|
|
|
return ret
|
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
def _center(str, width):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Center a string in a field."""
|
|
|
|
n = width - len(str)
|
|
|
|
if n <= 0: return str
|
|
|
|
return ' '*((n+1)/2) + str + ' '*((n)/2)
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
# XXX The following code knows that print separates items with space!
|
|
|
|
|
|
|
|
def prweek(week, width):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Print a single week (no newline)."""
|
|
|
|
for day in week:
|
|
|
|
if day == 0: s = ''
|
|
|
|
else: s = `day`
|
|
|
|
print _center(s, width),
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
def weekheader(width):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Return a header for a week."""
|
|
|
|
str = ''
|
|
|
|
if width >= 9: names = day_name
|
|
|
|
else: names = day_abbr
|
|
|
|
for i in range(7):
|
|
|
|
if str: str = str + ' '
|
|
|
|
str = str + _center(names[i%7][:width], width)
|
|
|
|
return str
|
|
|
|
|
1994-08-01 08:34:53 -03:00
|
|
|
def prmonth(year, month, w = 0, l = 0):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Print a month's calendar."""
|
|
|
|
w = max(2, w)
|
|
|
|
l = max(1, l)
|
|
|
|
print _center(month_name[month] + ' ' + `year`, 7*(w+1) - 1),
|
|
|
|
print '\n'*l,
|
|
|
|
print weekheader(w),
|
|
|
|
print '\n'*l,
|
|
|
|
for week in monthcalendar(year, month):
|
|
|
|
prweek(week, w)
|
|
|
|
print '\n'*l,
|
1990-10-13 16:23:40 -03:00
|
|
|
|
1993-06-20 18:02:22 -03:00
|
|
|
# Spacing of month columns
|
2000-02-02 11:10:15 -04:00
|
|
|
_colwidth = 7*3 - 1 # Amount printed by prweek()
|
|
|
|
_spacing = ' '*4 # Spaces between columns
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
def format3c(a, b, c):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""3-column formatting for year calendars"""
|
|
|
|
print _center(a, _colwidth),
|
|
|
|
print _spacing,
|
|
|
|
print _center(b, _colwidth),
|
|
|
|
print _spacing,
|
|
|
|
print _center(c, _colwidth)
|
1990-10-13 16:23:40 -03:00
|
|
|
|
|
|
|
def prcal(year):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Print a year's calendar."""
|
|
|
|
header = weekheader(2)
|
|
|
|
format3c('', `year`, '')
|
|
|
|
for q in range(January, January+12, 3):
|
|
|
|
print
|
|
|
|
format3c(month_name[q], month_name[q+1], month_name[q+2])
|
|
|
|
format3c(header, header, header)
|
|
|
|
data = []
|
|
|
|
height = 0
|
|
|
|
for month in range(q, q+3):
|
|
|
|
cal = monthcalendar(year, month)
|
|
|
|
if len(cal) > height: height = len(cal)
|
|
|
|
data.append(cal)
|
|
|
|
for i in range(height):
|
|
|
|
for cal in data:
|
|
|
|
if i >= len(cal):
|
|
|
|
print ' '*_colwidth,
|
|
|
|
else:
|
|
|
|
prweek(cal[i], 2)
|
|
|
|
print _spacing,
|
|
|
|
print
|
|
|
|
|
1999-06-09 12:07:38 -03:00
|
|
|
EPOCH = 1970
|
|
|
|
def timegm(tuple):
|
2000-02-02 11:10:15 -04:00
|
|
|
"""Unrelated but handy function to calculate Unix timestamp from GMT."""
|
|
|
|
year, month, day, hour, minute, second = tuple[:6]
|
|
|
|
assert year >= EPOCH
|
|
|
|
assert 1 <= month <= 12
|
|
|
|
days = 365*(year-EPOCH) + leapdays(EPOCH, year)
|
|
|
|
for i in range(1, month):
|
|
|
|
days = days + mdays[i]
|
|
|
|
if month > 2 and isleap(year):
|
|
|
|
days = days + 1
|
|
|
|
days = days + day - 1
|
|
|
|
hours = days*24 + hour
|
|
|
|
minutes = hours*60 + minute
|
|
|
|
seconds = minutes*60 + second
|
|
|
|
return seconds
|