from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def sso_plugin():
    """
    Для тестирования патча: ( cd /opt/eva-app; python3 -m patch.202401311645_sso_plugin )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    print('Запуск патча sso_plugin')
    for sso in tqdm(models.CmfAuthOpenIdPlugin.list(fields=['plugin.type'])):
        if sso.plugin.type != 'CmfAuthOpenIdPlugin':
            sso.plugin.type='CmfAuthOpenIdPlugin'
            sso.plugin.save()


if __name__ == "__main__":
    sso_plugin()