from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """
    Устанавливает целям рабочий календарь system:default
    для тех где использовался тестовый "Пример 9-5 календарь"
    и удаляет 9-5
    """

    # заменяем тестовый календарь, на реальный
    calendar = models.CmfCalendar.get(code='system:default')
    for goal in models.CmfSDeskSlaGoal.list(filter=['calendar.name', '=', 'Пример 9-5 календарь']):
        goal.calendar = calendar
        goal.save(only_data=True)

    # удаляем тестовые календари
    for calendar in models.CmfCalendar.list(filter=['name', '=', 'Пример 9-5 календарь']):
        calendar.delete()

if __name__ == "__main__":
    patch()