from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    projects = models.CmfProject.list(fields=['parent_id','project_type'])
    for obj in projects:
        if obj.project_type == 'project':
            obj.parent_id = 'CmfProject:Class'
            obj.tree_parent_id = 'CmfProject:Class'
        else:
            obj.parent_id = 'CmfWiki:Class'
            obj.tree_parent_id = 'CmfWiki:Class'
        obj.save(only_data=True)


if __name__ == "__main__":
    patch()