-# module calendar
-
##############################
# Calendar support functions #
##############################
-# This is based on UNIX ctime() et al. (also Standard C and POSIX)
-# Subtle but crucial differences:
-# - the order of the elements of a 'struct tm' differs, to ease sorting
-# - months numbers are 1-12, not 0-11; month arrays have a dummy element 0
-# - Monday is the first day of the week (numbered 0)
-# - years are expressed in full, e.g. 1970, not 70.
-# - timezone is currently hardcoded
-# - doesn't know about daylight saving time
+# Revision 2: uses funtions from built-in time module where possible.
-# These are really parameters of the 'time' module:
-epoch = 1970 # Time began on January 1 of this year (00:00:00 UTC)
-day_0 = 3 # The epoch begins on a Thursday (Monday = 0)
+# Import functions and variables from time module
+from time import gmtime, localtime, mktime
+from time import timezone, altzone, daylight, tzname
-# Localization: Minutes West from Greenwich
-timezone = -2*60 # Middle-European time with DST on
-# timezone = 5*60 # EST (sigh -- THINK time() doesn't return UTC)
+# Exception raised for bad input (with string parameter for details)
+error = 'calendar.error'
-# Return 1 for leap years, 0 for non-leap years
-def isleap(year):
- return year % 4 == 0 and (year % 100 <> 0 or year % 400 == 0)
+# Abbreviated names of months (1-based arrays!!!)
+month_abbr = [' ', 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', \
+ 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
+
+# Turn calendar time as returned by localtime() into a string
+def asctime(arg):
+ year, month, day, hours, mins, secs, wday, yday, isdst = arg
+ return '%s %s %02d %02d:%02d:%02d %04d' % (
+ day_abbr[wday], month_abbr[month], day,
+ hours, mins, secs, year)
+
+# UNIX-style ctime (except it doesn't append '\n'!)
+def ctime(secs):
+ return asctime(localtime(secs))
+
+######################
+# Non-UNIX additions #
+######################
+
+# Calendar printing etc.
+
+# 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.)
# Constants for months referenced later
January = 1
# Number of days per month (except for February in leap years)
mdays = [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
-# Exception raised for bad input (with string parameter for details)
-error = 'calendar error'
-
-# Turn seconds since epoch into calendar time
-def gmtime(secs):
- if secs < 0: raise error, 'negative input to gmtime()'
- secs = int(secs)
- mins, secs = divmod(secs, 60)
- hours, mins = divmod(mins, 60)
- days, hours = divmod(hours, 24)
- wday = (days + day_0) % 7
- year = epoch
- # XXX Most of the following loop can be replaced by one division
- while 1:
- yd = 365 + isleap(year)
- if days < yd: break
- days = days - yd
- year = year + 1
- yday = days
- month = January
- while 1:
- md = mdays[month] + (month == February and isleap(year))
- if days < md: break
- days = days - md
- month = month + 1
- return year, month, days + 1, hours, mins, secs, yday, wday
- # XXX Week number also?
-
-# Return number of leap years in range [y1, y2)
-# Assume y1 <= y2 and no funny (non-leap century) years
-def leapdays(y1, y2):
- return (y2+3)/4 - (y1+3)/4
-
-# Inverse of gmtime():
-# Turn UTC calendar time (less yday, wday) into seconds since epoch
-def mktime(year, month, day, hours, mins, secs):
- days = day - 1
- for m in range(January, month): days = days + mdays[m]
- if isleap(year) and month > February: days = days+1
- days = days + (year-epoch)*365 + leapdays(epoch, year)
- return ((days*24 + hours)*60 + mins)*60 + secs
-
# Full and abbreviated names of weekdays
day_name = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', \
'Friday', 'Saturday', 'Sunday']
day_abbr = ['Mon', 'Tue', 'Wed', 'Thu', 'Fri', 'Sat', 'Sun']
-# Full and abbreviated of months (1-based arrays!!!)
+# Full names of months (1-based arrays!!!)
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']
-
-# Zero-fill string to two positions (helper for asctime())
-def dd(s):
- while len(s) < 2: s = '0' + s
- return s
-
-# Blank-fill string to two positions (helper for asctime())
-def zd(s):
- while len(s) < 2: s = ' ' + s
- return s
-
-# Turn calendar time as returned by gmtime() into a string
-# (the yday parameter is for compatibility with gmtime())
-def asctime(arg):
- year, month, day, hours, mins, secs, yday, wday = arg
- s = day_abbr[wday] + ' ' + month_abbr[month] + ' ' + zd(`day`)
- s = s + ' ' + dd(`hours`) + ':' + dd(`mins`) + ':' + dd(`secs`)
- return s + ' ' + `year`
-
-# Local time ignores DST issues for now -- adjust 'timezone' to fake it
-def localtime(secs):
- return gmtime(secs - timezone*60)
-
-# UNIX-style ctime (except it doesn't append '\n'!)
-def ctime(secs):
- return asctime(localtime(secs))
-######################
-# Non-UNIX additions #
-######################
+# Return 1 for leap years, 0 for non-leap years
+def isleap(year):
+ return year % 4 == 0 and (year % 100 <> 0 or year % 400 == 0)
-# Calendar printing etc.
+# Return number of leap years in range [y1, y2)
+# Assume y1 <= y2 and no funny (non-leap century) years
+def leapdays(y1, y2):
+ return (y2+3)/4 - (y1+3)/4
# Return weekday (0-6 ~ Mon-Sun) for year (1970-...), month (1-12), day (1-31)
def weekday(year, month, day):
- secs = mktime(year, month, day, 0, 0, 0)
- days = secs / (24*60*60)
- return (days + day_0) % 7
+ secs = mktime((year, month, day, 0, 0, 0, 0, 0, 0))
+ tuple = localtime(secs)
+ return tuple[6]
# Return weekday (0-6 ~ Mon-Sun) and number of days (28-31) for year, month
def monthrange(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
return rows
# Caching interface to _monthcalendar
-mc_cache = {}
+_mc_cache = {}
def monthcalendar(year, month):
- key = `year` + month_abbr[month]
- try:
- return mc_cache[key]
- except KeyError:
- mc_cache[key] = ret = _monthcalendar(year, month)
+ key = (year, month)
+ if _mc_cache.has_key(key):
+ return _mc_cache[key]
+ else:
+ _mc_cache[key] = ret = _monthcalendar(year, month)
return ret
# Center a string in a field
-def center(str, width):
+def _center(str, width):
n = width - len(str)
- if n < 0: return str
- return ' '*(n/2) + str + ' '*(n-n/2)
+ if n <= 0: return str
+ return ' '*((n+1)/2) + str + ' '*((n)/2)
# XXX The following code knows that print separates items with space!
# Print a single week (no newline)
def prweek(week, width):
for day in week:
- if day == 0: print ' '*width,
- else:
- if width > 2: print ' '*(width-3),
- if day < 10: print '',
- print day,
+ if day == 0: s = ''
+ else: s = `day`
+ print _center(s, width),
# Return a header for a week
def weekheader(width):
str = ''
+ if width >= 9: names = day_name
+ else: names = day_abbr
for i in range(7):
if str: str = str + ' '
- str = str + day_abbr[i%7][:width]
+ str = str + _center(names[i%7][:width], width)
return str
# Print a month's calendar
-def prmonth(year, month):
- print weekheader(3)
+def prmonth(year, month, *rest):
+ if rest[2:]: raise TypeError, 'too many args'
+ w = 0
+ l = 0
+ if rest[0:]: w = rest[0]
+ if rest[1:]: l = rest[1]
+ 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, 3)
- print
+ prweek(week, w)
+ print '\n'*l,
-# Spacing between month columns
-spacing = ' '
+# Spacing of month columns
+_colwidth = 7*3 - 1 # Amount printed by prweek()
+_spacing = ' '*4 # Spaces between columns
# 3-column formatting for year calendars
def format3c(a, b, c):
- print center(a, 20), spacing, center(b, 20), spacing, center(c, 20)
+ print _center(a, _colwidth),
+ print _spacing,
+ print _center(b, _colwidth),
+ print _spacing,
+ print _center(c, _colwidth)
# Print a year's calendar
def prcal(year):
for i in range(height):
for cal in data:
if i >= len(cal):
- print ' '*20,
+ print ' '*_colwidth,
else:
prweek(cal[i], 2)
- print spacing,
+ print _spacing,
print
projects / thirdparty / Python / cpython.git / commitdiff
