* aifc.py: don't die on invalid MARK chunk

* calendar.py: remove stuff now built in time; some cleanup and
  generalization in the calendar printing
* cmd.py: use __init__.
* tzparse.py: This module is no longer necessary -- use builtin time instead!

4 files changed, 101 insertions, 130 deletions

diff --git a/Lib/aifc.py b/Lib/aifc.py
index 8c04ea3..7bdb9f0 100644
--- a/Lib/aifc.py
+++ b/Lib/aifc.py
@@ -563,11 +563,20 @@ class Aifc_read:
 
 	def _readmark(self, chunk):
 		nmarkers = _read_short(chunk)
-		for i in range(nmarkers):
-			id = _read_short(chunk)
-			pos = _read_long(chunk)
-			name = _read_string(chunk)
-			self._markers.append((id, pos, name))
+		# Some files appear to contain invalid counts.
+		# Cope with this by testing for EOF.
+		try:
+			for i in range(nmarkers):
+				id = _read_short(chunk)
+				pos = _read_long(chunk)
+				name = _read_string(chunk)
+				self._markers.append((id, pos, name))
+		except EOFError:
+			print 'Warning: MARK chunk contains only',
+			print len(self._markers),
+			if len(self._markers) == 1: print 'marker',
+			else: print 'markers',
+			print 'instead of', nmarkers
 
 class Aifc_write:
 	# Variables used in this class:
diff --git a/Lib/calendar.py b/Lib/calendar.py
index 160b8ba..9ef9e3d 100644
--- a/Lib/calendar.py
+++ b/Lib/calendar.py
@@ -1,29 +1,40 @@
-# 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
@@ -32,100 +43,34 @@ February = 2
 # 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
@@ -146,53 +91,68 @@ def _monthcalendar(year, month):
 	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):
@@ -211,8 +171,8 @@ 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
diff --git a/Lib/cmd.py b/Lib/cmd.py
index eedf613..7c512de 100644
--- a/Lib/cmd.py
+++ b/Lib/cmd.py
@@ -8,11 +8,13 @@ PROMPT = '(Cmd) '
 IDENTCHARS = string.letters + string.digits + '_'
 
 class Cmd:
-	
-	def init(self):
+
+	def __init__(self):
 		self.prompt = PROMPT
 		self.identchars = IDENTCHARS
 		self.lastcmd = ''
+	
+	def init(self):
 		return self
 
 	def cmdloop(self):
@@ -56,7 +58,7 @@ class Cmd:
 			func()
 		else:
 			import newdir
-			names = newdir.dir(self)
+			names = newdir.dir(self.__class__)
 			cmds = []
 			for name in names:
 				if name[:3] == 'do_':
diff --git a/Lib/tzparse.py b/Lib/tzparse.py
index 6cb0899..370d46b 100644
--- a/Lib/tzparse.py
+++ b/Lib/tzparse.py
@@ -48,7 +48,7 @@ def isdst(time):
 	import calendar
 	(tzname, delta, dstname, daystart, hourstart, dayend, hourend) = \
 		tzparams
-	year, month, days, hours, mins, secs, yday, wday = \
+	year, month, days, hours, mins, secs, yday, wday, isdst = \
 		calendar.gmtime(time - delta*3600)
 	return (daystart, hourstart) <= (yday+1, hours) < (dayend, hourend)