from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """Помечаем документы как импортированные для возможности переимпорта"""
    _filter = [['name', 'LIKE', '%drawio'], ['parent.import_original', '==', None]]
    done = []
    for attach in tqdm(models.CmfAttachment.list(filter=_filter, fields=['parent.import_original'])):
        if attach.parent.code.value in done:
            continue
        done.append(attach.parent.code.value)
        attach.parent.import_original = True
        attach.parent.value.save(only_data=True)
    print(done)


if __name__ == "__main__":
    patch()