from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """
    Для тестирования патча: ( cd /opt/crm; python3 -m patch.202205230433_tree_parent_set )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    for project in tqdm(models.CmfProject.list(filter=['project_type', '==', 'section'],
                                               fields=['parent', 'tree_parent'])):
        project.tree_parent_id = 'CmfWiki:Class'
        project.parent_id = 'CmfWiki:Class'
        project.tree_node_is_branch = True
        project.save()


if __name__ == "__main__":
    patch()
