from bs4 import BeautifulSoup

from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def patch():
    """
    Для тестирования патча: ( cd /opt/crm; python3 -m patch.202205230433_tree_parent_set )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    plugin = models.CmfPluginConfluence.get(order_by=['-cmf_created_at'], fields=['**'])
    cmf_import = models.CmfImport.get(order_by=['-cmf_created_at'], fields=['**'])
    from modules.confluence.api_import import ConfluenceAPIImport
    conf = ConfluenceAPIImport(plugin.confluence, cmf_import)
    g.debug(conf)
    imported_docs = models.CmfDocument.list(filter=[['cmf_import', '!=', None]], fields=['text'])
    for document in tqdm(imported_docs):
        conf.cmf_import.logger.info(f'******* Обрабатываем {document.code}')
        conf._convert_links(document)


if __name__ == "__main__":
    patch()
