Календарь

[r4ket4]

Хаюшки! Нужен календарик с возможностью помечать скажем каждый третий день от текущей даты. На питоне вроде можно дохера чего намутить (модуль calendar), но у меня мышление=0 - лох короче.

 

[f22]

from colorama import Fore
from datetime import datetime
import calendar


def painting_one_day(target_month):
    ''' Функция окрашивает один день в календаре'''
 
    # Получаем сегодняшний день
    curr_day = datetime.now().day
    # Форматируем
    curr_day = "{:>2}".format(curr_day)
    # Окрашиваем
    colored_day = f'{Fore.GREEN}{curr_day}{Fore.RESET}'

    if curr_day in target_month:
        target_month = target_month.replace(f'{curr_day}', f'{colored_day}', 1)

    print(target_month)


def painting_slice(target_month, step=3):
    ''' Функция окрашивает дни в календаре с указанным шагом'''

    # Получаем количество дней в месяце
    days_month = calendar.monthrange(curr_year, curr_month)[1]
 
    # Получаем сегодняшний день
    today = datetime.now().day

    # Пробегаем по всем дням начиная с сегодняшнего
    for day in range(today, days_month + 1, step):
        # Форматируем
        curr_day = "{:>2}".format(day)
        # Окрашиваем
        target_month = target_month.replace(f'{curr_day}', f'{Fore.GREEN}{curr_day}{Fore.RESET}', 1)
    print(target_month)

if __name__ == "__main__":
    curr_year = datetime.now().year
    curr_month = datetime.now().month
    month_output = calendar.month(curr_year, curr_month)
  
    painting_one_day(month_output)
    painting_slice(month_output)