from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """

    for status_opt in models.CmfStatusOpt.list(filter=['next_alarm', '>', 0], fields=['next_alarm']):
        status_opt.next_alarm *= 60 * 60
        status_opt.save()

    for status in models.CmfStatus.list(filter=['next_alarm', '>', 0], fields=['next_alarm']):
        status.next_alarm *= 60 * 60
        status.save()

if __name__ == "__main__":
    patch()
