U
    iEu                     @   sP   d dl m Z  d dlmZ d dlT d dlZd dlmZ G dd dejj	j
jZdS )    )cached_property)Path)*N)JiraApic                       s  e Zd Zejjjjjd ZdZ	ejjjjj
ddg Z
edd Zdd Zd	d
 Zdd Zdd Zedd ZedddZdd Zdd Zdd Zedd Z fddZ fddZd4d!d"Zd5d#d$Zd6d&d'Zd(d) Zd*d+ Zd,d- Zd.d/ Z d0d1 Z!d2d3 Z"  Z#S )7CmfPluginJira)jiraNtestget_all_projectsc              	   C   sb   | j s\| dddddddg t| jjj| jjj| jj | jj	 | jj
j| jjjd| _ | j S )	Npluginplugin.ext_urlzplugin.ext_loginzplugin.ext_tokenzplugin.ext_passwordzplugin.cloudzplugin.verify_ssl)urlusernametokenpassword
verify_sslcloud)_CmfPluginJira__jiraload_fieldsr   r
   ext_urlvalueZ	ext_loginZ	ext_tokenZdecryptZext_passwordr   r   )self r   (./modules/jira/models/cmf_plugin_jira.pyr      s      
 
 zCmfPluginJira.jirac              
   C   sD   ddl m} ddlm} | j}||| W  5 Q R  S Q R X d S )Nr   JiraAPIImportJiraCSVImport)modules.jira.api_importr   modules.jira.csv_importr   r   tmplt_import_settingsr   
cmf_importr   r   r   r   r   r   r      s    z#CmfPluginJira.tmplt_import_settingsc              
   C   sV   ddl m} ddlm} |jdkrJ| j}||| W  5 Q R  S Q R X || S )Nr   r   r   Zapi)r   r   r   r   typer   previewr    r   r   r   r#   !   s    
 zCmfPluginJira.previewc              
   C   sJ   ddl m} ddlm} d|_| j}||| W  5 Q R  S Q R X d S )Nr   r   r   u#   Импорт данных из Jira)r   r   r   r   namer   process_importr    r   r   r   r%   )   s
    zCmfPluginJira.process_importc                 C   sL   ddl m} || j|}t||r4t||}|  n|jd| d d S )Nr   r   u   Метод 'u   ' не существует)r   r   r   hasattrgetattrloggererror)r   r!   Zprocess_methodr   Zjira_importmethodr   r   r   process_entity0   s    

zCmfPluginJira.process_entityc                  C   s   g } | S )uR   Собственный магазин при выборе импорта из Jirar   )resr   r   r   calc_models_settings:   s    	z"CmfPluginJira.calc_models_settings)cmf_import_idc             
   O   s   ddl m} g }z`g }| jL}| D ]<}|d |d |d d}|drX|d |d< || q&W 5 Q R X W n` tk
r }	 zBt|	 t	j
j|dd	gd
}
dd || j|
 D }|s W 5 d }	~	X Y nX |S )Nr   r   idkeyr$   )r/   r0   r$   Z	isPrivater   zplugin.plugin.*r/   fieldsc                 S   s   g | ]}|qS r   r   .0objr   r   r   
<listcomp>_   s     z2CmfPluginJira.get_all_projects.<locals>.<listcomp>)r   r   r   get_projectsgetappend	Exceptiongdebugmodels	CmfImportZ_get_projects)r   r.   argskwargsr   r,   r   projectZproject_dataer!   r   r   r   r	   J   s(    

zCmfPluginJira.get_all_projectsc              
   O   s4   | j $}tdd | D W  5 Q R  S Q R X dS )uE   
        Проверяем подключение к апи
        c                 S   s   g | ]}|qS r   r   r3   r   r   r   r6   i   s     z&CmfPluginJira.test.<locals>.<listcomp>N)r   boolr7   )r   r?   r@   r   r   r   r   r   d   s    zCmfPluginJira.testc              
   C   s8   ddl m} | j}||| W  5 Q R  S Q R X d S Nr   r   )r   r   r   download_datar   r!   r   r   r   r   r   rE   k   s    zCmfPluginJira.download_datac              
   C   s8   ddl m} | j}||| W  5 Q R  S Q R X d S rD   )r   r   r   download_filesrF   r   r   r   rG   p   s    zCmfPluginJira.download_filesc              
   C   sn  ddl m} ddlm} dd l}d}t 6 t   tj	j
| ddgd}||jj|}	|jjj d| }
tjj
dd	d
|
 dgddgd}|std|
 |j|	_|	|j dd |	jtjj d D |	_g }|	j }|	j }t|	j|}|jd|  |j }t|D ]N}|	jj|	j d| |||dd}|!| |"  |	j#jd|  q d}|	j$|||t%j&dD ].}|	j#' r q|d7 }|(|d  q|D ]}|(d q|	j#jd |D ]}|)  q|j | }|	j#jd||*   d d}|+ s:||
 7 }q |	j#jd|  ||7 }W 5 Q R X W 5 Q R X |S ) Nr   r   )cmf_contextr   zplugin.plugin.source_hashr1   ::ext_idLIKEz%::z::%cust_field_conf_schemeimport_raw_json)filterr2   u)   Не найден проект с ext_id c                 S   s   i | ]}|d  |qS )titler   )r4   fieldr   r   r   
<dictcomp>   s    z4CmfPluginJira.process_issue_fork.<locals>.<dictcomp>r2   u@   Грузим задачи в несколько потоков: _process_issue_thread)issue_queueissue_queue_errorsrA   )targetr$   r@   u.   Параллельный обработчик )Zonly_processing   
issue_pathZDONEu&   Ждем обработки задачu*   ------------------------ скорость u    задач в секундуu?   Закончили обработку задач, ошибок ),r   r   Zcmf.apprH   datetimecmfutilZdisable_aclZdisable_notifyr=   r>   r8   r
   r   source_hashZ
CmfProjectZCmfErrorrL   Z_cache_prioritiesZsettingsCmfTaskZverbose_nameZjira_map_fieldsZqueueZQueueminZthreading_max_forksr(   infoZnowrange	threadingThreadrR   r9   startr!   Zget_project_tasksZconfigZIMPORT_ONLY_NEW_OR_UPDATEDZ	is_cancelZputjoinZtotal_secondsempty)r.   Z
project_idra   limitr   rH   rX   errorsr!   Z
api_importZproject_ext_idrA   Zissue_processrS   rT   Zthread_countsiprocZcntZissueZdiffZ
has_errorsr   r   r   process_issue_forku   s~    


	


z CmfPluginJira.process_issue_forkc                    sF   | ddg |jr:|jdkr:tj| jjjd|jd S t	 
|S )NrM   r   r[   Zbrowser0   )r   rM   
class_nameospathrb   r
   r   r   supercalc_ext_hrefr   r5   	__class__r   r   rn      s    zCmfPluginJira.calc_ext_hrefc                    s2   | dg |jdkr&|jr&|jd S t |S )NrM   r[   rW   )r   rj   rM   rm   calc_dump_pathro   rp   r   r   rr      s    
zCmfPluginJira.calc_dump_pathr$   c              
   C   sv  ddl }dt g d}dt d}d}t }t|jjd}	|	|}
|
 rt|
d<}|D ]0}t	
|}|t|d || f |d7 }q`W 5 Q R X n|
 rN|
 D ]}| sq|d	}| sqt|d2}t	|}|t|d || f |d7 }W 5 Q R X |d
kr|d d }dd |D }|d}| r| D ]n}| shqV|j rV|j|jkrV|j|krV|d || |jf |d  d7  < ||j qV|r|d || d|f |d}t|dJ}|D ]>}t	
|}|d || t|d f |d  d7  < qW 5 Q R X q||d}|d
krr||d< ||d< |S )u  
        Возвращает информацию о дампе сущности,
        общее количество объектов и множество объектов с кортежами вида (id, name/key).
        Для задач дополнительно добавляется информация об их вложениях и комментариях с
        множеством кортежей вида (task_key, id/name)

        Args:
            project (object): объект CmfProject
            entity (str): название сущности (название каталога или файла)
            field (str, optional): дополнительное поле сущности 
                                   для добавления в кортеж объекта вместе c id.
                                   Defaults to "name".

        Returns:
            dict: _description_
        r   N)countobjsundownloadedrs   rt   project_dirzr+r/   rV   z	info.jsontasksr2   
attachmentc                 S   s   h | ]}t |d  qS )r/   )str)r4   ry   r   r   r   	<setcomp>  s     z:CmfPluginJira._get_dump_stats_by_entity.<locals>.<setcomp>attachmentsrt   rs   ru   z, zcomments.jsoncomments)zipfilesetr   rM   r   r8   joinpathis_fileopenZjsonloadsaddrz   is_diriterdirexistsloadstemisdigitr$   remover9   rb   )r   rA   entityrP   r~   Zattachment_statsZcomment_statsZdumped_countZdumped_objsrw   Z	objs_pathfrowZobj_dataZobj_dirZobj_data_filer|   Zattachment_idsZattachments_diritemZcomment_data_fileZcomment_datastatsr   r   r   _get_dump_stats_by_entity   s    








 

 
z'CmfPluginJira._get_dump_stats_by_entityc                 C   sD  dd|gdddgg}|r"| | t|}|j|dd}t }d}	td	||	D ]}
|j|d
ddddgd|
|
|	 gd}|D ]}|jr|jdd 	dnd}|rt
||dnd}|jrt|jd }|j|}|dkr|r||jj|f q||dkr||jj|f q||r(|||f q||| q|qP||d}|S )u  
        Возвращает информацию об обработанных объектах сущности

        Args:
            project (object): объект CmfProject
            entity (str): название cmf модели
            filter_by (list, optional): дополнительный фильтр. Defaults to None.
            field (str, optional): дополнительное поле сущности 
                                   для добавления в кортеж объекта вида (id, name/key).
                                   Defaults to None.

        Returns:
            dict: _description_
        rA   =r!   !=NT)rN   include_archivedd   r   z--rJ   r$   zparent.coderM   )rN   r2   r   slicerI   rV   zversion:r/   CmfAttachment
CmfCommentrv   )r9   rY   Zget_model_by_namers   r   r^   ZslistrJ   splitlstripr'   rM   rz   r8   r   parentcode)r   rA   r   	filter_byrP   Zfilter_ZmodelZprocessed_countZprocessed_objsstepra   rt   r5   Zobj_idZ	obj_fieldr   r   r   r   _get_process_stats_by_entity+  sB    


 
z*CmfPluginJira._get_process_stats_by_entityFc           
         s   |r^dd |D  dd |D }t   t | }t   | } fdd|D }n||}|| }|sx|r|dndt|d}	|	S )	u  
        Возвращает результат проверки сдампленных и обработанных объектов
        в виде статуса и множества кортежей необработанных объектов

        Args:
            dump_objs (set): сдампленные объекты
            process_objs (set): обработанные объекты
            has_undownloaded (bool, optional): есть незагруженные вложения
            check_by_id (bool, optional): осуществлять проверку только по id. Defaults to False.
        c                 S   s   i | ]}|d  |d qS r   rV   r   r4   rg   r   r   r   rQ   t  s      z4CmfPluginJira._check_consistency.<locals>.<dictcomp>c                 S   s   i | ]}|d  |d qS r   r   r   r   r   r   rQ   u  s      c                    s   h | ]}| | fqS r   r   )r4   Zid_Z	dump_dictr   r   r{   x  s     z3CmfPluginJira._check_consistency.<locals>.<setcomp>u   НЕКОРРЕКТНОZOK)statusunprocessed)r   keysintersectionlist)
r   Z	dump_objsZprocess_objshas_undownloadedcheck_by_idZprocess_dictr   Zdiff_idsr   resultr   r   r   _check_consistencyb  s    
z CmfPluginJira._check_consistencyc                 C   s   | j d | |d}| j d|d   | j d | j|dddd	gd
d}| j d|d   d|d |d d}| j d | j|d |d dd}| j d|d   || |S )u   
        Возвращает статистику по спринтам проекта

        Args:
            project (object): объект CmfProject

        Returns:
            dict: данные статистики
        uF   Сбор информации о сдампенных спринтахZsprintsu%   Сдамплено спринтов: rs   uL   Сбор информации об обработанных спринтахCmfListlogic_type.coder   zlist.agile_sprint:defaultr$   r   rP   u'   Обработано спринтов: u   Спринтыr   dumped	processedu   Проверка консистентности между сдампленными и обработанными спринтамиrt   T)r   u   Спринты: r   r(   r<   r   r   r   updater   rA   
dump_statsprocess_stats
stats_datar   r   r   r   _get_project_sprint_stats  s4    

z'CmfPluginJira._get_project_sprint_statsc                 C   s   | j d | |d}| j d|d   | j d | j|dddd	gd
d}| j d|d   d|d |d d}| j d | |d |d }| j d|d   || |S )u   
        Возвращает статистику по релизам (версиям) проекта

        Args:
            project (object): объект CmfProject

        Returns:
            dict: данные статистики
        uT   Сбор информации о сдампенных релизах(версиях)zversions.jsonu1   Сдамплено релизов(версий): rs   uZ   Сбор информации об обработанных релизах(версиях)r   r   r   zlist.release:defaultr$   r   u3   Обработано релизов(версий): u   Релизыr   u   Проверка консистентности между сдампленными и обработанными релизами(версиями)rt   u   Релизы(версии): r   r   r   r   r   r   _get_project_version_stats  s,    

z(CmfPluginJira._get_project_version_statsc                 C   s   | j d | |d}| j d|d   | j d | j|ddd}| j d	|d   d
|d |d d}| j d | |d |d }| j d|d   || |S )u   
        Возвращает статистику по компонентам проекта

        Args:
            project (object): объект CmfProject

        Returns:
            dict: данные статистики
        uL   Сбор информации о сдампенных компонентахzcomponents.jsonu+   Сдамплено компонентов: rs   uR   Сбор информации об обработанных компонентахZCmfComponentr$   rP   u-   Обработано компонентов: u   Компонентыr   u   Проверка консистентности между сдампленными и обработанными компонентамиrt   u   Компоненты: r   r   r   r   r   r   _get_project_component_stats  s*    

z*CmfPluginJira._get_project_component_statsc                 C   s   | j d | |d}| j d|d   | j d | j|ddd}| j d	|d   d
|d |d d}| j d | |d |d }| j d|d   || |S )u   
        Возвращает статистику по доскам проекта

        Args:
            project (object): объект CmfProject

        Returns:
            dict: данные статистики
        uB   Сбор информации о сдампенных доскахZboardsu   Сдамплено досок: rs   uH   Сбор информации об обработанных доскахZCmfKanbanBoardr$   r   u!   Обработано досок: u
   Доскиr   u   Проверка консистентности между сдампленными и обработанными доскамиrt   u   Доски: r   r   r   r   r   r   _get_project_board_stats  s*    

z&CmfPluginJira._get_project_board_statsc                 C   sJ  | j d | j|ddd}| j d|d   | j d|d d   | j d	|d
 d   | j d | j|ddd}| j d|d   d|d |d d}dddg}| j d | j|ddddg|gd}| j d|d   d|d d |d d |d d}| j d | j|ddd d!g|gd}| j d"|d   d#|d
 d |d d}	| j d$ | |d% |d% }
| j d&|
d'   ||
 | j d( | j|d d% |d% t|d d d)}| j d*|d'   || | j d+ | |d
 d% |d% }| j d,|d'   |	| |||	gS )-u  
        Возвращает статистику по задачам проекта, их вложениям и комментариям

        Args:
            project (object): объект CmfProject

        Returns:
            dict: данные статистики
        ut   Сбор информации о сдампенных задачах, вложениях и комментарияхrx   r0   r   u   Сдамплено задач: rs   u%   Сдамплено вложений: r|   u-   Сдамплено комментариев: r}   uJ   Сбор информации об обработанных задачахr[   u!   Обработано задач: u   Задачиr   Z	parent_idrK   z	CmfTask:%uN   Сбор информации об обработанных вложенияхr   rM   r   N)r   u'   Обработано вложений: u   Вложенияru   )r   r   ru   r   uT   Сбор информации об обработанных комментарияхr   Z	log_levelr   r   u/   Обработано комментариев: u   Комментарииu   Проверка консистентности между сдампленными и обработанными задачамиrt   u   Задачи: r   u   Проверка консистентности между сдампленными и обработанными вложениями)r   u   Вложения: u   Проверка консистентности между сдампленными и обработанными комментариямиu   Комментарии: )r(   r<   r   r   r   r   rC   )r   rA   r   r   task_stats_dataZparent_filterZattach_process_statsZattachment_stats_dataZcomment_process_statsZcomment_stats_dataZtask_resultZattachment_resultZcomment_resultr   r   r   _get_project_task_stats  s    









z%CmfPluginJira._get_project_task_statsc           	      C   sv   g }|  |}|| | |}|| | |}|| | |}|| | |}|D ]}|| qb|S )u   
        Возвращает отчет сдампленных и обработанных объектов

        Args:
            project (object): объект CmfProject
        )r   r9   r   r   r   r   )	r   rA   r   Zsprint_stats_dataZversion_stats_dataZcomponent_stats_dataZboard_stats_datar   r   r   r   r   project_statsy  s    








zCmfPluginJira.project_stats)r$   )NN)FF)$__name__
__module____qualname__modulesr   r2   cmf_plugin_jirar   Zui_meta_skipr   Zapi_methodspropertyr   r#   r%   r+   staticmethodr-   rz   r	   r   rE   rG   ri   rn   rr   r   r   r   r   r   r   r   r   r   __classcell__r   r   rp   r   r      sB   
	


L
]
;  
 )%$$ar   )r   pathlibr   Zcmf.includeZ#modules.jira.fields.cmf_plugin_jirar   Zmodules.jira.apir   r   r2   r   r   r   r   r   r   <module>   s
   