holgi
/
work-helpers
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

autoformatted with ruff

master
Holger Frey 7 months ago
parent
34125d783c
commit
6f1fdbaf6b
1 changed files with 47 additions and 37 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 84
      work_helpers/ics2table.py

84
work_helpers/ics2table.py
Unescape View File

@ -6,7 +6,7 @@ import pyperclip
import requests import requests
# typing # typing
RawEvent = dict[str:tuple[str|datetime,str]] RawEvent = dict[str : tuple[str | datetime, str]]
Event = dict[str:str] Event = dict[str:str]
TableMetaData = dict[str:int] TableMetaData = dict[str:int]
@ -16,22 +16,24 @@ ICS_DATE_FORMAT = "%Y%m%dT%H%M%S"
GERMAN_DATE_FORMAT = "%d.%m.%Y %H:%M" GERMAN_DATE_FORMAT = "%d.%m.%Y %H:%M"
EARLIEST = datetime.now() - timedelta(days=30 * 3) EARLIEST = datetime.now() - timedelta(days=30 * 3)
LATEST = datetime.now() + timedelta(days=30 * 6) LATEST = datetime.now() + timedelta(days=30 * 6)
TABLE_COLUMN_ORDER = ("date", "location", "presenter") TABLE_COLUMN_ORDER = ("date", "location", "presenter")
# for mail sending # for mail sending
MAIL_RECIPIENTS = ",".join(
MAIL_RECIPIENTS = ",".join([ [
"cpistaff@imtek.de", "cpistaff@imtek.de",
"andrea.paola.layedra.almeida@uniklinik-freiburg.de", "andrea.paola.layedra.almeida@uniklinik-freiburg.de",
"jaro.homburger@biologie.uni-freiburg.de", "jaro.homburger@biologie.uni-freiburg.de",
"kubilay.acar@continental.com", "kubilay.acar@continental.com",
"Mario.diehlmann@basf.com", "Mario.diehlmann@basf.com",
"paul.fernandez@fmf.uni-freiburg.de", "paul.fernandez@fmf.uni-freiburg.de",
"sunil.shetty@uniklinik-freiburg.de", "sunil.shetty@uniklinik-freiburg.de",
"sofia.genoves94@gmail.com", "sofia.genoves@yahoo.com.ar" "sofia.genoves94@gmail.com",
]) "sofia.genoves@yahoo.com.ar",
]
)
MAIL_SUBJECT = "Updated Group Meeting Schedule" MAIL_SUBJECT = "Updated Group Meeting Schedule"
MAIL_START = """ MAIL_START = """
Dear All, Dear All,
@ -56,23 +58,27 @@ Holgi
# parsers # parsers
def _parse_text(ics_text:str) -> str:
def _parse_text(ics_text: str) -> str:
text = ics_text.strip() text = ics_text.strip()
return text.replace("\\,", ",") return text.replace("\\,", ",")
def _parse_presenter(ics_text:str) -> str:
def _parse_presenter(ics_text: str) -> str:
text = _parse_text(ics_text) text = _parse_text(ics_text)
first, sep, last = text.partition(":") first, sep, last = text.partition(":")
return first if not sep else last.strip() return first if not sep else last.strip()
def _parse_datetime(ics_text:str) -> str:
return datetime.strptime(ics_text.strip(), ICS_DATE_FORMAT) def _parse_datetime(ics_text: str) -> str:
return datetime.strptime(ics_text.strip(), ICS_DATE_FORMAT)
def _format_datetime(date: datetime) -> str: def _format_datetime(date: datetime) -> str:
return datetime.strftime(date, GERMAN_DATE_FORMAT) return datetime.strftime(date, GERMAN_DATE_FORMAT)
class Transformer(): class Transformer:
def __init__(self, name, parser, formater): def __init__(self, name, parser, formater):
self.name = name self.name = name
self.parser = parser self.parser = parser
@ -80,37 +86,41 @@ class Transformer():
vevent_transformer = { vevent_transformer = {
"DTSTART": Transformer("date", parser=_parse_datetime, formater=_format_datetime), "DTSTART": Transformer(
"LOCATION": Transformer("location", parser=_parse_text, formater=str), "date", parser=_parse_datetime, formater=_format_datetime
"SUMMARY": Transformer("presenter", parser=_parse_presenter, formater=str), ),
"LOCATION": Transformer("location", parser=_parse_text, formater=str),
"SUMMARY": Transformer("presenter", parser=_parse_presenter, formater=str),
} }
def get_ics(url:str) -> str: def get_ics(url: str) -> str:
try: try:
response = requests.get(url) response = requests.get(url)
except IndexError: except IndexError:
msg = f"Could not load calender from url" msg = f"Could not load calender from url"
raise IOError(msg) raise IOError(msg)
return response.text return response.text
def extract_event_info(response:str) -> Iterable[Event]: def extract_event_info(response: str) -> Iterable[Event]:
events = ( events = (
event for event in event
_extract_event_info(response) for event in _extract_event_info(response)
if EARLIEST < event["date"][0] < LATEST if EARLIEST < event["date"][0] < LATEST
) )
for event in sorted(events, key=lambda d: d["date"]): for event in sorted(events, key=lambda d: d["date"]):
yield {key: values[1] for key, values in event.items()} yield {key: values[1] for key, values in event.items()}
def _extract_event_info(response:str) -> Iterable[RawEvent]:
def _extract_event_info(response: str) -> Iterable[RawEvent]:
handle = (line for line in response.splitlines()) handle = (line for line in response.splitlines())
for line in handle: for line in handle:
if line.upper().startswith("BEGIN:VEVENT"): if line.upper().startswith("BEGIN:VEVENT"):
yield extract_vevent(handle) yield extract_vevent(handle)
def extract_vevent(handle:Iterable[str]) -> RawEvent:
def extract_vevent(handle: Iterable[str]) -> RawEvent:
event = {} event = {}
for line in handle: for line in handle:
if line.upper().startswith("END:VEVENT"): if line.upper().startswith("END:VEVENT"):
@ -124,7 +134,7 @@ def extract_vevent(handle:Iterable[str]) -> RawEvent:
return event return event
def create_wiki_table(events:Iterable[Event]) -> str: def create_wiki_table(events: Iterable[Event]) -> str:
table_meta_data = _get_table_meta_data(events) table_meta_data = _get_table_meta_data(events)
headers = {key: f"'''{key.capitalize()}'''" for key in TABLE_COLUMN_ORDER} headers = {key: f"'''{key.capitalize()}'''" for key in TABLE_COLUMN_ORDER}
lines = [ lines = [
@ -135,13 +145,13 @@ def create_wiki_table(events:Iterable[Event]) -> str:
return "\n".join(lines) return "\n".join(lines)
def _create_table_line(event:Event, meta:TableMetaData) -> str: def _create_table_line(event: Event, meta: TableMetaData) -> str:
columns = [event[key].ljust(meta[key]) for key in TABLE_COLUMN_ORDER] columns = [event[key].ljust(meta[key]) for key in TABLE_COLUMN_ORDER]
inner = " || ".join(columns) inner = " || ".join(columns)
return f"|| {inner} ||" return f"|| {inner} ||"
def _get_table_meta_data(events:Iterable[Event]) -> TableMetaData: def _get_table_meta_data(events: Iterable[Event]) -> TableMetaData:
meta = {} meta = {}
for event in events: for event in events:
for key, value in event.items(): for key, value in event.items():
@ -149,12 +159,13 @@ def _get_table_meta_data(events:Iterable[Event]) -> TableMetaData:
return meta return meta
def create_email_table(events:Iterable[Event]) -> str: def create_email_table(events: Iterable[Event]) -> str:
table_meta_data = _get_table_meta_data(events) table_meta_data = _get_table_meta_data(events)
date_key, _, name_key = TABLE_COLUMN_ORDER date_key, _, name_key = TABLE_COLUMN_ORDER
pad_date = table_meta_data[date_key] pad_date = table_meta_data[date_key]
lines = [ lines = [
(event[date_key].ljust(pad_date) + " | " + event[name_key]) for event in events (event[date_key].ljust(pad_date) + " | " + event[name_key])
for event in events
] ]
return "\n".join(lines) return "\n".join(lines)
@ -166,7 +177,7 @@ def group_schedule():
wiki_table = create_wiki_table(events) wiki_table = create_wiki_table(events)
email_table = create_email_table(events) email_table = create_email_table(events)
components = [ components = [
wiki_table, wiki_table,
"", "",
@ -177,9 +188,8 @@ def group_schedule():
MAIL_SUBJECT, MAIL_SUBJECT,
MAIL_START, MAIL_START,
email_table, email_table,
MAIL_END MAIL_END,
] ]
text = "\n".join(components) text = "\n".join(components)
print(text) print(text)
pyperclip.copy(text) pyperclip.copy(text)

Write Preview
Loading…
Cancel
Save