import zipfile

from modules.logs.log_config import get_logger
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()):
        ...
    """
    logger = get_logger('kn_client_delete_attach')
    start = 0
    step = 1000
    total = models.CmfAttachment.count(include_deleted=True)
    logger.info(f'total={total}')
    while True:
        atts = models.CmfAttachment.list(slice=[start, start + step], include_deleted=True,
                                         fields=['name', 'cmf_deleted'])
        for attach in tqdm(atts):
            try:
                if attach.cmf_deleted:
                    continue
                rdisk = models.CmfRFile.dp.data_driver.get_rd()
                rfile = rdisk.get_rfile(attach._file_name)
                if not rfile.exists or not rfile.st_size:
                    logger.info(f'Удаляем {rfile.abspath}')
                    attach.delete()
            except Exception as e:
                logger.exception(f'{attach._file_name}: {e}')
        if not atts:
            break
        start += step
        models.CmfAttachment.dp.commit()


if __name__ == "__main__":
    patch()