#!/usr/bin/python 


import os, os.path, sys
import HTMLParser


class Done(Exception):
	pass


class MyHTMLParser(HTMLParser.HTMLParser):

	id = None
	in_title = 0
	title = None

	def handle_starttag(self, tag, attrs):
		if tag == 'html':
			for name, value in attrs:
				if name =='id':
					self.id = value
		elif tag == 'title':
			self.in_title = 1
			self.title = ''

	def handle_endtag(self, tag):
		if tag == 'title':
			self.in_title - 0
		elif tag == 'head':
			raise Done

	def handle_data(self, data):
		if self.in_title:
			self.title += data


ext = '.pyhtml'
index = 'index' + ext


def walkfile(path, indent = 1, close = 0):
	source = os.path.normpath(path)
	href = source[:-len(ext)] + '.html'
	
	parser = MyHTMLParser()
	try:
		parser.feed(open(path, "rt").read())
	except Done:
		pass
	except:
		sys.stderr.write("error parsing %s\n" % path)
		raise
	else:
		raise Exception, "failed to parse %s" % path

	id = parser.id
	title = parser.title.strip()

	if close:
		sys.stdout.write('>\n')
	sys.stdout.write('\t'*indent + '<page')
	if id is not None:
		sys.stdout.write(' id="%s"' % id)
	if href is not None:
		sys.stdout.write(' href="%s"' % href)
	if title is not None:
		sys.stdout.write(' title="%s"' % title)
	

def walkdir(head, indent = 1, close = 0):
	path = os.path.join(head, index)
	if os.path.exists(path):
		walkfile(path, indent, close)
		close = 1
		
		tails = os.listdir(head)
		tails.sort()
		for tail in tails:
			path = os.path.join(head, tail)
		
			if os.path.isdir(path):
				if walkdir(path, indent + 1, close):
					close = 0
			elif path[-len(ext):] == ext and tail != index:
				walkfile(path, indent + 1, close)
				sys.stdout.write('/>\n')
				close = 0

		if close:
			sys.stdout.write('/>\n')
		else:
			sys.stdout.write('\t'*indent + '</page>\n')

		return 1
	else:
		return 0


def main():
	sys.stdout.write('<?xml version="1.0" encoding="ISO-8859-1"?>\n')
	sys.stdout.write('<sitemap')
	if walkdir('.', 1, 1):
		sys.stdout.write('</sitemap>\n')
	else:
		sys.stdout.write('/>\n')


if __name__ == '__main__':
	main()