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
fd
d Z e d d
Z
ddddZe edddZdddedddZd.dd Ze edd d Ze edd d Zdd! fd"d#
Z fd$d%Zd/d&d'Zd(d) Z fd*d+Zd,d- Z Z S )0
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 s& | | | t j||S r, )r" _calc_urlrB r' saver argskwargsr* r r! rD 4 s zCmfAttachment.savec C s | | j S r, )r6 r r) r r r! file_path: 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 rI rF rG r r r! rC ? s
zCmfAttachment._calc_url)returnc C s t ttj| j S )u< Полный фактический путь к файлу )strr r1 r2 r= r) r r r! full_path_fileF s z CmfAttachment.full_path_fileTFbackup smart_backup)contentc O s> t jjj }|| j}|jdddd |j|||d dS )u Загрузка файла FT)is_direxist_okparentsrO N) r7 r8 r9 r: r; r< r= Zcreatewrite)r rR rP rQ rF rG r@ rA r r r! upload_fileK s zCmfAttachment.upload_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_contentS s
zCmfAttachment.get_contentc C s t jjj }|| j}|jS r, )r7 r8 r9 r: r; r< r= r4 r? r r r! r Y 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= _ 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)prefixra ) load_fieldsrb r
startswithcmfutil
get_obj_by_idgZ current_userrc check_project_role_accessr& ZCmfPermissionErrorr= r< Z
move_to_trashcoder' delete)r ra rF rG projectrA r* r r! ro i s zCmfAttachment.deletec s6 t jf |}tjj| jj d| jj | jd |S )N_)r Z
target_dir) r' restorer7 r8 Zrestore_from_trashrn r r r= )r rG resr* r r! rr | s $zCmfAttachment.restorec K s t jj| j|dS )N)idrI )r7 r8 getr= )r rI _kwargsr r r! r< s zCmfAttachment.get_rfilec C s8 | j r4| j jdr4t| j }r4|jd| jd d S )Nrd zPPP-ATT-CREATEre )rb r ri rj rk rm r& )r rp r r r! _check_project_perm s z!CmfAttachment._check_project_permc s* | | | t j|| d S r, )r" rC rB r' _save_importrE r* r r! rx 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 )titlert Z
urlPreviewImgZ
urlPreviewr# Z
createDatez html.parser)
Zbs4ry rh replaceformatr rt r% r$ r# rz ZjsonZ
currentTag)r ry ZTAG_TMPLsr r r! create_preview_tag s 6z CmfAttachment.create_preview_tag)N)N) __name__
__module__ __qualname__Zcmfr7 Z CmfEntityZ ui_meta_skipr" r( r6 rB rD propertyrH rC rM rN bytesrW r^ boolr r= ro rr r< rw rx r
__classcell__r r r* r! r
s.
r
)r rJ 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