from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def set_default_calendar():
    """
    Для тестирования патча: ( cd /opt/eva-app; python3 -m patch.202310131243_set_default_calendar )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    print('Запуск патча set_default_calendar')
    with cmfutil.disable_acl(), cmfutil.disable_notify():
        default_calendar = models.CmfCalendar.get(is_default=True)
        if not default_calendar:
            system_calendar = models.CmfCalendar.get(code='system:default')
            if system_calendar:
                system_calendar.is_default = True
                system_calendar.save(only_data=True)


if __name__ == "__main__":
    set_default_calendar()