def test_timetable(self):
# FIXME test expansion
# FIXME test instert times
# FIXME test map_to_slot
lectures = Lecture.objects.get_lectures(2009, Semester.SPRING, 'adamcik')
timetable = Timetable(lectures)
timetable.place_lectures()
timetable.add_markers()
rows = []
bottom = {'bottom': True}
last = {'last': True}
bottomlast = {'bottom': True, 'last': True}
lectures = dict((l.id, l) for l in lectures)
lecture2 = {'lecture': lectures[2], 'rowspan': 2, 'remove': False, 'bottom': False}
lecture3 = {'lecture': lectures[3], 'rowspan': 2, 'remove': False, 'bottom': False}
lecture4 = {'lecture': lectures[4], 'rowspan': 6, 'remove': False, 'bottom': False}
lecture5 = {'lecture': lectures[5], 'rowspan': 2, 'remove': False, 'bottom': False, 'last': True}
lecture8 = {'lecture': lectures[8], 'rowspan': 1, 'remove': False, 'bottom': False}
lecture9 = {'lecture': lectures[9], 'rowspan': 12, 'remove': False, 'bottom': True, 'last': True}
lecture10 = {'lecture': lectures[10], 'rowspan': 1, 'remove': False, 'bottom': False, 'last': True}
lecture11 = {'lecture': lectures[11], 'rowspan': 1, 'remove': False, 'bottom': True, 'last': True}
rows.append([[lecture2, lecture4, last], [lecture9], [lecture10], [last], [last]])
lecture2 = copy(lecture2)
lecture2['remove'] = True
lecture9 = copy(lecture9)
lecture9['remove'] = True
lecture9['bottom'] = False
lecture4 = copy(lecture4)
lecture4['remove'] = True
rows.append([[lecture2, lecture4, lecture5], [lecture9], [last], [last], [last]])
lecture5 = copy(lecture5)
lecture5['remove'] = True
rows.append([[lecture3, lecture4, lecture5], [lecture9], [last], [last], [last]])
lecture3 = copy(lecture3)
lecture3['remove'] = True
rows.append([[lecture3, lecture4, last], [lecture9], [last], [last], [last]])
rows.append([[{}, lecture4, last], [lecture9], [last], [last], [last]])
rows.append([[{}, lecture4, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[lecture8, {}, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[bottom, bottom, bottomlast], [lecture9], [lecture11], [bottomlast], [bottomlast]])
for i, (t, r) in enumerate(zip(timetable.table, rows)):
self.assertEquals(t, r)
def test_timetable(self):
# FIXME test expansion
# FIXME test instert times
# FIXME test map_to_slot
lectures = Lecture.objects.get_lectures(2009, Semester.SPRING, 'adamcik')
timetable = Timetable(lectures)
timetable.place_lectures()
timetable.add_markers()
rows = []
bottom = {'bottom': True}
last = {'last': True}
bottomlast = {'bottom': True, 'last': True}
lectures = dict((l.id, l) for l in lectures)
lecture2 = {'lecture': lectures[2], 'rowspan': 2, 'remove': False, 'bottom': False}
lecture3 = {'lecture': lectures[3], 'rowspan': 2, 'remove': False, 'bottom': False}
lecture4 = {'lecture': lectures[4], 'rowspan': 6, 'remove': False, 'bottom': False}
lecture5 = {'lecture': lectures[5], 'rowspan': 2, 'remove': False, 'bottom': False, 'last': True}
lecture8 = {'lecture': lectures[8], 'rowspan': 1, 'remove': False, 'bottom': False}
lecture9 = {'lecture': lectures[9], 'rowspan': 12, 'remove': False, 'bottom': True, 'last': True}
lecture10 = {'lecture': lectures[10], 'rowspan': 1, 'remove': False, 'bottom': False, 'last': True}
lecture11 = {'lecture': lectures[11], 'rowspan': 1, 'remove': False, 'bottom': True, 'last': True}
rows.append([[lecture2, lecture4, last], [lecture9], [lecture10], [last], [last]])
lecture2 = copy(lecture2)
lecture2['remove'] = True
lecture9 = copy(lecture9)
lecture9['remove'] = True
lecture9['bottom'] = False
lecture4 = copy(lecture4)
lecture4['remove'] = True
rows.append([[lecture2, lecture4, lecture5], [lecture9], [last], [last], [last]])
lecture5 = copy(lecture5)
lecture5['remove'] = True
rows.append([[lecture3, lecture4, lecture5], [lecture9], [last], [last], [last]])
lecture3 = copy(lecture3)
lecture3['remove'] = True
rows.append([[lecture3, lecture4, last], [lecture9], [last], [last], [last]])
rows.append([[{}, lecture4, last], [lecture9], [last], [last], [last]])
rows.append([[{}, lecture4, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[lecture8, {}, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[{}, {}, last], [lecture9], [last], [last], [last]])
rows.append([[bottom, bottom, bottomlast], [lecture9], [lecture11], [bottomlast], [bottomlast]])
for i, (t, r) in enumerate(zip(timetable.table, rows)):
self.assertEquals(t, r)
def pdf(request, year, semester_type, slug, size=None, week=None):
if size is not None and size not in ['A4', 'A5', 'A6', 'A7']:
raise http.Http404
semester = Semester(year=year, type=semester_type)
color_map = ColorMap(hex=True)
margin = 0.5*units.cm
width, height = pagesizes.landscape(pagesizes.A5)
width -= 2*margin
height -= 2*margin
time_width = 0.06 * width
day_width = (width-time_width) / 5
filename = '%s-%s-%s' % (year, semester.type, slug)
if week:
filename += '-%s' % week
response = http.HttpResponse(mimetype='application/pdf')
response['Content-Disposition'] = 'attachment; filename=%s.pdf' % filename
lectures = Lecture.objects.get_lectures(year, semester.type, slug, week)
rooms = Lecture.get_related(Room, lectures)
courses = Course.objects.get_courses(year, semester.type, slug)
for course in courses:
color_map[course.id]
timetable = Timetable(lectures)
if lectures:
timetable.place_lectures()
timetable.do_expansion()
timetable.insert_times()
if week:
timetable.set_week(semester.year, int(week))
paragraph_style = default_styles['Normal']
paragraph_style.fontName = 'Helvetica-Bold'
paragraph_style.fontSize = 10
paragraph_style.leading = 12
table_style = _tablestyle()
data = [[platypus.Paragraph(render_title(semester, slug, week), paragraph_style)]]
data[-1].extend([''] * sum(timetable.span))
table_style.add('SPAN', (0,0), (-1, 0))
# Add days
data.append([''])
for span, date, name in timetable.header():
if date:
data[-1].append(dateformat.format(date, 'l - j M.'))
else:
data[-1].append(unicode(name))
if span > 1:
extra = span - 1
table_style.add('SPAN', (len(data[-1])-1, 2), (len(data[-1])-1+extra, 2))
data[-1].extend([''] * extra)
# Convert to "simple" datastruct
for row in timetable.table:
data_row = []
for cells in row:
for cell in cells:
time = cell.get('time', '')
lecture = cell.get('lecture', '')
if lecture:
if lecture.type and lecture.type.optional:
paragraph_style.fontName = 'Helvetica'
code = lecture.alias or lecture.course.code
content = [platypus.Paragraph(html.escape(code), paragraph_style)]
paragraph_style.leading = 8
if lecture.type:
content += [platypus.Paragraph(u'<font size=6>%s</font>' % lecture.type.name.replace('/', ' / '), paragraph_style)]
content += [platypus.Paragraph(u'<font size=6>%s</font>' % u', '.join(rooms.get(lecture.id, [])), paragraph_style)]
paragraph_style.leading = 12
paragraph_style.fontName = 'Helvetica-Bold'
elif time:
content = time.replace(' - ', '\n')
else:
content = ''
if cell.get('remove', False):
data_row.append('')
else:
data_row.append(content)
data.append(data_row)
# Calculate widths and line that splits days
col_widths = [time_width]
for w in timetable.span:
x = len(col_widths)
table_style.add('LINEBEFORE', (x, 2), (x, -1), 1, outer_border)
col_widths.extend([float(day_width)/w] * w)
# Set row heights
row_heights = [16, 12]
row_heights += [(height-(8*2)) / (len(data)-2)] * (len(data)-2)
# Create spans, setup backgrounds and put content in KeepInFrame
for lecture in timetable.lectures:
offset = 0
for o in timetable.span[:lecture['j']]:
offset += o
x1 = offset + lecture['k'] + 1
y1 = lecture['i']+2
x2 = x1 + lecture['width'] - 1
y2 = y1 + lecture['height'] - 1
table_style.add('SPAN', (x1, y1), (x2, y2))
table_style.add('BACKGROUND', (x1, y1), (x2, y2),
colors.HexColor(color_map[lecture['l'].course_id]))
content = data[y1][x1]
data[y1][x1] = platypus.KeepInFrame(col_widths[x1]*lecture['width'],
row_heights[2]*lecture['height'],
content, mode='shrink')
page = canvas.Canvas(response, pagesizes.A4)
page.translate(margin, pagesizes.A4[1]-margin)
if 'A4' == size:
page.translate(0.5*margin, 2.5*margin-pagesizes.A4[1])
page.scale(1.414, 1.414)
page.rotate(90)
elif 'A6' == size:
page.scale(0.707, 0.707)
elif 'A7' == size:
page.scale(0.5, 0.5)
table = tables.Table(data, colWidths=col_widths, rowHeights=row_heights,
style=table_style)
table.wrapOn(page, width, height)
table.drawOn(page, 0, -height)
note = request.META.get('HTTP_HOST', '').split(':')[0]
page.setFont('Helvetica', 10)
page.setFillColor(colors.HexColor('#666666'))
page.drawString(width - page.stringWidth(note) - 2, -height+2, note)
page.showPage()
page.save()
response['X-Robots-Tag'] = 'noindex, nofollow'
return response