from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """
    Для тестирования патча: ( cd /opt/crm; python3 -m patch.20220XXXXXXX_PATCHNAME )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    projects_list = models.CmfProject.list()
    for project in tqdm(projects_list):
        archive_folder = models.CmfFolder.get(tree_parent=project, name='Архив', fields=['sys_type'])
        if not archive_folder:
            archive_folder = models.CmfFolder(
                tree_parent=project,
                name='Архив',
                sys_type='archive',
                system=True)
            archive_folder.save()
        else:
            if archive_folder.sys_type != 'archive':
                archive_folder.sys_type = 'archive'
                archive_folder.system = True
                archive_folder.save(only_data=True)


if __name__ == "__main__":
    patch()