U
]d / @ 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 Z fddZdd Z dd Z
d
d Z fd d
Z e
dd ZddddZe
edddZdddeddd Zd d Zd2d d Ze
edd!d"Ze
edd#d$Zdd% fd&d'
Z fd(d)Zd3d*d+Zd,d- Z fd.d/Z d0d1 Z Z S )4
CmfAttachment) file_existsc 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 [А-Яа-я] . - ) name
is_changedrefindallvaluelensplitmathZfloorstripr Z
short_str_enc)selfZ
rus_simbolZ
os_max_lengthZfile_name_lengthextZ slice_length r (./modules/files/models/cmf_attachment.py
_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& loadospathjoinconfig
UPLOAD_DIR
get_files_direxistsmakedirs)r
upload_dirr r r! r6 # s
zCmfAttachment.upload_dirc C s4 t jjj }|| j}|jr*|j| _nd| _d S )Nr ) modelsCmfRFiledp data_driverget_rd get_rfile
_file_namer4 st_sizer rdiskrfiler r r!
_calc_st_size, s
zCmfAttachment._calc_st_sizec C s4 t jjj }|| j}|jr0t j | dd d S )NZ
attachmentZ
after_save)
r7 r8 r9 r: r; r< r= r4 ZCmfAutomationCrudTriggerZ crud_hookr? r r r! _call_automation_hook4 s z#CmfAttachment._call_automation_hookc s2 | | | t j||}| |S r, )r" _calc_urlrB r' saverC )r argskwargsresr* r r! rE ; s zCmfAttachment.savec C s | | j S r, )r6 r r) r r r! file_pathC s zCmfAttachment.file_pathN)fieldsc O sH | j js| jjsd S dtj| j | _| j d| _| j d| _ d S )Nz/filesz.meta/preview.pdfz.meta/preview_img.png)
r r r# Zis_nullurllibparseZquoter= r$ r% )r rJ rF rG r r r! rD H s
zCmfAttachment._calc_url)returnc C s t ttj| j S )u< Полный фактический путь к файлу )strr r1 r2 r= r) r r r! full_path_fileO s z CmfAttachment.full_path_fileTFbackup smart_backup)contentc O s> t jjj }|| j}|jdddd |j|||d dS )u Загрузка файла FT)is_direxist_okparentsrP N) r7 r8 r9 r: r; r< r= createwrite)r rS rQ rR rF rG r@ rA r r r! upload_fileT s zCmfAttachment.upload_filec O s | j dd}|j|| d S )NT)rW )r< Z stream_write)r rF rG rA r r r! upload_stream_file[ s z CmfAttachment.upload_stream_filec
K s@ | }||}t|d d}| W 5 Q R S Q R X d S )Nabspathrb)r< Z get_versionopenread)r versionrG rA Z
rfile_versionfiler r r! get_content` s
zCmfAttachment.get_contentc C s t jjj }|| j}|jS r, )r7 r8 r9 r: r; r< r= r4 r? r r r! r f s zCmfAttachment.file_existsc C s> | j j}| j | jr0tj| j | j j}tjd|S )uT Имя файла вложения в зависимости от родителя /)r r r& r- r. r/ r0 r3 )r file_namer r r! r= l s
zCmfAttachment._file_name)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)prefixrd ) load_fieldsre r
startswithcmfutil
get_obj_by_idgZ current_userrf check_project_role_accessr& ZCmfPermissionErrorr= r< Z
move_to_trashcoder' delete)r rd rF rG projectrA r* r r! rr v s zCmfAttachment.deletec s6 t jf |}tjj| jj d| jj | jd |S )N_)r Z
target_dir) r' restorer7 r8 Zrestore_from_trashrq r r r= )r rG rH r* r r! ru s $zCmfAttachment.restorec K sd dd l }tjj| j|d}|s`|r`tj| jdd}| j|_tjj|j|jj dd|_
| |S )Nr )idrJ F)r rT T)rv mkdir) r. r7 r8 getr= r rv r/ dirnamer r& rE )r rJ rW _kwargsr. rH r r r! r< s zCmfAttachment.get_rfilec C s8 | j r4| j jdr4t| j }r4|jd| jd d S )Nrg zPPP-ATT-CREATErh )re r rl rm rn rp r& )r rs r r r! _check_project_perm s z!CmfAttachment._check_project_permc s* | | | t j|| d S r, )r" rD rB r' _save_import)r rF rG r* 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 )titlerv Z
urlPreviewImgZ
urlPreviewr# Z
createDatez html.parser)
Zbs4r} rk replaceformatr rv 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__Zcmfr7 Z CmfEntityZ ui_meta_skipr" r( r6 rB rC rE propertyrI rD rN rO bytesrY rZ ra boolr r= rr ru r< r{ r| r
__classcell__r r r* r! r
s2
r
)r rK 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