Talk:May Week

import itertools
from pprint import pprint
from collections import namedtuple

Day = namedtuple('Day', 'balls june_events garden_parties other')

def process_table(t):
	print('=' * 80)
	print()

	t = iter(t)

	next(t)

	yield '{| class="wikitable" style="width: 100%; table-layout: fixed"\n'
	for line in t:
		if line.startswith('! '):
			headings = [heading.strip() for heading in line[2:].split('||')]
			break

	rows = [
		[event.strip() for event in line[2:].split('||')]
		for line in t
		if not line.startswith('|-')
	]

	pprint(rows)

	days = itertools.zip_longest(*rows)
	days = [[event for event in col if event] for col in days]

	grouped_days = []
	for day in days:
		day = set(day)

		balls = {e for e in day if 'ball' in e.lower()}
		day -= balls

		june_events = {e for e in day if 'june' in e.lower()}
		day -= june_events

		garden_parties = {e for e in day if 'garden' in e.lower()}
		day -= garden_parties

		other = day

		grouped_days.append(Day(balls, june_events, garden_parties, other))

	pprint(grouped_days)

	yield '! !! May balls !! June events !! Garden parties !! Other events\n'

	for h, d in zip(headings, grouped_days):
		yield '|-\n'
		yield '\n\n'
		yield '! scope="row" | ' + h + '\n'
		yield '| <!-- balls -->\n'
		yield '\n'.join(x + '\n' for x in sorted(d.balls))
		yield '| <!-- june events -->\n'
		yield '\n'.join(x + '\n' for x in sorted(d.june_events))
		yield '| <!-- garden parties -->\n'
		yield '\n'.join(x + '\n' for x in sorted(d.garden_parties))
		yield '| <!-- other events -->\n'
		yield '\n'.join(x + '\n' for x in sorted(d.other))

	yield '|}'

	# return''.join(reversed(t))

with open('mayweek.txt') as f, open('mayweek.new.txt', 'w') as out:
	table = []
	for line in f:
		if line.startswith('{|') or table:
			table.append(line)

		if line.startswith('|}'):
			line = ''.join(process_table(table))
			table = []

		if not table:
			out.write(line)