U
[Se0 @ s d dl Z d dlZd dlmZmZ d dlZd dlmZ d dlT d dlm Z d dl
m Z d dl m
Z
d dlmZ G d d
d
eje
ZdS ) N)datetimetimezone)Path)*)CmfModel) cmf_hashlib)CmfPluginConvertMixin)cmf_attachmentc s" e Zd Zejjjd Zdd Zdd Z fddZ dd Z
d
d Z d d
Z fddZ
edd ZddddZeedddZddded d d Zd d Zd6d!d"Zeedd#d$Zeedd%d&Zd'd( Zdd) fd*d+
Z fd,d-Z d7d.d/Z d0d1 Z fd2d3Z d4d5 Z Z!S )8
CmfAttachment) file_existsc C s: | j js d S tj| j jd }d}t||kr6|| _d S )N )name
is_changedospathsplitextvaluelenZ file_type)self extensionZext_max_length r (./modules/files/models/cmf_attachment.py_calc_file_type s z CmfAttachment._calc_file_typec C s | j js d S td| j j}d}t| j jt| t|d }d}t| j ddkrh| j dd }||krt|d }| j jd |dt| d t
| j jd
| _ |r| j d| | _ d S ) Nu [А-Яа-я] .r - ) r r refindallr r splitmathZfloorstripr Z
short_str_enc)r Z
rus_simbolZ
os_max_lengthZfile_name_lengthextZ slice_lengthr r r
_calc_name s 8zCmfAttachment._calc_namec s t ddddg S )Nurl url_previewurl_preview_imgparent)supersave_preload_fieldsr __class__r r r. * s z!CmfAttachment.save_preload_fieldsc C sD | j | j sd S tjtj| j }tj|s@t | |S N)
r, loadr r joinconfig
UPLOAD_DIR
get_files_direxistsmakedirs)r
upload_dirr r r r: - s
zCmfAttachment.upload_dirc C s4 t jjj }|| j}|jr*|j| _nd| _d S )Nr ) modelsCmfRFiledp data_driverget_rd get_rfile
_file_namer8 st_sizer rdiskrfiler r r
_calc_st_size6 s
zCmfAttachment._calc_st_sizec C s4 t jjj }|| j}|jr0t j | dd d S )NZ
attachmentZ
after_save)
r; r< r= r> r? r@ rA r8 ZCmfAutomationCrudTriggerZ crud_hookrC r r r _call_automation_hook> s z#CmfAttachment._call_automation_hookc s: | | | | t j||}| |S r2 )r( _calc_urlr rF r- saverG )r argskwargsresr0 r r rI E s zCmfAttachment.savec C s | | j S r2 )r: r r/ r r r file_pathN s zCmfAttachment.file_pathNfieldsc O sb | j js| jjsd S tjdtj | j
| _tj| j dd| _ tj| j dd| _ d S )Nz/filesz.metaz preview.pdfzpreview_img.png)
r r r) Zis_nullr r r4 urllibparseZquoterA r* r+ )r rO rJ rK r r r rH S s
zCmfAttachment._calc_url)returnc C s t ttj| j S )u< Полный фактический путь к файлу )strr r5 r6 rA r/ r r r full_path_fileZ s z CmfAttachment.full_path_fileTFbackup smart_backup)contentc O s> t jjj }|| j}|jdddd |j|||d dS )u Загрузка файла FT)is_direxist_okparentsrU N) r; r< r= r> r? r@ rA createwrite)r rX rV rW rJ rK rD rE r r r upload_file_ s zCmfAttachment.upload_filec O s | j dd}|j|| d S )NT)r\ )r@ Z stream_writer rJ rK rE r r r upload_stream_filef s z CmfAttachment.upload_stream_filec k sR | }||}|r |d }n| j dgdjj}t|d }|V W 5 Q R X d S )NabspathrN rb)r@ Z get_versionra r open)r versionrK rE Z
rfile_versionra filer r r get_contentj s
zCmfAttachment.get_contentc C s t jjj }|| j}|jS r2 )r; r< r= r> r? r@ rA r8 rC r r r r t s zCmfAttachment.file_existsc C s8 | j | j r(tj| j | jjS tddd dS )uT Имя файла вложения в зависимости от родителя u Вложение должно быть привязано к объекту, нельзя сохранять в корень диска.T)abortN) r, r3 r r r4 r7 r r Z cmf_alertr/ r r r rA z s
zCmfAttachment._file_namec O s | }|j||S r2 )r@ get_fullviewr_ r r r rh s zCmfAttachment.get_fullview)forcec s | dddg | jr| jjdrt| j }rtj| jkr|z|j d| j
d W q t k
rx |j d| j
d Y qX n|j d| j
d | j r|
}|r|j| jjd t j|d |i|S )
Nroot_parent_idr, cmf_owner CmfProject:zPPP-ATT-DELETE-ALLobjzPPP-ATT-DELETE-OWN)prefixri ) load_fieldsrj r
startswithcmfutil
get_obj_by_idgZ current_userrk check_project_role_accessr, ZCmfPermissionErrorrA r@ Z
move_to_trashcoder- delete)r ri rJ rK projectrE r0 r r rw s zCmfAttachment.deletec s6 t jf |}tjj| jj d| jj | jd |S )N_)r Z
target_dir) r- restorer; r< Zrestore_from_trashrv r r rA )r rK rL r0 r r rz s $zCmfAttachment.restorec K sd dd l }tjj| j|d}|s`|r`tj| jdd}| j|_tjj|j|jj dd|_
| |S )Nr )idrO F)r rY T)r{ mkdir) r r; r< getrA r r{ r dirnamer r, rI )r rO r\ _kwargsr rL r r r r@ s zCmfAttachment.get_rfilec C s8 | j r4| j jdr4t| j }r4|jd| jd d S )Nrl zPPP-ATT-CREATErm )rj r rq rr rs ru r, )r rx r r r _check_project_perm s z!CmfAttachment._check_project_permc s* | | | t j|| d S r2 )r( rH rF r- _save_import)r rJ rK r0 r r r s zCmfAttachment._save_importc
C s^ ddl m} d}| ddddg ||dd jf | j| j| j| j| j | j
j d
d }|j S ) Nr )
BeautifulSoupap
r+ r* r) cmf_created_at
r )titler{ Z
urlPreviewImgZ
urlPreviewr) Z
createDatez html.parser)
Zbs4r rp replaceformatr r{ r+ r* r) r ZjsonZ
currentTag)r r ZTAG_TMPLsr r r create_preview_tag s 6z CmfAttachment.create_preview_tag)N)NF)"__name__
__module__ __qualname__Zcmfr; Z CmfEntityZ ui_meta_skipr r( r. r: rF rG rI propertyrM rH rS rT bytesr^ r` rf boolr rA rh rw rz r@ r r r
__classcell__r r r0 r r
s6
r
)r% rP r r shutilpathlibr Z cmf.includeZ
cmf.modelsr Zcmf.utilr Z&common.models.cmf_plugin_convert_mixinr Zmodules.files.fieldsr r
r r r r s