U
3gL @ s` d dl Z d dlZd dlmZ d dlT d dlmZ d dlmZ d dl m
Z
G dd de
j eZ dS ) N)Path)*) cmf_hashlib)CmfPluginConvertMixin)cmf_attachmentc s e Zd Zejjjd Zejj e
j dddddddd d
d d d
dddg Z e dddZ e
dddZdd ZedddZeed d d Zd d Zd!d" Zd#d$ Z fd%d&Zd'd( Zd)d* Zd+d, Z fd-d.Zed/d0 Z d1d2d3d4Z eedd5d6Z d7d8d7d9e d:d;d<Z d=d> Z!d?d@ Z"dbdAdBZ#ee$ddCdDZ%eeddEdFZ&dGdH Z'd8dI fdJdK
Z( fdLdMZ)dcdNdOZ* fdPdQZ+ fdRdSZ, fdTdUZ-dVdW Z.dXdY Z/dZd[ Z0 fd\d]Z1 fd^d_Z2e3ddd`daZ4 Z5S )e
CmfAttachment) file_existsZupsert get_fullviewgen_url_for_webdav get_versionsremove_version get_url_file
get_local_urlZ mark_viewedZ
public_getZ
public_createZpublic_get_rfileZpublic_get_versionsZ
api_createZ
api_updateZ
api_delete)returnc C s. | ddg | }|s&td g S | S )Nst_sizeZcmf_modified_at DEV: Не найден Rfile, вероятно не найден файл в хранилище /files для текущего CmfAttachment) load_fields get_rfile cmf_alertr 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 r )r r argskwargsr r r r r + s
z CmfAttachment.remove_versionc O s* | }|std d S |j|f||S )Nr )r r r
)r Zeditorr r r r r r r
3 s
z CmfAttachment.gen_url_for_webdavos_typec C s$ | }|std d S |j|dS Nr r )r r r
r r r r r r r
; s
zCmfAttachment.get_url_file)r r c C s$ | }|std d S |j|dS r )r r r r r r r r C 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_typeK 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_nameU s 8zCmfAttachment._calc_namec C s. t | jjtjr*| dg | jjr*d| _d S )Nzparent.privateT)
isinstanceparentr( modelsZ
CmfCommentr privater r r r
_calc_privated s zCmfAttachment._calc_privatec s t dddddg S )Nurl url_previewurl_preview_imgr< Zembedded)supersave_preload_fieldsr? __class__r r rE j 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 rP m s
zCmfAttachment.upload_dirc C s4 t jjj }|| j}|jr*|j| _nd| _d S )Nr ) r= CmfRFiledp data_driverget_rdr
_file_namerN r r rdiskr r r r
_calc_st_sizev s
zCmfAttachment._calc_st_sizec C s4 t jjj }|| j}|jr0t j | dd d S )NZ
attachmentZ
after_save)
r= rQ rR rS rT r rU rN ZCmfAutomationCrudTriggerZ crud_hookrV r r r _call_automation_hook~ s z#CmfAttachment._call_automation_hookc sB | | | | | t j||}| |S rH )r: _calc_urlr+ rX r@ rD saverY )r r r resrF r r r[ s zCmfAttachment.savec C s | | j S rH )rP r# r? r r r file_path 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$ rA Zis_nullr% r& rJ urllibparsequoterU rB rC )r r_ r r r r r rZ s
zCmfAttachment._calc_urlc C s t ttj| j S )u< Полный фактический путь к файлу )strr rK rL rU r? r r r full_path_file s z CmfAttachment.full_path_fileTFbackup smart_backup make_preview)contentc O sH t jjj }|| j}|jdddd |j||||d | dS )u Загрузка файла FT)is_direxist_okparentsre N)
r= rQ rR rS rT r rU createwriter[ ) r ri rf rg rh r r rW r r r r upload_file s
zCmfAttachment.upload_filec C s | }| dS )u Делаем отметку в мете что эту версию файла менять нельзя поскольку она опубликована
N)r
do_publishr r r r rp s zCmfAttachment.do_publishc O sD | j dd}|si }tjj|d< tjj|d< |j|| | d S )NT)rm task_attachment_max_size document_attachment_max_size)r gZglobal_settingsrq rr Z stream_writer[ 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_versionrv r( open)r r r r Z
rfile_versionrv filer r r get_content s
zCmfAttachment.get_contentc C s t jjj }|| j}|jS rH )r= rQ rR rS rT r rU rN rV r r r r s zCmfAttachment.file_existsc C s8 | j | j r(tj| j | jjS tddd dS )uT Имя файла вложения в зависимости от родителя u Вложение должно быть привязано к объекту, нельзя сохранять в корень диска.T)abortN) r< rI r% r& rJ rM r# r( r r? r r r rU s
zCmfAttachment._file_namec O s | }|j||S rH )r r rt r r r r s zCmfAttachment.get_fullview)forcec s< t j|d|i|}| jr8| }|r8|j| jjd |S )Nr| )prefix)rD deleterU r Z
move_to_trashcoder( )r r| r r r\ r rF r r r~ s zCmfAttachment.deletec s6 t jf |}tjj| jj d| jj | jd |S )N_)r# Z
target_dir) rD restorer= rQ Zrestore_from_trashr r( r# rU )r r r\ rF r r r s $zCmfAttachment.restorec K sd dd l }tjj| j|d}|s`|r`tj| jdd}| j|_tjj|j|jj dd|_
| |S )Nr )idr_ F)r# rj T)r mkdir) r% r= rQ getrU r# r r& dirnamer( r< r[ )r r_ rm _kwargsr% r\ 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_idcmfutil
get_obj_by_idcheck_project_role_accessr< rD check_edit_perm)r projectrF 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)
r r rs current_userr r r< ZCmfPermissionErrorrD check_delete_permr? rF r r r s z CmfAttachment.check_delete_permc s2 | | | | t j|| d S rH )r: rZ r+ rX rD _save_importr r r rF 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 )
BeautifulSoupa
rC rB rA cmf_created_at
r. )titler Z
urlPreviewImgZ
urlPreviewrA Z
createDatez html.parser)
Zbs4r r replaceformatr# r rC rB rA r ZjsonZ
currentTag)r r ZTAG_TMPLsr r r create_preview_tag
s 7z CmfAttachment.create_preview_tagc O s | j r tjj| df|| d S )NZattachment_created)Zis_newr= CmfEventdo_eventr r r r _do_event_saveM s z CmfAttachment._do_event_savec O s t jj| df|| d S )NZattachment_deleted)r= r r r r r r _do_event_deletedQ s z CmfAttachment._do_event_deletedc sB t t }| dg | jr:| j D ]}|| q*t|S )Nr< )setrD
get_ownersr r< addlist)r r r ZownersZ parent_ownerrF r r r T s zCmfAttachment.get_ownersc s4 t j||}t| jd}|| W 5 Q R X d S )Nrw )rD copyrx rd ru )r r r Zobj_copyfrF r r r ^ s zCmfAttachment.copyc C s |r|t jkr@t jst jr dS |r2t jj|kr2dS t j}t j}nd}tj|ddddddgd}|j rrt j
d krrd
S |j s|j j
d kr|jrdS |j sdS |j|jkrdS |j D ]`}|jd kr|j r||j | S |j|jkr dS |dkrtjj|dd
}|j|kr dS qdS )u
https://bcrm.carbonsoft.ru/project/Document/DOC-014685#spec-014684b
Проверяем видимость аттача для g.current_person (или check_for_person)
При использовании check_for_user метод можт выполняться долго! Обязательно перед вызовом
проверяйте, что аттач имеет флаг private
T.
spectators cmf_owner_idr> r zproject.logic_prefix)Zsimpler_ ZapiFzproject.servicedeskZCmfProjectRole)Zid_onlyN)rs r Zdisable_permissionsZacl_admin_modeZcurrent_personr Zcurrent_person__member_ofr r r> Z api_scoper r Z logic_prefixZ
user_localr
class_nameZin_project_roler= Z
CmfAccessListZsubject_full_group_list)clsZ
attachment_idZowner_idZcheck_for_userZcheck_for_user__member_ofZattachZsubjectr r r check_visibilityc s8
z CmfAttachment.check_visibility)N)NF)NN)6__name__
__module__ __qualname__Zcmfr= Z CmfEntityZ ui_meta_skipr r Z api_methodsr r r intr r
rc r
r r+ r: r@ rE rP rX rY r[ propertyr] rZ rd bytesro rp ru rz boolr rU r r~ r r r r r r r r r r classmethodr
__classcell__r r rF r r sz
C
r ) r7 r` pathlibr Z cmf.includeZcmf.utilr Z&common.models.cmf_plugin_convert_mixinr Zmodules.files.fieldsr r r r r r s