summer_planning.py
View Options

	import datetime
	import enum
	import html.parser
	import os
	import pprint
	import re

	import colorama
	import requests

	START_DAY = datetime.date.fromisoformat("2021-06-28") # must be a monday
	END_DAY = datetime.date.fromisoformat("2021-08-31")
	INTERVAL_RE = re.compile(r"~?([0-9-]+)\s→\s~?([0-9-]+)")
	URL = "https://intranet.softwareheritage.org/api.php?action=parse&page=Summer_planning_2021&format=json"
	AUTH_LOGIN = os.environ["SWH_INTRANET_LOGIN"]
	AUTH_PASSWORD = os.environ["SWH_INTRANET_PASSWORD"]


	class Modes(enum.Enum):
	TITLE = enum.auto()
	IN_SECTION = enum.auto()
	IN_TABLE = enum.auto()


	class Parser(html.parser.HTMLParser):
	stack = []
	current_mode = None
	current_table = None
	current_person = None

	def __init__(self):
	super().__init__()
	self.tables = {}

	def handle_starttag(self, tag, attrs):
	self.stack.append((tag, attrs))
	if tag == "h2":
	self.current_mode = Modes.TITLE
	self.current_person = ""
	elif tag == "table":
	self.current_mode = Modes.IN_TABLE
	self.current_table = []
	elif tag == "tr":
	self.current_table.append([])
	elif tag in ("td", "th"):
	self.current_table[-1].append("")

	def handle_endtag(self, tag):
	while self.stack[-1][0] != tag:
	del self.stack[-1]
	if tag == "h2":
	self.current_mode = Modes.IN_SECTION
	elif tag == "table":
	self.tables[self.current_person] = self.current_table
	self.current_mode = Modes.IN_SECTION
	self.current_table = None

	def handle_data(self, data):
	data = data.strip()
	if not data:
	return
	if self.current_mode == Modes.TITLE:
	if not self.current_person:
	self.current_person = data
	elif self.current_mode == Modes.IN_TABLE:
	self.current_table[-1][-1] += data


	def print_calendar(tables):
	one_week = datetime.timedelta(weeks=1)
	six_days = datetime.timedelta(days=1)
	current_day = START_DAY

	current_line = []
	def append_cell(s, color=""):
	current_line.append(f"{color}{s:^15.15}{colorama.Style.RESET_ALL}")
	def flush_row():
	nonlocal current_line
	print(" ".join(current_line))
	current_line = []

	# header
	append_cell("week:")
	while current_day < END_DAY:
	append_cell(current_day.isoformat())
	current_day += one_week
	flush_row()

	append_cell("(date=monday)")
	flush_row()

	for (person, table) in tables.items():
	current_day = START_DAY
	intervals = [INTERVAL_RE.match(row[0]) for row in table]
	intervals = [tuple(map(datetime.date.fromisoformat, interval.groups())) for interval in intervals if interval]
	append_cell(person)
	while current_day < END_DAY:
	color = colorama.Fore.WHITE + colorama.Back.BLACK
	status = "working"
	for (start, end) in intervals:
	if start <= current_day and current_day+six_days <= end:
	color = colorama.Fore.WHITE + colorama.Back.BLUE
	status = "not working"
	elif current_day <= start <= current_day+six_days or current_day <= end <= current_day+six_days:
	color = colorama.Fore.BLACK + colorama.Back.YELLOW
	status = "partial"
	append_cell(status, color=color)
	current_day += one_week
	flush_row()


	def main():
	data = requests.get(URL, auth=(AUTH_LOGIN, AUTH_PASSWORD))
	html = data.json()["parse"]["text"]["*"]

	p = Parser()
	p.feed(html)

	print_calendar(p.tables)


	if __name__ == "__main__":
	main()

File Metadata

Mime Type: text/x-python
Expires: Jun 4 2025, 7:41 PM (10 w, 3 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3399323

summer_planning.py
No OneTemporary
Actions

summer_planning.py
View Options

File Metadata

Event Timeline

summer_planning.pyNo OneTemporaryActions

summer_planning.pyView Options

File Metadata

Event Timeline

summer_planning.py
No OneTemporary
Actions

summer_planning.py
View Options