from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def create_cmdb_folders():
    """
    Для тестирования патча: ( cd /opt/eva-app; python3 -m patch.202408081549_create_cmdb_folders )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    print('Запуск патча create_cmdb_folders')

    with cmfutil.disable_acl(), cmfutil.disable_notify():
        asset_scheme = models.CmfAssetScheme.get(asset_code_prefix="AST")

        if asset_scheme:
            asset_scheme._scaffold_assets()


if __name__ == "__main__":
    create_cmdb_folders()