from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def checklists_items_count_fix():
    """
    Для тестирования патча: ( cd /opt/eva-app; python3 -m patch.202507020925_checklists_items_count_fix )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    print('Запуск патча checklists_items_count_fix')
    models.CmfTask.bulk_update(filter=['checklists_items_count', '==', None],
                               values={'checklists_items_count': 0})
    models.CmfTask.bulk_update(filter=['checklists_items_done_count', '==', None],
                               values={'checklists_items_done_count': 0})


if __name__ == "__main__":
    checklists_items_count_fix()