U
    Y g                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlZd dlZd dlT d dlmZ d	Zd
Zee ZdZdZG dd dZdS )    N)Path)TemporaryDirectory)Dict)parse)datetime)*)RFilei  i  gHzG?100%c                   @   s  e Zd ZdZddddddgZeedd	d
ZedddZedddZ	eedddZ
eeedddZeeedddZedd ZedddZdkddZeddd d!Zd"d# Zddd$d%Zed&d' Zddd(d)Zeddd*d+Zeddd,d-Zeeeed.d/d0Zeedd1d2Zeed3d4d5Zeeedd6d7d8Zeeedd9d:d;Zeedd<d=d>Zeedd?d@dAZeeddBdCdDZ dddEdFZ!eeeeddGdHdIZ"dleeeedKdLdMZ#edmeeeeedNdOdPZ$eeedQdRdSZ%eddTdUZ&dVdW Z'dXdY Z(eddZd[Z)e*d\d] Z+ednedd^d_Z,dodadbZ-ee.dcddd`dedfdg Z/ee.dhddd`dedidj Z0dS )pCmfPluginConvertMixinuO   
    Используется в классах CmfDocument, CmfAttachment
    save_to_pdfsave_to_docxexport_to_docxexport_to_pdfexport_obj_list_to_pdfexport_tree_to_pdf)content	extensionc           	   	   C   s   ddl m} tj| jj}|| d| j  d| }t	j
| dd}t	j
|sft	| t	j
||}t|d}|| W 5 Q R X t	j
d|  d|}t|S )	up  
        Метод сохраняет файл директорию объекта и возвращает путь для скачивания
        :param content: Контент в байт
        :param extension: Расширение файла
        :return: file_path: Строка пути с кодированием символов для URL
        r   )cmfutil_.Texportwbfiles)Zcmf.utilr   modelsCmfDocumentZ_get_filenamenamevalueZtranslit_stripcodeospathjoinZget_files_direxistsmakedirsopenwriter   Zquote)	selfr   r   r   	file_nameZfull_export_pathZfile_full_pathfile	file_path r)   +./common/models/cmf_plugin_convert_mixin.py_save_export_file'   s     
z'CmfPluginConvertMixin._save_export_file)returnc                 O   sD   t d |  }| t|dddd}| |dd}| |d}|S )	u   
        Сохраняет HTML -> PDF
        Используется преобразование через DOCX, из-за этого нет разрыва страниц.

        :return: Данные pdf в виде строки base64
        !   УСТАРЕВШАЯ ВЕРСИЯutf-8encoding.htmldocx:MS Word 2007 XML.docxpdf:writer_pdf_Exportpdf	cmf_alert_handle_htmlobj_to_formatbytesr+   )r%   argskwargshtml
docx_bytes	pdf_bytesurl_file_pathr)   r)   r*   r   =   s    z!CmfPluginConvertMixin.save_to_pdfc                 O   s6   t d |  }| t|dddd}| |d}|S )uu   
        Сохраняет HTML -> DOCX

        :return: Данные docx в виде строки base64
        r-   r.   r/   r1   r2   docxr6   )r%   r;   r<   r=   r>   r@   r)   r)   r*   r   M   s
    z"CmfPluginConvertMixin.save_to_docx)	html_textr,   c           	   
   C   s~   ddl m} ddlm} | }| }t J}| ||}||| t }|	| |
d | W  5 Q R  S Q R X dS )ud   
        Преобразует HTML -> DOCX

        :return: Данные в байтах
        r   )Document)
HtmlToDocxN)rA   rC   ZhtmldocxrD   r   _handle_view_htmlZadd_html_to_documentioBytesIOsaveseekgetvalue)	r%   rB   rC   rD   ZdocumentZ
new_parsertmp_dir_namer=   	docx_filer)   r)   r*   html_to_docx[   s    

z"CmfPluginConvertMixin.html_to_docx)r   r,   c                 C   s   t |  S )up   
        Преобразует байты в строку base64

        :return: Строка base64
        )base64Z	b64encodedecode)r   r)   r)   r*   _get_base64o   s    z!CmfPluginConvertMixin._get_base64)r   r,   c                 C   s   |  dr(td t|  dd }n\|  drPt|  ddt t }n4|  drtt|  ddt }ntdt t }|S )u?  
        Метод масштабирует значение ширины под формат А4 и переводит его в пиксели
        :param value: Строка значения ширины в % или пикселях
        :return: pixels: Число с плавающей точкой
        %d    ptpx)endswithA4_PAGE_WIDTH_PXfloatstripreplacePT_TO_PXSCALE_A4)r   Zpixelsr)   r)   r*   _get_value_in_pixelsx   s    


z*CmfPluginConvertMixin._get_value_in_pixelsc                 C   s   | j dr| j dsdS t| j d dd}t| j d dd}||kr|| }|d }| d| j d< ||  d| j d< dS )u   
        Метод масштабирует изображение под ширину ячейки таблицы
        :param img: Тег изображения
        :param cell_width: Ширина ячейки
        widthheightNrU   rS   g)\(@)attrsgetrX   rY   rZ   )imgZ
cell_widthZ	img_widthZ
img_heightproportion_ratior)   r)   r*   _auto_size_img_for_cell   s    z-CmfPluginConvertMixin._auto_size_img_for_cellc                 C   s  |   }|jddD ]}||d q|dD ]"}t|jdk r4||d q4| | | | | 	| | 
| | | | | | | |jddd}|jd	d
d}d|_|d|d |j|jddd |j| ||d |j| t|S )u   
        Обработка страницы html для сохранения

        :param html_text: Страница html
        :return: Обработанная страница html
        spoiler-iconclass_p   brr=   zru-RU)langstyletext/csstypea  
        @page { size: 21.0cm 29.7cm; margin: 1cm }
        body {
            font-family: Arial, Helvetica, FreeSans, 'Liberation Sans', sans-serif;
            font-size: 10pt;
            line-height: 1.2;
        }
        td { page-break-inside: avoid !important; }
        tr { page-break-after: avoid !important; }
        td p { margin-top: 0.2cm; margin-bottom: 0.2cm }
        r   headmetazUTF-8)charsetbody)handle_include_pagefind_allreplace_withnew_taglencontentsappend_handle_html_card_preview_handle_html_svg_to_png_handle_html_img_handle_html_infopanel_handle_html_layout_handle_html_table_handle_html_cleanstringinsertrp   rs   str)r%   soupspanrh   r=   rl   r)   r)   r*   r8      s,    






z"CmfPluginConvertMixin._handle_htmlNc           
      C   s   ddl m} |dkrt }| dg | jjr6| jjnd}||d}|}|| jj |jddD ]V}|j	
d	}|r||krtjj
|d
}|r||}	||	 q|  qb|  qb|S )u  
        Подгружает html text из Include страницы
        :param parent_pages: ID родителей страницы для проверки на зацикливание
        :return: bs4.BeautifulSoup soup: Объект библиотеки
        r   BeautifulSoupNtextrS   html.parserzmacros-include-pagerf   zdata-param-pageid)id)bs4r   setload_fieldsr   r   addr   ru   r`   ra   r   r   rt   rv   	decompose)
r%   Zparent_pagesr   rB   r   ZpagesZinclude_pageZpage_idZinclude_documentZinclude_htmlr)   r)   r*   rt      s$    



z)CmfPluginConvertMixin.handle_include_pagec                 C   s   | j ddD ]z}|d}d|jd< |jdd| d}|jdd| d}| d}|| || || || qd	S )
u  
        Метод парсит DIV вложения и оставляет в нем только нужные теги: изображение, имя файла и дата

        :param bs4.BeautifulSoup soup: Объект библиотеки
        zapp-tinymce-card-previewrf   rb   Z150pxr^   zapp-tinymce-file-name-viewdivzapp-tinymce-file-create-viewN)ru   findr`   wraprw   rz   rv   )r   Zpreviewrb   r&   Zfile_creater   r)   r)   r*   r{      s    





z/CmfPluginConvertMixin._handle_html_card_previewc              	   C   s  ddl }ddlm} |dD ]}zhd}|jdd }rz|jd}dd	d
| jjgdd
| dgg}t	j
j|dgd}|r|jd|jjd}	n|jd}
|jd}|jd}|r| }|d |jd< |d |jd< d}|jdt|idD ](}|jd}d}t||||jd< q|jt|d}| |}|
rH|sb|t|}	|	j\}
}|jdd| t|
t|d}	||	 W q tk
r   dt| }t| Y qX qdS )ud  
        Метод заменяет тег SVG тегом IMG с изображением в PNG.
        Конвертирует SVG в PNG, если SVG в диаграммах DrawIO, то сначала пытается найти готовый PNG диаграммы
        :param bs4.BeautifulSoup soup: Объект библиотеки
        r   NImagesvgzdrawio-contentrf   zdata-drawio-titleZANDZ	parent_id==r   z.pngurlfilterfieldsrb   )srcr^   r_   viewbox      z"rgb\((\d+)[\s,]+(\d+)[\s,]+(\d+)\)fillr`   zrgb(\1, \2, \3))Z
bytestringzdata:image/png;base64,)r   r^   r_   u3   Не удалось заменить SVG на PNG: )cairosvgPILr   ru   find_parentr`   ra   r   r   r   CmfAttachmentrw   r   splitrecompilesubZsvg2pngr   rP   r#   rF   rG   sizerv   	Exceptionlogging	exception)r%   r   r   r   r   
attachmentZdrawioZdrawio_titler   rb   r^   r_   r   Zsvg_sizepatternZtag_with_fillZrgbreplZimg_byteZ
img_base64err_msgr)   r)   r*   r|     sH     

 z-CmfPluginConvertMixin._handle_html_svg_to_pngc           "      C   s  ddl m} ddl}| h}|dD ]T}|jd}|jd}|rtdd|	d	d
}|
drtd t|	dd }nt|}|rtdd|	d	d
}|
drtd t|	dd }nt|}|jd}|sq*|dtj s|dtj dr:|	dtj d}|	dtj dd}|dsR|dr|jd}	|	dd}t|}
t|
j}|	rtjj|	dgd}n*ddd|gdd|
jgg}tjj|dgd}|rt|j}nt|	dtj}| s ttjd }z\||}|j\}}t|d4}| | }|j  }d| d| |jd< W 5 Q R X W n8 t!k
r   |"|j# t$%d | d! Y q*Y nX |d"rzt&j|d#d$}|j'd%kst(d&|j' ||)d
d' d }|d(krd)}| |j*}d| d| |jd< t| d*t+, d+d
| }t|d,}|-|j* W 5 Q R X ||}|j\}}W nh t!k
r } zH|.d-}d.d/i|_d0|_/|"| t$%d1 t0d2|  W Y q*W 5 d}~X Y nX |d3rz6td4d|}t12|}|t34|}|j\}}W nh t!k
r~ } zH|.d-}d.d/i|_d0|_/|"| t$%d5 t0d6|  W Y q*W 5 d}~X Y nX |r|s|| }|| }|r|s|| }|| }|s|s|}|}|| }|t5 }|| }|t6ks|dkr|rt6| nd'}t6}|t7|9 }| d7|jd< | d7|jd< |d.r*d8|d. kr*|j.d9d:d;} |j.d<|j}!| 8|! |"|  q*W 5 Q R X dS )=u#  
        Метод заменяет адрес картинки на base64 и масштабирует размер изображения под формат А4 относительно исходного

        :param bs4.BeautifulSoup soup: Объект библиотеки
        r   r   Nrb   r^   r_   z[^0-9|%|\.]rS   ,r   rQ   rR   r   zhttps://zhttp://z:8080/z	files/objzdata-attach-idz%25r   )r   r   ORr   r   r   /files/common/static/file-outline.pngrbdata:image/z;base64,u	   Файл u/    не является изображениемhttp)ri      )timeout   u2   Код статуса ответа сервера ri   )ZpngZbmpZgifZtiffZtifZjpegZjpgz/imgz%Y%m%d-%H%M%Szwb+rh   rl   z0color: red; font-style: italic; font-size: 10pt;u?   Не удалось отобразить изображениеuL   Не удалось скачать изображения документаuN   Не удалось скачать изображения документа: 
data:imagez^data:image/.+;base64,uG   Не удалось конвертировать изображениеuI   Не удалось конвертировать изображение: rU   zmargin-left: auto;r   z text-align: center; width: 100%;rl   )rb   )9r   r   tempfiler   ru   r`   ra   r   r   rZ   rV   WEB_PAGE_WIDTH_PXrX   
startswithconfigAPP_FQDNr   Zurlparseunquoter   r   r   r   full_path_file
UPLOAD_DIRr!   PROJECT_DIRr#   r   rP   readformatlowerr   rv   r   r   r   requestsstatus_codeAssertionErrorrfindr   r   Znowr$   rw   r   r7   rN   	b64decoderF   rG   r\   rW   absrz   )"r%   r   r   r   tmp_dirZimg_tagr^   r_   r   Z	attach_idr   Zurl_path_unquoter   Z_filterZattach_pathrb   Zwidth_originalZheight_originalZattachment_byteZattach_base64Z	mime_typeresZextension_imgr(   fpeZtag_iZimg_base64_dataZimg_byte_dataZscalerc   coefficientZimg_wrapperZ	clone_imgr)   r)   r*   r}   2  s    

*




$











 
z&CmfPluginConvertMixin._handle_html_imgc                 C   s  | j ddD ]t}| jddddd}|jd}|dkrDd	|jd
< nb|dkrXd|jd
< nN|dkrld|jd
< n:|dkrd|jd
< n&|dkrd|jd
< n|dkrd|jd
< || jddd |j| jdddd |jdd}|dk	r|jrd|jjd< d|jjd< |jj	|jd g |jd!d}| jddd"d}| jd#d$d%d&}|	d'd( |j
D  |	|g |j	|g || qdS ))u   
        Подменяет инфопанели. Создает панели в виде table вместо div
        :param bs4.BeautifulSoup soup: Объект библиотеки
        :return:
        zak-editor-panelrf   tabler	   0)r^   cellpaddingcellspacingzdata-panel-typeZnotez/background: #eae6ff; border: none; padding: 0cmrl   infoz/background: #deebff; border: none; padding: 0cmsuccessz/background: #e3fcef; border: none; padding: 0cmalarmz/background: #feece5; border: none; padding: 0cmerrorZwarningz/background: #fffae6; border: none; padding: 0cmtrzbackground: transparentr   tdtopz1width: 36px; orphans: 0; widows: 0; margin: 0.2cm)Zvalignrl   zak-editor-panel__iconNZ	absmiddleZalignborder zak-editor-panel__contentz$orphans: 0; widows: 0; margin: 0.2cmZfont3zfont-size: 12pt)r   rl   c                 S   s   g | ]}|d kr|qS )
r)   ).0r   r)   r)   r*   
<listcomp>  s      z@CmfPluginConvertMixin._handle_html_infopanel.<locals>.<listcomp>)ru   rw   r`   ra   rz   r   r   rb   r   extendry   rv   )r   Zpanelr   Z
panel_typeZiconZpanel_contentZtd_tagZtag_fontr)   r)   r*   r~     s>    
z,CmfPluginConvertMixin._handle_html_infopanelc                  C   s  dd l }dddddddd	d
ddddddddddd}|dD ]n}i }|jddd}|rd|| |dddD ]&}|rv|d\}}	|	 || < qv|dr|ddtkr|dt}
|dttkr|
|d< n
t |d< d|d< |dsd |d< |d!sd |d!< |d"s,d#|d"< |d$sH|d$d%|d$< i }|j	d k	rfg }| 
|d}
t|j	d&}t|j	jd'd(D ]\}}i }t||jjsq|jdddD ]*}|r|d\}}	|	 || < q|d }	r| 
|	}|||< |
|8 }
nd ||< || qt|dkr\|
t| }|D ]}|||< qL|j	  tt| }|d)D ]*}d'}d}| 
|d}
|d*d+gD ]}i }d,|d"< |d-r|jd-}|||d.< |dddD ]*}|r|d\}}	|	 || < q|tkr2d/}n|s`|dpL|d }r| 
|}np||}|d0}|r|rt|d' t|| D ]"}||kr q||| 7 }q|t|7 }n|d'7 }|rF| }d1}|D ](}|jrt|jd2kr|} qq| d3|d< |
|8 }
|d4D ]}| || q0n|d'7 }q|dkr|
| }|jd*dd id5D ]2}| d3|d< |d4D ]}| || qqvqq>d S )6Nr   z#e3fcefz#e6fcffz#b3d4ffz#fffae6z#feece5z#eae6ffz#f4f5f7z#b3f5ffz#abf5d1z#ffc400z#fff0b3z#ffbdadz#c0b6f2z#b3bac5z#4c9affz#79e2f2z#57d9a3z#ff8f73z#998dd9)zLight greenz
Light tealZBluezLight yellowz	Light redzLight purplez
Light grayZTealZGreenzDark yellowZYellowZRedZPurpleZGrayz	Dark Bluez	Dark tealz
Dark greenzDark redzDark purpler   r   ztableView-content-wraprf   rl   rS   ;:r^   r   r   r   Zbordercolorz#eeeeeer   1colri   )startr   r   thZnonezdata-highlight-colorZbgcolor   colspan
   z[<br/>]rU   rb   r   )r   ru   r   rv   ra   r   rY   FULL_SCREEN_VALUErW   Zcolgroupr]   rx   	enumeratechildren
isinstanceelementTagr`   rz   r   roundsumvaluesZhas_attrrangeintry   r   rd   ) r%   r   r   Zcell_colorsr   Zstyle_table_dictparentparamsr   r   Ztable_widthZ
cols_widthZcol_without_widthZ	col_countZcol_numr   Zstyle_col_dictZwidth_pxcZcols_sumr   	nums_colsZ
table_cellZtd_styleZtable_cell_color_nameZtd_widthr   iZtable_cell_childrenZtable_cell_widthZtable_cell_childrb   r)   r)   r*   r     s    











z(CmfPluginConvertMixin._handle_html_tablec                    s   ddl m  | dddiD ]}d|d< d|_| d	} fd
d|jD }t|}td| d}|D ]*}d|_|d| d|d< |	| qf|	| qdS )u   
        Функция добавляет стиль для макета, используем table, так как flex не поддерживается в LibreOffice

        :param bs4.BeautifulSoup soup: Объект библиотеки
        r   r   r   zdata-layout-sectiontruer	   r^   r   r   c                    s   g | ]}t | r|qS r)   )r   r   elr  r)   r*   r     s     
 z=CmfPluginConvertMixin._handle_html_layout.<locals>.<listcomp>rR   r   r   zdata-column-widthrQ   N)
r   r   ru   r   rw   r   rx   r   ra   rz   )r   Zlayoutr   columnsr   Zwidth_columncolumnr)   r  r*   r   r  s    
z)CmfPluginConvertMixin._handle_html_layoutc                 C   s"   | j ddgdD ]}|  qdS )u   
        Метод очищает HTML от лишних тегов

        :param bs4.BeautifulSoup soup: Объект библиотеки
        ztable-context-menuzcell-dropbox-buttonrf   N)ru   r   )r   tagr)   r)   r*   r     s    z(CmfPluginConvertMixin._handle_html_clean)rB   tmp_pathr,   c           !   
   C   s  ddl m}m} ddlm} || }g }|dD ]>}d|jddkrZ|t	|dd	 |jd
s2d|jd
< q2|dD ]}|t	|dd	 q||dD ]}|
  q|dD ]}|
  qt|D ]\}}	|	d }|jdd}
|jdd}|jd}|dr|d\}}|dd}|dd}t|d| d|  }t|d$}|t| t||jd< W 5 Q R X |dr
t|}tjjddtjgdgd}|rt|j}nt|dtj}| r
t||jd< |
s
||}|j\}
}d}tdd d! t|
D }|d"rBttj d# }|!d$d%kr\t"|n|!d$}||!d| }t#|}|j$d&kr4t|}|d'| |  % }t|d}||j& W 5 Q R X z>||}|j\}
}t||jd< |r|n|
r|
n|}W n$ |k
r0   t||jd< Y nX nt||jd< ||krP|n|}|r|| rj|| nd(}||jd< tdd)d! t|D } t| | |jd< qt|S )*u   
        Функция заменяет адрес картинки, файла на фактический

        :return: HTML с фактическими адресами картинок, файлов
        r   )r   UnidentifiedImageErrorr   azcmf-fileclassrS   r'   )objro   Zhrefrb   Zimagerl   titler  r^   r_   r   r   r   r   z;base64Zfile_r   r   r   r   r   r   r   i  c                 s   s   | ]}|  r|V  qd S Nisdigitr  r)   r)   r*   	<genexpr>  s      z:CmfPluginConvertMixin._handle_view_html.<locals>.<genexpr>r   r   ?r   Zname_ri   c                 s   s   | ]}|  r|V  qd S r  r  r  r)   r)   r*   r    s      )'r   r   r	  r   r   ru   r`   ra   rz   dictr   r   r   r   rZ   r   r#   r$   rN   r   r   r   r   r   r   r   r   r   r!   r   r   r    r   r   rx   r   r   Zresolver   )!rB   r  r   r	  r   r   r   r  indexZ_filer^   r_   Zfull_urlZtype_imgZb64dataZ	name_fileZfile_imgZurl_pathr   Zattachment_pathrb   Zwidth_defaultZwidth_valueZpath_defaultZend_nameZfile_extensionZresponser   filepathr'   Z	width_newr   Zheight_valuer)   r)   r*   rE     s    





"



z'CmfPluginConvertMixin._handle_view_htmlc              
   C   s  ddl m}m} ddlm} tjd||fd}tj	tj
tjtdd}tj
|ddd	d
dd}| }d|_|  |dd| |d ||}|dD ]}	d|	jd< q|dD ]}	d|	jd< d|	jd< qt 0}
|| | |
 t| W  5 Q R  S Q R X dS )uw   
        Конвертирует HTML -> PDF

        :return: Возвращает байты PDF файла
        r   )FPDF	HTMLMixinr   	_HTML2PDF)basesz..modulesZappZassetsZfontsz	noto-sanszNotoSans-Regular.ttfr.   ZNotoSansrS   r   r	   r^   r   Z100r   r   N)Zfpdfr  r  r   r   types	new_classr   r   abspathr    dirname__file__Zcore_fonts_encodingZadd_pageZadd_fontZset_fontZfindAllr`   r   Z
write_htmlrE   Zprettifyr:   output)r%   rB   r  r  r   r  Z	path_fontr5   r   r  rK   r)   r)   r*   html_to_pdf  s&     

z!CmfPluginConvertMixin.html_to_pdf)	path_docxr,   c              
   C   sH   ddl m} tj|rDt|d}||}|jW  5 Q R  S Q R X dS )u7   
        Конвертирует DOCX -> HTML
        r   )convert_to_htmlr   N)Zmammothr$  r   r   r!   r#   r   )r%   r#  r$  rL   resultr)   r)   r*   docx_to_html  s
    z"CmfPluginConvertMixin.docx_to_html)path_imgr?   r,   c                 C   s   t | | dS u5   
        Конвертирует PDF -> PNG
        N)r   pdf_bytes_to_imgr'  r?   r)   r)   r*   r)    s    z&CmfPluginConvertMixin.pdf_bytes_to_img)path_pdfr'  r,   c                 C   s   t | | dS r(  )r   
pdf_to_img)r+  r'  r)   r)   r*   r,     s    z CmfPluginConvertMixin.pdf_to_img)	path_htmlr'  r,   c              	   C   sB   t j|r>t|d"}| | }| j||d W 5 Q R X dS )u6   
        Конвертирует HTML -> PNG
        rr*  N)r   r   r!   r#   r"  r   r)  )r%   r-  r'  	file_htmlr?   r)   r)   r*   html_to_img'  s    z!CmfPluginConvertMixin.html_to_img)	path_xlsxr-  r,   c                 C   s   ddl m } ||| dS )u7   
        Конвертирует XLSX -> HTML
        r   	xlsx2htmlNr2  )r%   r1  r-  r3  r)   r)   r*   xlsx_to_html0  s    z"CmfPluginConvertMixin.xlsx_to_html)r'  path_pngr,   c                 C   s$   ddl m} ||}|| dS )u<   
        Сохранение в формате PNG
        r   r   N)r   r   r#   rH   )r%   r'  r5  r   rb   r)   r)   r*   save_img8  s    
zCmfPluginConvertMixin.save_imgc              
      s  ddl m} ddlm} ddlm  ddlm ||ddd}| }| }tt	|D ]}|
|| }	|dkr|j}
|| |
_n|j|| d}
|	jD ]*}|\}}}}|
j|d	 ||d	 |d
 q fddt|	jD ],}|
fdd|	j||	|dD  qqZ|| dS )u6   
        Конвертирует XLS -> XLSX
        r   )workbook)open_workbook)XL_CELL_DATE)xldate_as_tupleT)Zformatting_infoZragged_rows)r  ri   )Z	start_rowZend_rowZstart_columnZ
end_columnc                    sJ   | j }| j krF|d}|dd dkr>tj|dd   }nt| }|S )Nr   r   )r   r   r   )r   Zctyper   time)cellr   Zdatetime_tup)r9  r:  r)   r*   _get_xlrd_cell_value[  s    

z?CmfPluginConvertMixin.xls_to_xlsx.<locals>._get_xlrd_cell_valuec                 3   s   | ]} |V  qd S r  r)   )r   r<  )r=  r)   r*   r  f  s   z4CmfPluginConvertMixin.xls_to_xlsx.<locals>.<genexpr>)Zend_colxN)Zopenpyxlr7  Zxlrdr8  r9  r:  ZWorkbooksheet_namesr   rx   Zsheet_by_nameZactiver  Zcreate_sheetZmerged_cellsZmerge_cellsZnrowsrz   Z	row_sliceZrow_lenrH   )r%   Zpath_xlsr1  r7  r8  Zbook_xlsZ	book_xlsxr>  Zsheet_indexZ	sheet_xlsZ
sheet_xlsxZcrangeZrloZrhiZcloZchirowr)   )r9  r=  r:  r*   xls_to_xlsxA  s2    
 
z!CmfPluginConvertMixin.xls_to_xlsx)dataoutput_pathsuffixr,   c                 C   s   t | || dS )u(  
        Конвертация данных в PDF через libreoffice
        :param data: Данные файла в виде байт
        :param output_path: Путь сохранения итогового pdf файла
        :param suffix: Расширение файла
        N)r   
obj_to_pdf)rA  rB  rC  r)   r)   r*   rD  l  s    z CmfPluginConvertMixin.obj_to_pdfrS   )rA  rC  mimetyper,   c              
   C   sZ  ddl }ddlm} | ||d|}| @}t|d }t|d}	|	| | t	|}
W 5 Q R X W 5 Q R X |j
|
ddd	d
}|dd}||}|d}|rR|d}|rR| }|jddD ]}|dd}d}|d D ]T}|d| }|||d d d }|||d d }|||| 7 }q||  |d< qt	|S )u  
        Конвертация данных в HTML через libreoffice

        :param data: Данные файла источника в виде байт
        :param suffix: Расширение файла источника

        :return: Строка HTML
        r   Nr   r=   z	file.htmlr   ignore)errors u   �u   rS   rp   rl   Trf   r  r   {ri   })r   r   r   r9   r   r   r#   r$   get_encoding_typer   rO   rZ   r   Zget_textru   ra   )r%   rA  rC  rE  r   r   Z
bytes_htmlr  Zfull_path_htmlr/  r0   rB   r   rp   rl   Z
style_textr  	style_tagZstyle_classZ	class_tagZstart_pos_classZstyle_startZ	style_endr)   r)   r*   obj_to_htmlv  s4    	

"

z!CmfPluginConvertMixin.obj_to_html)rA  rC  	to_formatrE  r,   c              
   C   sH   zt | |||}W n. tk
rB } zt|  W 5 d}~X Y nX |S )u*  
        Конвертация данных в HTML через libreoffice
        :param data: Данные файла в виде байт
        :param suffix: Расширение файла
        :param to_format: Формат файла который хотим получить
        N)r   r9   r   gdebug)rA  rC  rN  rE  Zconverted_dataerrr)   r)   r*   r9     s    
z#CmfPluginConvertMixin.obj_to_format)	path_filer,   c              
   C   s   ddl m} | }t| d8}|D ]$}|| |jr"|jdr" qHq"|  W 5 Q R X |jd}|sddddd	d
dg}|D ]d}zFtj| d|d,}|	  |
d |}W 5 Q R  W  qW 5 Q R X W q tk
r   d}Y qX q|S )u   
        Определяем кодировку файла

        :param path_file: Путь к файлу
        :return: Кодировка файла
        r   )UniversalDetectorr   r0   r.   zwindows-1251zwindows-1250zwindows-1252asciiz
iso-8859-1latin1r.  r/   )Zchardet.universaldetectorrS  r#   ZfeedZdoner%  ra   closerF   	readlinesrI   UnicodeDecodeError)rR  rS  Zdetectorr'   liner0   	encodingsZformat_encodingr)   r)   r*   rK    s*    

z'CmfPluginConvertMixin.get_encoding_typec                 C   s   ddl m} | dg || jjp$d}|d}|jddd}d	|_|jd| |jj	
| |jd
dD ]}||d qn|dD ]"}t|jdk r|
|d q| | | | | | | | | | | | | | t|S )Nr   r   text_render<p></p>rp   rl   rm   rn   zg@page { size: 21.0cm 29.7cm } td,th { padding: 0.45cm; } td p,th p { margin: 0;word-break: break-all; }re   rf   rh   ri   rj   )r   r   r   r[  r   rw   r   r=   r   rp   rz   ru   rv   rx   ry   r{   r|   r}   r~   r   r   r   r   )r%   r   r   head_tagrL  r   rh   r)   r)   r*   prepare_document_to_docx  s*    







z.CmfPluginConvertMixin.prepare_document_to_docxc                 C   s<   |   }| t|dddd}| |dd}| |d}|S )Nr.   r/   r1   r2   r3   r4   r5   r^  r9   r:   r+   )r%   prepared_htmlr>   r?   r@   r)   r)   r*   r     s
    z#CmfPluginConvertMixin.export_to_pdfc                 C   s.   |   }| t|dddd}| |d}|S )Nr.   r/   r1   r2   rA   r_  )r%   r`  r>   r@   r)   r)   r*   r     s    z$CmfPluginConvertMixin.export_to_docxc                 C   sz   |  ddg ddlm} || jjp&dd}|d}| jj|_|d| | 	| |jdd	d
id}|
| t|S )Nr[  Zprojectr   r   r\  r   Zh1r   rl   zpage-break-after: always;r   )r   r   r   r[  r   rw   r   r   r   r}   rz   r   )r%   r   r   Zname_tagZ	break_tagr)   r)   r*   prepare_document_to_export  s    



z0CmfPluginConvertMixin.prepare_document_to_exportc           
      C   st   dg}|D ]d}t |}dd | D }|D ]@}t|d,}| }|d}	||	_|jj|	 W 5 Q R X q,q
|S )Nz&/opt/eva-app/dist/cmf-angular/tinymce/c                 S   s"   g | ]}t |d rt |qS )z.css)r   rV   )r   css_filer)   r)   r*   r     s      z:CmfPluginConvertMixin._append_css_tags.<locals>.<listcomp>r.  rl   )	r   Ziterdirr#   r   rw   r   r=   rp   rz   )
clsr   ZfoldersZfolder_nameZfolderZcss_listrb  r'   ZcssZcss_tagr)   r)   r*   _append_css_tags  s    
z&CmfPluginConvertMixin._append_css_tagsc                 C   s  ddl m} dd l}t  tjd tjj	  W 5 Q R X || pHdd}|
d}|jd| |j
ddd	id
}|jj| tj|}d|jjjd< t }|sd}| d}| d| d}	t|d}
|
t| W 5 Q R X |jdddddd|	 | gdd}tjtj|d}t|	d}|j|ddd |  W 5 Q R X t  tj| W 5 Q R X tjjtjd|ddt j! d|j"j# d d! W 5 Q R X d S )"Nr   r   i,  r\  Zlxmlrp   rq   rr   r.   r   z%mce-content-body mce-content-readonlyr  r   z/export.htmlr   z.pdfwZchromiumz
--headlessz--disable-gpuz--no-pdf-header-footerz--no-sandboxz--print-to-pdf=T)check)r   r   r   F)ZbackupZmake_previewu&   Закончен экспорт в PDFu   
                Закончен экспорт в PDF
                Ссылка для скачивания:
                https://z
                )Zpersonr   r  Zforce_notify_current_personmsg)$r   r   
subprocessr   Zdisable_aclr   ZCmfDeferredJobZset_estimate_durationZdpZcommitrw   r=   r   rp   rz   r   rd  rs   r`   r   r#   r$   r   runr   rO  Zcurrent_userZupload_stream_filerH   Zset_result_attachmentZ	CmfNotifyZplace_notifyr   r   r   r   )r=   r&   r   rh  r   r]  Zcharset_tagr   Zhtml_file_nameZpdf_file_namer'   procZ
new_attachfr)   r)   r*   _export_html_to_pdf"  s^    




z)CmfPluginConvertMixin._export_html_to_pdfFc              	   C   s   d}|d krt  }|r(| g| j  }n
| j }t|D ]\}}|jdsZ|jdr:|jdrhq:||kr:|| nq:|jdrz|  || 7 }W n t	k
r   Y nX ||
 7 }q:|S )NrS   CmfDocument:z
CmfFolder:zCmfProject:)r   Z
tree_nodesloadr   r   r   r   _acl_check_readra  CmfPermissionErrorcollect_html_nested)r%   export_selfZ	processedZ	html_bodyZnodesr   Znoder)   r)   r*   rq  c  s,    

z)CmfPluginConvertMixin.collect_html_nestedu   Экспорт PDF-файлаT)descriptionZshow_bg_progressbarZ
system_jobc                 C   s,   t | }|jdd}tjj||jd dS )u\   
        Выгружает объект и всех его потомков в pdf
        T)rr  )r&   N)r   get_obj_by_idrq  r   r   rl  r   )obj_idr  Zhtml_nestedr)   r)   r*   r     s    
z(CmfPluginConvertMixin.export_tree_to_pdfu   Экспорт в pdfc              	   C   sf   d}| D ]L}| dsqt|}|rz|  || 7 }W q tk
rR   Y qX qtj| dS )u   
        Выгружает все объекты из списка в pdf
        obj_list -- список id выгружаемых объектов
        rS   rm  N)	r   r   rt  ro  ra  rp  r   r   rl  )Zobj_listr=   ru  r  r)   r)   r*   r     s    

z,CmfPluginConvertMixin.export_obj_list_to_pdf)N)rS   )rS   )N)FN)1__name__
__module____qualname____doc__Zapi_methodsr:   r   r+   r   r   rM   staticmethodrP   rX   r]   rd   r8   rt   r{   r|   r}   r~   r   r   r   rE   r"  r&  r)  r,  r0  r4  r6  r@  rD  rM  r9   rK  r^  r   r   ra  classmethodrd  rl  rq  Zcmf_deferred_jobr   r   r)   r)   r)   r*   r
      s   	
2
!/ 
* 	[		+	(!#
@
!	r
   )rN   ZhashlibrF   pathlibr   r   r   typingr   Zurllibr   r   r  r   Zcmf.includeZrdiskr   r   rW   r\   r[   r   r
   r)   r)   r)   r*   <module>   s"   