from patch.include import *
from tqdm import tqdm


@app_context(commit=True)
def set_share_companies():
    """
    Для тестирования патча: ( cd /opt/eva-app; python3 -m patch.202412020917_set_share_companies )
    Здесь можно работать с моделями через models.CmfTask и т.д.
    Для прогрессбара используйте:
    for task in tqdm(models.CmfTask.list()):
        ...
    """
    print('Запуск патча set_share_companies')
    limit = 100
    start = 0
    while True:        
        obj_list = models.CmfTask.list(
            filter=['parent_logic_prefix', '=', 'project.servicedesk'],
            fields=['company', 'cmf_owner.company'],
            slice=[start, start + limit]
        )
        start += limit

        if not obj_list:
            break
        
        for obj in tqdm(obj_list):
            # https://bcrm.carbonsoft.ru/project/Task/SD-1625010126
            obj.company = obj.company if obj.company else obj.cmf_owner.company.value
            obj.share_companies = [obj.company]
            obj.save(only_data=True)

if __name__ == "__main__":
    set_share_companies()