util.py

import datetime


def get_semester(date: datetime.datetime) -> str:
    """
    Semesters are coded B161 (winter 2016/2017), B162 (summer 2016/2017) etc.

    They are obtained by comparing dates:
        January-February year X -> winter semester of year (X-1)/X
        March-September year X  -> summer semester of year (X-1)/X
        October-December year X  -> winter semester of year X/(X+1)
    :return: String representing the semester, such as B161.
    """
    year = date.year % 1000
    month = date.month

    if month in [1, 2]:
        return f"B{year-1}1"
    elif month in [3, 4, 5, 6, 7, 8, 9, 10, 11]:
        return f"B{year-1}2"
    elif month in [12]:
        return f"B{year}1"


def semester_id_to_str(semester_id):
    year = semester_id[1:3]
    wintersummer_flag = int(semester_id[3])

    year_full = 2000 + int(year)
    winsum_str = "zimní" if wintersummer_flag == 1 else "letní"

    return f'{winsum_str.capitalize()} semestr {year_full}/{year_full+1}'


def timestamp_to_date_str(timestamp: float) -> str:
    return datetime.datetime.fromtimestamp(timestamp).strftime("%d.%m.%Y")


def sanitize_fn(fn: str) -> str:
    return "".join([c for c in fn if c.isalpha() or c.isdigit() or c == ' ']).rstrip()