import xmlgen

class HTMLTemplate(object):
	def __init__(self):
		with xmlgen.Tag('html') as self.doc:
			with self.doc.child('head') as self.head:
				self.title = self.head.child('title')
			self.body = self.doc.child('body')

	def set_title(self, title):
		self.title.text(title)

	def add_script(self, src, type='text/javascript', async=False, defer=False):
		script = self.head.child('script', src=src, type=type)
		if async: script.attr('async', 'async')
		if defer: script.attr('defer', 'defer')
		return script

	def add_stylesheet(self, src, type='text/css', media=None):
		stylesheet = self.head.child('link', rel='stylesheet', type=type, href=src)
		if media is not None:
			stylesheet.attr('media', media)


	def __enter__(self, *_, **__):
		return self.body
	def __exit__(self, *_, **__):
		pass

	def __str__(self):
		return unicode(self.doc).encode('utf-8')