B
EfgA @ sT d dl m Z d dlmZ d dlmZ d dlT d dlmZ d dlZG dd deZ dS ) )cached_property)JiraBaseImport)CmfPluginImportMixin)*)fieldsNc @ sl e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z e
dd Z e dd Z
dd Ze dd ZdS )
JiraCSVImportu5 Класс загрузки объектов из CSVc C s | j }| |S )N)
cmf_importZcalc_import_excel_settings _set_default_import_settings)selfsettings r ./modules/jira/csv_import.pytmplt_import_settings
s
z#JiraCSVImport.tmplt_import_settingsc
C s t }| }t| jjd jdz}t|}xht|D ]\\}}|dkrV| jj d rVq8|
||}|s8| |\}} | r8|j r8|j
dkr8||j q8W W d Q R X tj| dS )Nr r first_rowFT)set_get_issue_id_to_code_infoopenr attachments file_pathcsvreader enumerate
json_settings_process_task_without_deal_process_personemail
does_not_workaddmodels CmfPersonZregister_persons)
r
Znew_user_emailsissue_id_to_code_infofr irowZtaskperson
is_createdr r r
process_import s
z JiraCSVImport.process_importc
C s i }x(t | jjD ]\}}|||d < qW d}d}d|krF|d }d|krV|d }d|krf|d }d|krv|d }|dks|dkri S i }t| jjd jdL}t|}x:t |D ].\}} |dkr| jj d rq| | || | < qW W dQ R X |S )
u Заранее получаем список маппинг ID задачи к Коду, так некоторые ссылки указаны по IDtitleNu ключ задачиz issue keyu' идентификатор задачиzissue idr r r )
r r csv_import_settings_fieldslowerr r r r r r )
r
Zcolumn_name_to_num_mappingr# settingZcolumn_key_numZ
column_id_numZissue_id_to_code_mappingr" r r$ r r r
r $ s.
z(JiraCSVImport._get_issue_id_to_code_infoc
C s` g }ddg g d}d}d}t | jjd jd}t|}| jjd rNt| xt|D ]\}}|dkrjP g } |d | xt| jj
D ]v\}
} | d
sq| d
d d d } | d
krd}n | dkrd}| dkr|dkr|d | d
d | ||
qW qXW W d Q R X |d r&| | |rBd|d< t j
j|d< n|r\d|d< t jj|d< |S )Nu СделкиZ handshake)r( icontitlesrowsFr r r r. model_fieldZfield_qualname.CmfTaskTr )r r2 r- Zcaptionu Задачиr( r, u Пользователи)r r r r r r r nextr appendr) splitr r2 r, r )
r
resultZdealsZis_taskZ is_personr" r r# r$ Zdata_rowjr Z
model_namer r r
previewC sD
zJiraCSVImport.previewc C s t | dsi | _|| jkrd| j|< xt| jjD ]z\}}|| |kr2|d dd dd }x>t| jjD ].\}}|d |rnd|| i| j|< P qnW | j| r2P q2W | j| S )u Получение имени автора по названию столбцу, соответствующему названию столбцу с его id
Args:
author_id: id автора
row: строчка из excel
_get_jira_comment_info_cacheNr( name)hasattrr9 r r r) r5 r*
startswith)r
Z author_idr$ r# r+ column_namer7 Zsetting2r r r
_get_jira_author_infog s
z#JiraCSVImport._get_jira_author_infoc C s t j}i }xnt| jjD ]^\}}|d r|| s2q|d d t jjkrHq|d d }|| ||< |dkr|| ||< qW |sdS t| j||\}}||fS )Nr0 verbose_name
field_namer )NF)r r r r r) rB r process_any_table_fields) r
r$
main_modelnormalized_fieldsr# r+ rC new_objr& r r r
r ~ s
z JiraCSVImport._process_personc C s: t }x.tjj D ] \}}t|tjr|| qW |S )N)r r r2 r items
issubclassZ
CmfRelBaser )r
ZrelsrC Zfieldr r r
_tasks_rel s
zJiraCSVImport._tasks_relc C s t dd| S )Nz(\!(.+?)\|.+?\!) )resub)textr r r
_delete_images s z JiraCSVImport._delete_imagesc C s t j}i }d}d }g }g }d } t t }
} i } x t| jjD ] \}
}|d }d|kr|||
r||| |||
d|kr||
r||
|kr|| ||||
d|kr||
r||
} |d r@||
sq@|d d ||
}}| ||}|d d t j
j krb|j d d
d }t
|d krb|| |d
| |d || |d
d |d d t jj kr|dkrd|i}|| |< |d d t jj krq@|dkrd}nF|dkr|}n6|dkr|r@|
| q@n|dkr |r@| | q@|| jkrFd}d|kr d}t||ts8i ||< ||| |< q@|||< q@W |r`||d< |sjd S |r|rd|i|d d< d|d d < t|d d |d d < |rd |kr|d d d | d!7 < d"|kr|d" | t| j||\}}d"|krd#d$dg}x:|D ]2}t ||r t| jt j |j t ||d% q W | rt| jt j | |j |j d&\}}t| jt j ||d% xZ|
D ]R}t j!j|d'}|st| jt j!d|i t| jt j"|t j!j|d'd% qW t#| j|| |S )(NFr( u внешняя ссылкаu родительu спринтr0 rC rB ; )maxsplit r r/ )cmf_created_at
cmf_authorrN cmf_ownerr= statusTactivity
spectatorstagsu идентификаторcodecommentsZcmf_model_activityr2 Z cmf_modelZ status_typerN z
parent responsiblerU )leftright)r= r] Z tree_parent)r= )$r r2 r r r r) r* r4 Z_get_link_for_descriptionZ_normalize_jira_value
CmfCommentrB r5 lenZ_translate_monthrA rO
CmfProjectr rJ
isinstancegetdictr Zget_status_typejoinupdaterD getattrZCmfProjectExecutorsr] ZCmfListZCmfListCmfTaskr ZCmfTaskSpectatorsZ process_tags)r
r$ r! rE rF Zset_activity_fieldZ
activity_namer\ ZlinksZ list_nameZspectator_namesZ tag_namesZ parent_infor# r+ r@ rC valueZcomment_valuesZrel_field_namerG _Zproject_executorsexecutorZlist_objZspectator_namer% r r r
r s
z(JiraCSVImport._process_task_without_dealc C s | s| S dd }ddddddddd d
d d d
}|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd|t jd |t jd d }|d |t jdi ||t jd|t jd |t jd d! |d"|t jdi xh| D ]`}xX|d# D ]L}|d$ }|||}|d% d kr>||kr>|| r>|| |d%< q>W q0W | S )&Nc S s | j |gd}|r|d S d S )N)Z fields_namer )Zimport_shop_fields)ZmodelrC Zresr r r
_get_field_name s zCJiraCSVImport._set_default_import_settings.._get_field_nameu резюмеu ключ задачиu описаниеu' идентификатор задачиu статусu ключ проектаu приоритетu
авторu создательu комментарий) Zsummaryu темаu pезюмеkey descriptionidrW ZprojectpriorityZauthorZreportercommentr= r[ rN rX rW rq r^ rU rV rT Zcmf_modified_atZ
alarm_dateZdeadliner] rY rZ )u резюмеu ключ задачиu описаниеu тип задачиu статусu приоритетu исполнительu
авторu создательu созданоu обновленоu6 пользовательское поле (start date)u срок исполненияu название проектаu наблюдателиu
меткиr r )z user namer z user statusu' руководитель проектаr r( r0 )r r2 rh ra r rc r* re )r rm Zsynonymsmappingr+ Z column_infor@ r r r
r s\
"z*JiraCSVImport._set_default_import_settingsN)__name__
__module__ __qualname____doc__r r' r r8 rA r r rJ staticmethodrO r r r r r r
r
s $ ur )
r Zmodules.jira.base_importr Z%common.models.cmf_plugin_import_mixinr Z cmf.includeZcmfr r r r r r r
s