U
    ~iu                     @   sT   d dl mZ d dlmZ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 )    )Path)urlparse
urlunsplit)*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pluginz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   Z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                 K   sP   ddl m} || j|}t||r8t||}|f | n|jd| d d S )Nr   r   u   Метод 'u   ' не существует)r   r   r   hasattrgetattrloggererror)r   r    Zprocess_methodkwargsr   Zjira_importmethodr   r   r   process_entity2   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>a   s     z2CmfPluginJira.get_all_projects.<locals>.<listcomp>)r   r   r   get_projectsgetappend	Exceptiongdebugmodels	CmfImportZ_get_projects)r   r.   argsr)   r   r,   r   projectZproject_dataer    r   r   r   r
   L   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   k   s     z&CmfPluginJira.test.<locals>.<listcomp>N)r   boolr7   )r   r?   r)   r   r   r   r   r	   f   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   rD   m   s    zCmfPluginJira.download_datac              
   C   s8   ddl m} | j}||| W  5 Q R  S Q R X d S rC   )r   r   r   download_filesrE   r   r   r   rF   r   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_errorsr@   )targetr#   r)   u.   Параллельный обработчик )Zonly_processing   
issue_pathZDONEu&   Ждем обработки задачu*   ------------------------ скорость u    задач в секундуu?   Закончили обработку задач, ошибок ),r   r   Zcmf.apprG   datetimecmfutilZdisable_aclZdisable_notifyr=   r>   r8   r   r   source_hashZ
CmfProjectZCmfErrorrK   Z_cache_prioritiesZsettingsCmfTaskZverbose_nameZjira_map_fieldsZqueueZQueueminZthreading_max_forksr'   infoZnowrange	threadingThreadrQ   r9   startr    Zget_project_tasksZconfigZIMPORT_ONLY_NEW_OR_UPDATEDZ	is_cancelZputjoinZtotal_secondsempty)r.   Z
project_idr`   limitr   rG   rW   errorsr    Z
api_importZproject_ext_idr@   Zissue_processrR   rS   Zthread_countsiprocZcntZissueZdiffZ
has_errorsr   r   r   process_issue_forkw   s~    


	


z CmfPluginJira.process_issue_forkc                    s\   | dg |jrP|jdkrPt|jd }|jd }t|j|jd| ddfS t |S )NrL   rZ   r   r0   zbrowse/ )	r   rL   
class_namer   r   schemenetlocsupercalc_ext_href)r   r5   Zself_urlr0   	__class__r   r   rn      s    
zCmfPluginJira.calc_ext_hrefc                    s2   | dg |jdkr&|jr&|jd S t |S )NrL   rZ   rV   )r   rj   rL   rm   calc_dump_path)r   r5   ro   r   r   rq      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rr   rs   project_dirzr+r/   rU   z	info.jsontasksr2   
attachmentc                 S   s   h | ]}t |d  qS )r/   )str)r4   rx   r   r   r   	<setcomp>
  s     z:CmfPluginJira._get_dump_stats_by_entity.<locals>.<setcomp>attachmentsrs   rr   rt   z, zcomments.jsoncomments)zipfilesetr   rL   r   r8   joinpathis_fileopenZjsonloadsaddry   is_diriterdirexistsloadstemisdigitr#   remover9   ra   )r   r@   entityrO   r}   Zattachment_statsZcomment_statsZdumped_countZdumped_objsrv   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_
        r@   =r    !=NT)rM   include_archivedd   r   z--rI   r#   zparent.coderL   )rM   r2   r   slicerH   rU   zversion:r/   CmfAttachment
CmfCommentru   )r9   rX   Zget_model_by_namerr   r~   r]   ZslistrI   splitlstripr&   rL   ry   r8   r   parentcode)r   r@   r   	filter_byrO   Zfilter_ZmodelZprocessed_countZprocessed_objsstepr`   rs   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   rU   r   r4   rf   r   r   r   rP   x  s      z4CmfPluginJira._check_consistency.<locals>.<dictcomp>c                 S   s   i | ]}|d  |d qS r   r   r   r   r   r   rP   y  s      c                    s   h | ]}| | fqS r   r   )r4   Zid_Z	dump_dictr   r   rz   |  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_consistencyf  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%   Сдамплено спринтов: rr   uL   Сбор информации об обработанных спринтахCmfListlogic_type.coder   zlist.agile_sprint:defaultr#   r   rO   u'   Обработано спринтов: u   Спринтыr   dumped	processedu   Проверка консистентности между сдампленными и обработанными спринтамиrs   T)r   u   Спринты: r   r'   r<   r   r   r   updater   r@   
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   Сдамплено релизов(версий): rr   uZ   Сбор информации об обработанных релизах(версиях)r   r   r   zlist.release:defaultr#   r   u3   Обработано релизов(версий): u   Релизыr   u   Проверка консистентности между сдампленными и обработанными релизами(версиями)rs   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+   Сдамплено компонентов: rr   uR   Сбор информации об обработанных компонентахZCmfComponentr#   rO   u-   Обработано компонентов: u   Компонентыr   u   Проверка консистентности между сдампленными и обработанными компонентамиrs   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   Сдамплено досок: rr   uH   Сбор информации об обработанных доскахZCmfKanbanBoardr#   r   u!   Обработано досок: u
   Доскиr   u   Проверка консистентности между сдампленными и обработанными доскамиrs   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   Сбор информации о сдампенных задачах, вложениях и комментарияхrw   r0   r   u   Сдамплено задач: rr   u%   Сдамплено вложений: r{   u-   Сдамплено комментариев: r|   uJ   Сбор информации об обработанных задачахrZ   u!   Обработано задач: u   Задачиr   Z	parent_idrJ   z	CmfTask:%uN   Сбор информации об обработанных вложенияхr   rL   r   N)r   u'   Обработано вложений: u   Вложенияrt   )r   r   rt   r   uT   Сбор информации об обработанных комментарияхr   Z	log_levelr   r   u/   Обработано комментариев: u   Комментарииu   Проверка консистентности между сдампленными и обработанными задачамиrs   u   Задачи: r   u   Проверка консистентности между сдампленными и обработанными вложениями)r   u   Вложения: u   Проверка консистентности между сдампленными и обработанными комментариямиu   Комментарии: )r'   r<   r   r   r   r   rB   )r   r@   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   r@   r   Zsprint_stats_dataZversion_stats_dataZcomponent_stats_dataZboard_stats_datar   r   r   r   r   project_stats}  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-   ry   r
   r	   rD   rF   rh   rn   rq   r   r   r   r   r   r   r   r   r   __classcell__r   r   ro   r   r   
   sB   
	


L
]
;  
 )%$$ar   )pathlibr   urllib.parser   r   Zcmf.includeZ#modules.jira.fields.cmf_plugin_jirar   Zmodules.jira.apir   r   r2   r   r   r   r   r   r   <module>   s
   