U
    _d<y                     @   s   d dl Z d dlmZmZ d dlmZ d dlmZmZ d dlmZ d dl	Z	d dl
mZ d dlmZmZ d dlZd dlmZ d dlZd dlZd d	lmZ d d
lT d dlmZ d dlmZ d dlmZ G dd dejZdS )    N)copydeepcopy)Decimal)datetimetimezone)Logger)Path)DictIterator)cached_property)commit_all_ds)*)fields
cmf_import)
log_configc                       s<  e Zd Zejjd ZdZeedddZ	dd Z
eddd	Zd
d Z fddZdd Zdd Zdd Zed9ddZedd Zeejdd Zd:edddZdd  Zedd!d"Zd#d$ Zd%d& Zd'd( Zeejddd)d*Zd+d, Z  fd-d.Z! fd/d0Z"d1d2 Z#e$dd3d4Z%d5d6 Z&d7d8 Z'  Z(S );	CmfImport)CSV_SETTING_NAMEcsv_import_settings_fieldsloggerZCSV)returnc                 C   s(   t j}tjrt j}tj|  dd|dS )uE    Журнал логирования процесса импорта    i@KL )ZbackupCountZmaxByteslevel)loggingINFOconfigDEBUGr   Z
get_loggerget_files_dir)selfr    r   ./common/models/cmf_import.pyr      s    zCmfImport.loggerc                 O   sF   t j|  tj}t jt jtj|s8| j	
d t jd|S )uK   
        Возвращает путь текущему файлу
        initfiles)ospathjoinr   r   LOG_NAMEexistsr   
UPLOAD_DIRr   info)r   argskwargs	file_pathr   r   r    get_log_filename"   s    zCmfImport.get_log_filenamec                    s`   t j tj}t jt jtj|s8j	
d d  fddt  D }|S )u   
        Список файлов после logrotate, чтобы можно было загрузить историю
        r!   Tc                    sB   g | ]:}t jt j |rtd |rt jd |qS )zdata\.log(\.\d+)?$r"   )r#   r$   isfiler%   rematchr   ).0fdir_pathr   r   r    
<listcomp>3   s    z/CmfImport.get_log_filenames.<locals>.<listcomp>)r#   r$   r%   r   r   r&   r'   r   r(   r   r)   listdir)r   r*   r+   r,   Z	file_listr   r3   r    get_log_filenames+   s    
zCmfImport.get_log_filenamesc              	   O   s   dd l }t| d}|d}tj|r6t| ||d,}| 	 D ]}|
dtj}|| qLW 5 Q R X tjd|  dS )Nr   Tzall_logs.zipwz/filesr"   )zipfiler   r   joinpathr#   r$   r'   removeZipFiler7   replacer   r(   writer%   )r   r*   r+   r9   r4   Zzip_pathZmyzipr,   r   r   r    get_all_logs_by_zip8   s    

zCmfImport.get_all_logs_by_zipc                    s$   | j s| jjdd| _ t   d S )NZOPENstatus_type)statusworkflowget_default_statussuper_calc_statusr   	__class__r   r    rF   D   s    zCmfImport._calc_statusc                 C   s  | j d rd S | jr(| j | j d< d S | jdkr6d S | j d dtj tjjd | j d dtj	 tj	jd | j d dtj
 tj
jd | j d dtj tjjd | j d dtj tjjd | j d d	tj tjjd d S )
NZmodels_fieldsexcelu   Поля контактов)Zcaptionr   iconu   Поля задачu   Поля заметокu   Поля проектовu"   Поля списков задачu   Поля групп)json_settingspluginZcalc_models_settingstypeappendmodels	CmfPersonZimport_shop_fieldsrK   CmfTaskZ
CmfComment
CmfProjectCmfListZCmfPersonGrouprG   r   r   r    _calc_models_settingsI   s>    

zCmfImport._calc_models_settingsc                 K   sH   |  ddddg | jd r d S | jj| d| jd< | j| _|   d S NrL   attachmentsrN   rM   import_settingsr   )load_fieldsrL   rM   calc_import_settingssaver   r+   r   r   r    rZ   d   s    
zCmfImport.calc_import_settingsc                 K   sH   |  ddddg | jd r d S | jj| d| jd< | j| _|   d S rV   )rY   rL   rM   tmplt_import_settingsr[   r\   r   r   r    r]   n   s    
zCmfImport.tmplt_import_settings c                 C   s4   d|  d d} td|  d|  t|  d S )Nu1   Ошибка при импорте данных. .ue   . Проверьте загружаемый файл и повторите попытку снова.zCmfImportError: : )stripr   errorZ	cmf_alert)msger   r   r    
_log_errorv   s    zCmfImport._log_errorc                 C   s:   |  dg | jd D ]}|d | jkr|d   S qg S )NrL   rX   namer   )rY   rL   r   )r   settingsr   r   r    r   |   s
    z$CmfImport.csv_import_settings_fieldsc              
   O   s*  dt jd< t  t  tjj| ddddgd}|j	d |j
jd	d
d|_|  t  | }|r|j	d |j
jdd|_|  n0|jjdkr|j	d |j
jdd|_|  |jjd jtjd}W 5 Q R X W 5 Q R X tjj|tjjj|jdttj dd| ddd dS )uJ   
        Запускаем в фоне импорт данных
        1importrC   rB   progresszplugin.plugin.*idr   u%   ***Запускаем импорт***ZIN_PROGRESSimport-started)rA   status_codeu2   ***Импорт успешно завершен***ZCLOSEDr@   u7   ***Импорт завершен с ошибками***import-errorrn   r   z/files/u+   Импорт данных завершен z%d.%m.%Y %H:%Mz
, <a href=u(   >Журнал логирования</a>T)objZ	person_idrf   rc   Zis_notify_current_personN)!r#   environcmfutildisable_acldisable_notifyrP   r   getr   r)   rC   rD   rB   r[   commit_with_eventprocess_importrA   ZhandlersZbaseFilenamer=   r   r(   Z	CmfNotifyZplace_notifygZcurrent_personrl   valuerf   r   nowr   utc)Z	import_id_args_kwargsrq   resZlog_pathr   r   r    start   s0    

,zCmfImport.startN)rowc              	   C   s4  d}|dkr|j }|sg }i }t| jD ]Z\}}|d r(|| r(|d d dd |jkr^q(|d d |kr(|| ||d d < q(|s|S || d}	tt|	d	}
|D ]}|j|}t	|tj
st	|tjr,d}| D ]8}|jd
dd|| gd	d|| ggdgd}|r qq|r,t|	|| qt	|tjr<qt|	|||  || }t	|tjr~t|d}t|tjj}t	|tjrq||d|g q| jd r|j|dgd}|s0|d	}|r|j|dgdd}|r|
|	_|j|	jdgd}|r$|	j dttjd|	_|	}|  |S )u  
        Импортируем объект

        :param list row: Строка из файла
        :param models.* model: Модель для создания объекта
        :param list _filter: Для фильтрации объекта
        :param str key: Ключевое имя для сопоставления модели

        :return: Новый или найденный объект
        :rtype: Optional[models.*]
        Nmodel_fieldfield_qualnamer_   r   verbose_name
field_namer   codeORrf   ==z**)filterr   ZRUunionT)r   r   Zinclude_deleted)rf   r   _z%Y%m%d-%H%M%S)r   	enumerater   split
class_namestrgetattrr   rv   
issubclassZCmfRelationZCmfGenericRelationZrelated_modelssetattrZ
CmfRelBaseZCmfPhonephonenumbersparseZformat_numberZPhoneNumberFormatZINTERNATIONALZ
CmfNumericrO   rL   r   rf   r   r{   r   r|   r[   )r   r   model_filterkeyresult_fieldsisettingZnew_objZnew_coder   ZfieldZrel_objZ	rel_modelvalZphoner   r   r   r    _process_obj   sp    

zCmfImport._process_objc              
   C   s   |  dddddg | jrz0t  | jj| dW  5 Q R  W S Q R X W n tk
r } zl| dt  }t| t	| dt
|jd	  }| j| || _| jjd
d| _|   W Y dS d}~X Y nX | jdkr|  S dS )u9   
        Импортируем файл в бд
        rJ   rX   	first_rowrM   rW   r   r`   
ro   rp   FN)rY   rM   rs   enable_import_moderx   	Exception	traceback
format_excr   r   	format_tb__traceback__r   	exception	error_msgrC   rD   rB   r[   rN   _process_import_excel)r   errerr_msg
error_textr   r   r    rx      s     
&
zCmfImport.process_importc           4      C   sf  ddl m} | jr| jd s dS d}zi }i }i }i }i }i }i }	|  }
t|
D ](\}}|dkrr| jd rrqTi }i }d}d}t| jD ],\}}|d r|| sq|d d d	d }||i }|d d
 d	}|d }|| }|dkr|}qn|dkr|}qt	|dkrP|dkr2|||d < q||i }|| ||d < nZ|dkrt|dkrt|
|| }n6|dkr||| }n|dkrd||| i}|||< |||< q|di }|dd}||d}|s|| tj|\}}|||d< || tj|\}}|di }|drD|d |d< |di }|r|	|}|s|dkrtjjddtjjddd}ntjjddtjjddd}||	|< ||d < ||d< ||d!< ||d"< d#d$d%|jgdd%|d gg}|| tj||\} }| j| |   | ||d < |rH||g }!|!|  |!||< |rt||d g }"|"| |"||d < |d&i d'd}#|#r|# d(|jj }$||$i }%||%d!< |#|%d'< |%d)g }&|&|  |&|%d)< |%||$< ||d}'|'sftjjd#d*d%|jgd'd%d+ggd,}(d#d*d%|(jgd'd%d-gg})d-||(d.d/}*|j| tj|*|)d0\}'}|'||d< || jtj|'| d1 qT|  D ]\}$}+tjjd#d*d%|+d! jgd'd%d2ggd,}(d#d*d%|(jgd'd%|+d' gdd3d4gg})tjj|)d,},|,stj|+d! |(|+d' tj!jd5dd6},|,  |+d) D ]} |,| _"| jd7d8 q"q|  D ]>\}}-||}.|.rH|-D ]}/|/j"#  |.|/_"|/  qdqH|  D ]N\}}0||}.|.r|0D ].}1||1}/|/r|/j"#  |.|/_"|/  qqW n~ t$k
r` }2 z^t%|2 d9| d:|2 d;t&'|2j(d<  }3| j)*|3 |3| _+| j,j-d=d>| _.|   W Y dS d?}2~2X Y nX d7S )@u   
        Запуск импорта из CSV

        :return: Истина в случае успешного импорта
        r   )CmfPluginImportMixinrX   Fr   r^   r   r   r_   r   parent_taskZchild_tasks   	executorslogic_prefixrS   priorityrB   rA   zproject.agiler   rR   Z
cmf_authorZ	cmf_ownerzproject.basez
task.base%baser   )Z	cmf_modelprefixZactivityztask.agile:softdev%ZsoftdevrC   projectparentZANDZ	parent_idr   rT   rf   r   tasksZtree_parent_idZSprints)r   u
   ДоскаZkanban)rf   r   tree_parentZ	list_type)r   r   
obj_fields
filter_obj)leftrightZEpics=z	task.epicztask.epic:default)r   r   rf   Z
logic_typeT)Z	only_datau   Строка r`   r   r   ro   rp   N)/Z%common.models.cmf_plugin_import_mixinr   rW   rL   Z	_get_filer   r   r   rv   lenZget_logic_prefixZget_priorityZget_status_type
setdefaultZprocess_any_table_fieldsrP   rS   rQ   CmfWorkflowZcalc_workflowZCmfActivityrl   rR   r   rO   r[   r   rz   Z	CmfFolderrT   r   ZCmfListCmfTaskitemsZCmfLogicTyper   loadr   r   r   r   r   r   r   r   rC   rD   rB   )4r   r   Z
number_rowZall_parent_taskZall_child_tasksZall_epic_tasksZ	all_tasksZ
cash_boardZcash_projectZcash_workflow_taskZ	file_readr   executorr   Zparent_task_codeZchild_tasks_codeidxr   Z
model_nameZ_rowZfields_namer   rz   Zproject_fieldsr   r   r   Zexecutor_objZtask_fieldsZvalue_statusrC   Zfilter_taskZtask_objZtmp_parent_tasksZtmp_child_tasksZ	epic_nameZepic_keyZtmp_dataZ	tmp_tasksZlist_objZparent_folderr   r   dataZepic_objr   r   Zsub_taskZ
tasks_codeZsub_task_coder   r   r   r   r    r     sL   








 



 




"zCmfImport._process_import_excelc                 C   s>  |  |tj}|r|sdS d}| jD ] }|d r$|d d dkr$|}q$|dg tjj|jt|d |j	jdgd}|stj|t|d |j	| d	}|
  |jjd
d t|jjD ]*}|jj| jkrq|  |jj| q| j |tjtjjdd|jgdd|jggd}|ddg ||_||_|jr:|
  |S )u;   
        Импорт товара в сделку
        Nr   r   zCmfProduct.pricescurrencyfile_valprice)
product_idr   Zcurrency_idr   )productr   r   r   T)forceZdeal_idr   r   )r   r   r   Zsales_order)r   rP   Z
CmfProductr   rY   ZCmfPricerv   rl   r   r   r[   Zpricesr   r   Z
deal_itemsrz   r   deleter;   ZCmfDealItemsr   dealr   
is_changed)r   r   r   r   Zprice_settingrg   r   itemr   r   r    _process_product  s@    
 zCmfImport._process_productc           	      C   s  |s|r|j sd S d }| jD ] }|d r|d d dkr|}q|sFd S g }tjj|d d}|sxtj|d d}|  |dd|jg |r|dd|jg n|dd|j jg | |tj	|}|
d	d
g |j s|r||_ n|j |_ |js||_|jr
|  |S )Nr   r   zCmfDeal.pipeliner   )rf   Zpipeline_idr   Z
company_idcompanypipeline)r   r   rP   ZCmfPipelinerv   r[   rO   rl   r   ZCmfDealrY   r   r   )	r   r   Zcontactr   Zpipeline_settingrg   r   r   r   r   r   r    _process_deal  s8    
zCmfImport._process_dealc                 C   s   dd l }tj  |  }||\}}}t|| d }|dkrZ| jd| d dS t	j
j| jdgd}|jjd	krd
|_|jr|  t  dS dS )Nr   d      u8   Импорт прерван, место на диске z% < 20%TrB   rk   )import-cancelingimport-canceledro   import-stoppingr   F)shutilrs   r   Zimport_heartbeatget_download_path
disk_usageintr   rb   rP   r   rv   rl   rB   r   r   r[   rw   )r   r   Zdownload_pathtotalusedfreeZfree_percentrq   r   r   r    	is_cancel  s     
zCmfImport.is_cancelc                  O   s0  t   t   tjj|d ddgd}|jjdkr|jjdkrf|jj	dd|_|
  t  td	D ]t}td
 tjj|jddgd}|jjdkr q|dkrnd|_|jj	dd|_|
   W 5 Q R  W 5 Q R  dS qnn|jj	dd|_|
  t  zzd}|jd |j  t|j}|rd| }|jD ]N}| j|7  _|
  t|tjrnqB|jd|  |  t  qB|jj	dd|_|jd W nX tk
r } z8|jd|  t| t||_|jj	dd|_W 5 d}~X Y nX W 5 |
  X W 5 Q R X W 5 Q R X dS )u/   
        Отменяем импорт
        rl   rB   rC   rk   )rm   r   rm   r   rp   <   r   r   ;   u5   Не удалось остановить импортro   Nu#   ***** Отменяем импортr   u   Удаляем r   u,   Успешно отменили импортu   Ошибка удаления )rs   rt   ru   rP   r   rv   rB   r   rC   rD   r[   rw   rangeZgeventsleeprl   r   r   r)   Zimport_objectsr   r   rj   
isinstancerQ   r   r   r   r   r   )r*   r+   rq   r   r   r   steprd   r   r   r    cancel"  sV    





&zCmfImport.cancelc                 O   s    |  ddddg | jj| dS )uk   
        Предварительный просмотр получившихся объектов
        rL   rW   rN   rM   r   )rY   rM   preview)r   r*   r+   r   r   r    r   Y  s    zCmfImport.previewc                    s   t   dddddg S )NrL   rW   rN   rB   rM   )rE   save_preload_fieldsrG   rH   r   r    r   `  s    zCmfImport.save_preload_fieldsc                    s   |    t jf |S N)rU   rE   r[   r\   rH   r   r    r[   c  s    zCmfImport.savec                 C   s@   | j r
d S | jr,| jjdkr,tjjdd| _ tjjdd| _ d S )NZCmfPluginJirazimport.system.jira:defaultr   zimport.system:default)rC   rM   r   rP   r   rv   rG   r   r   r    _calc_workflowg  s
    zCmfImport._calc_workflowc                 C   sN   |  dg ttj| jj  d}|| jjjj	}|j
ddd |S )Nzplugin.source_hashZ	_raw_dataT)exist_okparents)rY   r   r   r(   r:   rM   Zui_namelowersource_hashrz   mkdir)r   r   r   r   r    r   o  s
    zCmfImport.get_download_pathc                 C   s   | j |  d S r   )rM   download_datarG   r   r   r    r   v  s    zCmfImport.download_datac                 C   s   | j |  d S r   )rM   download_filesrG   r   r   r    r   y  s    zCmfImport.download_files)r^   )NN))__name__
__module____qualname__r   r   Zui_meta_skipr   r   r   r   r-   listr7   r?   rF   rU   rZ   r]   staticmethodre   r   Z
celery_appZtaskr   r   rx   boolr   r   r   r   r   r   r   r[   r   r   r   r   r   __classcell__r   r   rH   r    r      sF   	

!N B!#5r   )Zcsvr   r   Zdecimalr   r   r   r   r   r   pathlibr   typingr	   r
   r   r   ZxlrdZopenpyxlZcmf.data_providers.baser   Zcmf.includeZcmfr   Zcommon.fieldsr   Zcmf.modules.logsr   r   r   r   r   r    <module>   s"   