tt_prettyprinter.py

# -*- coding: utf-8 -*-
import string
from tt_generator import *

class HTMLPrettyPrinter:

	def __init__(self):
		self.course_colors = {}
		self.color_gen = color_generator()

	def color(self, course_name):
		if not course_name in self.course_colors.keys():
			self.course_colors[course_name] = self.color_gen.next()
		return self.course_colors[course_name]

	def print_timetables(self, timetables, total_combinations):
		table_id = 0
		print intro(len(timetables), total_combinations)
		for timetable in timetables:
			table_id += 1
			print self.format_timetable(timetable, table_id)
		print outro()

	def format_timetable(self, timetable, table_id):
	# initialize content for each cell
		content = [["<td class='period-empty-slot' >&nbsp;</td>\n" for x in xrange(32)] for y in xrange(6)]

		for weekday in range(Weekday.MONDAY, Weekday.SUNDAY):
			day_slots = [slot for slot in timetable.lessons if slot.day == weekday]
			day_slots.sort(key=lambda x: x.start.minutes, reverse=False)
			l=[] #lista de grupos de aulas por dia. cada grupo contém as aulas que se sobrepoe. as aulas que nao se sobrepoe ficam sozinhas num grupo

			super_slots = []
			while len(day_slots) != 0:
				slot = day_slots.pop(0)
				if len(super_slots) != 0:
					a = False
					for super_slot in super_slots:
						if super_slot[-1].end.minutes == slot.start.minutes:
							super_slot.append(slot)
							a = True
							break
					if not a:
						ss = []
						ss.append(slot)
						super_slots.append(ss)
				else:
					ss = []
					ss.append(slot)
					super_slots.append(ss)

			for ss in super_slots:
				ss.sort(key=lambda x: x.start.minutes, reverse=False)

			super_slots.sort(key=lambda x: x[0].start.minutes, reverse=False)

			while len(super_slots) != 0:
				grupo = []
				grupo.append(super_slots.pop(0))

				for ss in super_slots:

					fake_slot = LessonSlot(weekday, ss[0].start, ss[-1].end, None)

					if fake_slot.overlaps_with_group(grupo): #se se sobrepoe
						grupo.append(ss)
						super_slots.remove(ss)

				l.append(grupo)

			for grupo in l:
				self.format_group(grupo, content)

		# format the HTML for the timetable
		html_result = [table_intro(table_id)]
		for row in xrange(0,32):
			html_result.append("			  <tr>\n")
			html_result.append("				<th  class='period-hours' id='hour%i'>%s-%s</th>\n" % (row, time_from_index(row), time_from_index(row+1)))
			for column in xrange(0,6):
				html_result.append("				%s" % (content[column][row]))
			html_result.append("			  </tr>\n")
		html_result.append(table_outro())
		return string.join(html_result, '')


	def format_group(self, grupo, content):
		day = grupo[0][0].day
		if len(grupo) == 1: #se as aulas nao se sobrepuserem, faz como antigamente
			slot = grupo[0]
			self.format_simple_case(slot, content, slot[0].day, 0)
		else:
			earliest_start = int(min([time_index(slot[0].start) for slot in grupo]))
			latest_end = int(max([time_index(slot[-1].end) for slot in grupo]))

			height = latest_end-earliest_start + 1

			filler = "<td class='period-empty-slot' style='width: %i%% '>&nbsp;</td>\n" % (int(95/len(grupo)))
			micro_table = [[filler for x in xrange(0,height)] for y in xrange(0,len(grupo))]

			day_i = -1
			for slot in grupo:
				day_i = day_i+1
				self.format_simple_case(slot, micro_table, day_i, earliest_start)

			for row in xrange(0, height):
				s = [col[row] for col in micro_table]
				wid = 95
				for i in xrange(0, len(s)):
					w = wid/(len(s)-i)
					wid = wid - w
					s[i] = make_inner_td(s[i], w, i, len(s))

				content[day][row+earliest_start] ="<td class='cont-td'><table class='inner-table'>" + "\n".join(s) + "</table></td>"


	def format_simple_case(self, ss, content, day_i, earliest_start):
		for slot in ss:
			start_i = int(time_index(slot.start) - earliest_start)
			end_i = int(time_index(slot.end) - earliest_start)
			content[day_i][start_i] = "<td class='period-first-slot' "
			content[day_i][end_i-1] = "<td class='period-last-slot' "
			for time_i in xrange(start_i+1, end_i-1):
				content[day_i][time_i] = "<td class='period-middle-slot' "
			for time_i in xrange(start_i, end_i):
				content[day_i][time_i] += "style='background-color: %s'  " \
											% (self.color(slot.course_name()))
				content[day_i][time_i] += "headers='weekday%i hour%i'  " \
												   % (day_i, time_i)
				content[day_i][time_i] += "title='%s-%s'> " \
										% (slot.start, slot.end)
				if time_i != start_i:
					content[day_i][time_i] += "&nbsp;</td>\n"

			content[day_i][start_i] += "%s&nbsp;&nbsp;(%s)&nbsp;%s</td>\n"  \
									% (slot.course_name(), slot.lesson_category(), slot.room)


def make_inner_td(td, n, i, s):
	if not "empty" in td:
		t = td.split("style='")
		u = t[0]
		u += "style=' width: %i%% ;" % (n)
		u += t[1]
	else:
		u=td

	a = u.split('-slot')
	if i == 0:
		b = a[0] + "-slot left-inner-td" + a[1]
	else:
		if i == s-1:
			b = a[0] + "-slot right-inner-td" + a[1]
		else:
			b = a[0] + "-slot inner-td" + a[1]

	return u

def time_index(time):
	return (time.minutes - 8*60) / 30

def time_from_index(index):
	return Time(index/2 + 8, (index%2) * 30)

def color_generator():
	pallete = [ "#A8FFFF", "#AFEEEE", "#00FFFF", "#87CEFA", "#A8D4FF", \
				"#B0C4DE", "#BAEDD3", "#7FFFD4", "#51FFA9", "#40E0D0", \
				"#D8BAED", "#ADFF2F", "#32CD32", "#00FF7F", "#F9A8FF", \
				"#FFBA51", "#FF9A00", "#FFA07A", "#F7AFB3", "#FFC0A8", \
				"#FFC0CB", "#FFD700", "#FFDEAD", "#FFECA8", "#F0E68C", \
				"#FFFFFF", "#F0FFF0", "#F0FFFF", "#F8F8FF", "#F5F5F5", \
				"#FFF5EE", "#F5F5DC", "#FFFFF0", "#FDF5E6", "#FAEBD7", \
				"#FFE4E1", "#FAF0E6"]
	for color in pallete:
		yield color

def intro(total_selected, total_combinations):
	return """\
<!DOCTYPE html>
<html>
  <head>
	<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
	<title>Horários gerados</title>
	<link rel='stylesheet' type='text/css' media='screen' href='https://fenix.tecnico.ulisboa.pt/cursos/meic-a/static/css/publicPages.css' />
	<link rel='stylesheet' type='text/css' media='screen,print' href='tt-style.css' />
	<script type='text/javascript'>
	  var current = 1;
	  var total = %i;

	  function showTable(i) {
		document.getElementById('tt'+i).style.display = 'block';
	  }
	  function hideTable(i) {
		document.getElementById('tt'+i).style.display = 'none';
	  }
	  function updateLabel() {
		document.getElementById('label').innerHTML = current+'/'+total;
	  }
	  function goRight() {
		hideTable(current);
		current = (current+total)%%total+1;
		showTable(current);
		updateLabel();
	  }
	  function goLeft() {
		hideTable(current);
		current = (current+total-2)%%total+1;
		showTable(current);
		updateLabel();
	  }
	</script>
  </head>
  <body>
	<p style='text-align: center'>
	  <span id='label' style='font: bold 20px verdana'></span><br>
	  <span style='%sfont: 10px verdana'>(A mostrar os horários mais compactos dum total de %i combinações)</span>
	</p>
	<table border='0'>
	  <tr>
		<td style='width: 50px'> <img src='left.png' onclick='goLeft()'> </td>
		<td style='width: 100%%'>
""" % (total_selected, "" if (total_selected < total_combinations) else "display:none;", total_combinations)

def table_intro(id_number):
	return """\
		  <div style='display: none' id='tt%i' class='mtop15'>
			<table class='timetable' style='margin-left: auto; margin-right: auto' cellspacing='0' cellpadding='0' width='98%%'>
			  <tr>
				<th>Horas/Dias</th>
				<th colspan='1' width='14%%' id='weekday0'>Segunda</th>
				<th colspan='1' width='14%%' id='weekday1'>Terça</th>
				<th colspan='1' width='14%%' id='weekday2'>Quarta</th>
				<th colspan='1' width='14%%' id='weekday3'>Quinta</th>
				<th colspan='1' width='14%%' id='weekday4'>Sexta</th>
				<th colspan='1' width='14%%' id='weekday5'>Sábado</th>
			  </tr>
""" % (id_number)

def table_outro():
	return """\
			</table>
		  </div>
"""

def outro():
	return """\
		</td>
		<td style='width: 50px'> <img src='right.png' onclick='goRight()'> </td>
	  </tr>
	</table>
	<script type='text/javascript'>
	  showTable(current);
	  updateLabel();
	</script>
  </body>
</html>"""