U
]e{; @ 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 ZedddZe dddZ
dd Z e d
d d Z
e e d
ddZdd Zdd Z fddZdd Zdd Zdd Z fd d Zed d Zd d!d"d#Zee dd$d%Zd&d'd(ed)d*d+Zd,d- Z dKd.d/Z ee dd0d1Z ee dd2d3Z d4d5 Z!d'd6 fd7d8
Z" fd9d:Z#dLd;d<Z$ fd=d>Z% fd?d@Z& fdAdBZ'dCdD Z(dEdF Z)dGdH Z* fdIdJZ+ Z,S )M
CmfAttachment) file_exists)returnc C s | }|std g S | S N DEV: Не найден Rfile, вероятно не найден файл в хранилище /files для текущего CmfAttachment) get_rfile cmf_alert get_versions)selfrfile r (./modules/files/models/cmf_attachment.pyr s
zCmfAttachment.get_versions)versionc O s, | }|std d S |j|d|i|S )Nr r )r r remove_version)r r argskwargsr r r r r s
z CmfAttachment.remove_versionc O s* | }|std d S |j|f||S r
)r r gen_url_for_webdav)r Zeditorr r r r r r r ! s
z CmfAttachment.gen_url_for_webdavos_typec C s$ | }|std d S |j|dS Nr r )r r get_url_filer r r r r r r ) s
zCmfAttachment.get_url_file)r r c C s$ | }|std d S |j|dS r )r r
get_local_urlr r r r r 1 s
zCmfAttachment.get_local_urlc C s: | j js d S tj| j jd }d}t||kr6|| _d S )N )name
is_changedospathsplitextvaluelenZ file_type)r extensionZext_max_lengthr r r _calc_file_type9 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_nameC s 8zCmfAttachment._calc_namec s t ddddg S )Nurl url_previewurl_preview_imgparent)supersave_preload_fieldsr __class__r 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 rL U s
zCmfAttachment.upload_dirc C s4 t jjj }|| j}|jr*|j| _nd| _d S )Nr ) modelsCmfRFiledp data_driverget_rdr
_file_namerJ st_sizer rdiskr r r r
_calc_st_size^ s
zCmfAttachment._calc_st_sizec C s4 t jjj }|| j}|jr0t j | dd d S )NZ
attachmentZ
after_save)
rM rN rO rP rQ r rR rJ ZCmfAutomationCrudTriggerZ crud_hookrT r r r _call_automation_hookf s z#CmfAttachment._call_automation_hookc s: | | | | t j||}| |S rD )r: _calc_urlr+ rV r? saverW )r r r resrB r r rY m s zCmfAttachment.savec C s | | j S rD )rL r# rA r r r file_pathv 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& rF urllibparseZquoterR r< r= )r r] r r r r r rX { s
zCmfAttachment._calc_urlc C s t ttj| j S )u< Полный фактический путь к файлу )strr rG rH rR rA r r r full_path_file s z CmfAttachment.full_path_fileTFbackup smart_backup)contentc O sF t jjj }|| j}|jdddd |j|||d | dS )u Загрузка файла FT)is_direxist_okparentsrb N)
rM rN rO rP rQ r rR createwriterY )r re rc rd r r rU r r r r upload_file s
zCmfAttachment.upload_filec O s$ | j dd}|j|| | d S )NT)ri )r Z stream_writerY r r r r r r r upload_stream_file s z CmfAttachment.upload_stream_filec k sR | }||}|r |d }n| j dgdjj}t|d }|V W 5 Q R X d S )Nabspathr\ rb)r Z get_versionrn r( open)r r r r Z
rfile_versionrn filer r r get_content s
zCmfAttachment.get_contentc C s t jjj }|| j}|jS rD )rM rN rO rP rQ r rR rJ rT r r r r s zCmfAttachment.file_existsc C s8 | j | j r(tj| j | jjS tddd dS )uT Имя файла вложения в зависимости от родителя u Вложение должно быть привязано к объекту, нельзя сохранять в корень диска.T)abortN) r> rE r% r& rF rI r# r( r rA r r r rR s
zCmfAttachment._file_namec O s | }|j||S rD )r get_fullviewrl r r r rt s zCmfAttachment.get_fullview)forcec s8 | j r"| }|r"|j| jjd t j|d|i|S )N)prefixru )rR r Z
move_to_trashcoder( r? delete)r ru r r r rB r r rx s
zCmfAttachment.deletec s6 t jf |}tjj| jj d| jj | jd |S )N_)r# Z
target_dir) r? restorerM rN Zrestore_from_trashrw r( r# rR )r r rZ rB r r rz s $zCmfAttachment.restorec K sd dd l }tjj| j|d}|s`|r`tj| jdd}| j|_tjj|j|jj dd|_
| |S )Nr )idr] F)r# rf T)r{ mkdir) r% rM rN getrR r# r{ r& dirnamer( r> rY )r r] ri _kwargsr% rZ r r r r s zCmfAttachment.get_rfilec s6 | j r(t| j }r(|jd| jd n
t d S )NzPPP-ATT-CREATEobj)Z
project_idZcmfutilZ
get_obj_by_idcheck_project_role_accessr> r? check_edit_perm)r projectrB r r r s z CmfAttachment.check_edit_permc s | dddg | jrvtj| jkrbz| jjd| jd W qt tk
r^ | jjd| jd Y qtX q| jjd| jd n
t d S )Nr r> cmf_ownerzPPP-ATT-DELETE-ALLr zPPP-ATT-DELETE-OWN)
load_fieldsr gZ current_userr r r> ZCmfPermissionErrorr? check_delete_permrA rB r r r s z CmfAttachment.check_delete_permc s2 | | | | t j|| d S rD )r: rX r+ rV r? _save_importr r r rB 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 r replaceformatr# r{ r= r< r; r ZjsonZ
currentTag)r r ZTAG_TMPLsr r r create_preview_tag s 6z CmfAttachment.create_preview_tagc O s | j r tjj| df|| d S )NZattachment_created)Zis_newrM CmfEventdo_eventr r r r _do_event_save& s z CmfAttachment._do_event_savec O s t jj| df|| d S )NZattachment_deleted)rM r r r r r r _do_event_deleted* s z CmfAttachment._do_event_deletedc sB t t }| dg | jr:| j D ]}|| q*t|S )Nr> )setr?
get_ownersr r> addlist)r r r ZownersZ parent_ownerrB r r r - s zCmfAttachment.get_owners)N)NF)-__name__
__module__ __qualname__ZcmfrM Z CmfEntityZ ui_meta_skipr r intr r r` r r r+ r: r@ rL rV rW rY propertyr[ rX ra bytesrk rm rr boolr rR rt rx rz r r r r r r r r
__classcell__r r rB r r
sH
Br
)r7 r^ 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