U
ƩhW @ 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 fddZ e dddZ
edddZdd Zedd d Zeed d d Zd!d" Zd#d$ Z fd%d&Zd'd( Zd)d* Zd+d, Zd-d. Z fd/d0Ze e d1d1d2d3d4 Z e d5d6 Z d7d8d9d:Z!e edd;d<Z"d1d=d1d1d>e#d?d@dAZ$dBdC Z%dDdE Z&dkdFdGZ'e e(ddHdIZ)e eddJdKZ*dLdM Z+d=dN fdOdP
Z, fdQdRZ-dldSdTZ. fdUdVZ/ fdWdXZ0 fdYdZZ1d[d\ Z2d]d^ Z3d_d` Z4 fdadbZ5 fdcddZ6e7dmdedfZ8e e d1d1dgdhdidj Z9 Z:S )n
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_deletec s d | _ t j|| d S N)content_updatedsuper__init__selfargskwargs __class__ (./modules/files/models/cmf_attachment.pyr ! s zCmfAttachment.__init__)returnc C s. | ddg | }|s&td g S | S )Nst_sizeZcmf_modified_at DEV: Не найден Rfile, вероятно не найден файл в хранилище /files для текущего CmfAttachment) load_fields get_rfile cmf_alertr r rfiler r r r % s zCmfAttachment.get_versions)versionc O s, | }|std d S |j|d|i|S )Nr r# )r r r )r r# r r r" 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
7 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 G s
zCmfAttachment.get_local_urlc C s: | j js d S tj| j jd }d}t||kr6|| _d S )N ) file_name
is_changedospathsplitextvaluelen file_type)r extensionZext_max_lengthr r r _calc_file_typeO s z CmfAttachment._calc_file_typec C s. t | jjtjr*| dg | jjr*d| _d S )Nzparent.privateT)
isinstanceparentr/ modelsZ
CmfCommentr privater r r r
_calc_privateX s zCmfAttachment._calc_privatec s t dddddg S )Nurl url_previewurl_preview_imgr5 Zembedded)r save_preload_fieldsr8 r r r r= ^ s z!CmfAttachment.save_preload_fieldsc C sD | j | j sd S tjtj| j }tj|s@t | |S r )
r5 loadr, r- joinconfig
UPLOAD_DIR
get_files_direxistsmakedirs)r
upload_dirr r r rE a s
zCmfAttachment.upload_dirc C s4 t jjj }|| j}|jr*|j| _nd| _d S )Nr ) r6 CmfRFiledp data_driverget_rdr
_file_namerC r r rdiskr" r r r
_calc_st_sizej s
zCmfAttachment._calc_st_sizec C s4 t jjj }|| j}|jr0t j | dd d S )NZ
attachmentZ
after_save)
r6 rF rG rH rI r rJ rC ZCmfAutomationCrudTriggerZ crud_hookrK r r r _call_automation_hookr s z#CmfAttachment._call_automation_hookc C s" | j r t| j}|r |j| _d S r )is_newAPPget_cache_project
project_idZ default_attach_full_search_typeZfull_search_typer projectr r r _calc_full_search_typey s z$CmfAttachment._calc_full_search_typec sh | | | | | t j||}tj | |
| j rdt tj
j| jgd |S )N)r ) _calc_urlr3 rM r9 rU r saver6 CmfBackbonePeerZobj_after_save_hookrN rO Zschedule_deferred_jobr extract_text_from_attachid)r r r resr r r rW s zCmfAttachment.saveT) only_once
system_jobc C s dd l }ddlm} tjj| dddgd}d}d}t|j}|jd krjd
}d }t||j j
d t j
}n|jd
krvnd S t||:}|r| }n&||} d}| jD ]}
||
7 }qW 5 Q R X |d d |_|jdd d S )Nr )unquoter; r1 Zfull_file_pathrZ fieldsTr)z.docz.docxz.dotz.dotxz.docmz.dotmz.rtfz.pdfz.xpsz.fdfz.xlsz.xlsxz.xlsmz.pptz.pptxz.odpz.odsFrbz/files/)z.txtz.csvz.mdz.html i )Z only_data)PyPDF2 urllib.parser^ r6 r getr full_path_filer1 r; r/ replacer@ rA openreadZ PdfReaderZpagesZ extract_texttextrW ) Z attach_idrd r^ attachZ
force_readmode file_pathfilerk readerZpager r r rY s,
z&CmfAttachment.extract_text_from_attachc C s | | j S r )rE r* r8 r r r rn s zCmfAttachment.file_pathNr` c O sb | j js| jjsd S tjdtj | j
| _tj| j dd| _ tj| j dd| _ d S )Nz/filesz.metaz preview.pdfz
thumbnail.jpg)
r* r+ r: Zis_nullr, r- r? urllibparsequoterJ r; r< )r r` r r r r r rV s
zCmfAttachment._calc_urlc C s t ttj| j S )u< Полный фактический путь к файлу )strr r@ rA rJ r8 r r r rg s z CmfAttachment.full_path_fileFbackup smart_backup make_preview
mark_dirty)contentc
O sV t jjj }|| j} | jdddd | j|||||d d| _ |
d| _ dS )u Загрузка файла FT)is_direxist_okparentsrv N) r6 rF rG rH rI r rJ createwriter rW )
r r{ rw rx ry rz r r rL r" r r r upload_file s zCmfAttachment.upload_filec C s. | }|r| ntd| ddd dS )u Делаем отметку в мете что эту версию файла менять нельзя поскольку она опубликована
uW При обработке документа обнаружен пустой файл: u . Возможно, произошла ошибка при загрузке файла. Пожалуйста, загрузите файл повторно.FabortN)r
do_publishr r! r r r r s
zCmfAttachment.do_publishc O sP | j dd}|si }tjj|d< tjj|d< |j|| d| _| d| _d S )NT)r task_attachment_max_size document_attachment_max_sizeF)r gZglobal_settingsr r Z stream_writer rW 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 )Nabspathrq rb )r Z get_versionr r/ ri )r r# r r" Z
rfile_versionr ro r r r get_content s
zCmfAttachment.get_contentc C s t jjj }|| j}|jS r )r6 rF rG rH rI r rJ rC rK r r r r s zCmfAttachment.file_existsc C s< | ddg | jr,tj| j | jjS tddd dS )uT Имя файла вложения в зависимости от родителя r5 r* u Вложение должно быть привязано к объекту, нельзя сохранять в корень диска.Tr N) r r5 r, r- r? rB r* r/ r r8 r r r rJ s zCmfAttachment._file_namec O s | }|j||S r )r r r r r r r s zCmfAttachment.get_fullview)forcec sH t j|d|i|}| jr8| }|r8|j| jjd tj | |S )Nr )prefix)
r deleterJ r Z
move_to_trashcoder/ r6 rX Zobj_after_delete_hook)r r r r r[ r" r r r r s zCmfAttachment.deletec s6 t jf |}tjj| jj d| jj | jd |S )N_)nameZ
target_dir) r restorer6 rF Zrestore_from_trashr r/ r* rJ )r r r[ r r r r s $zCmfAttachment.restorec K sd dd l }tjj| j|d}|s`|r`tj| jdd}| j|_tjj|j|jj dd|_
| |S )Nr r_ F)r r| T)rZ mkdir) r, r6 rF rf rJ r* rZ r- dirnamer/ r5 rW )r r` r _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)rR rP rQ check_project_role_accessr5 r check_edit_permrS r r r r + s z CmfAttachment.check_edit_permc s | dddg | jr|t| j}tj| jkrjz|jd| jd W qz t k
rf |jd| jd Y qzX q|jd| jd n
t
d S )NrR r5 cmf_ownerzPPP-ATT-DELETE-ALLr zPPP-ATT-DELETE-OWN) r rR rP rQ r current_userr r r5 ZCmfPermissionErrorr check_delete_permrS r r r r 2 s z CmfAttachment.check_delete_permc s* | | | t j|| d S r )rV r3 rM r _save_importr r r r r A s zCmfAttachment._save_importc
C s` ddl m} d}| dddddg ||d d
jf | j| j| j| j| j | j
j d d }|j S )
Nr )
BeautifulSoupa
r< r; r: cmf_created_atr*
rc )titlerZ Z
urlPreviewImgZ
urlPreviewr: Z
createDatez html.parser)
Zbs4r r rh formatr* rZ r< r; r: r ZjsonZ
currentTag)r r ZTAG_TMPLsr r r create_preview_tagG s 7z CmfAttachment.create_preview_tagc O s | j r tjj| df|| d S )NZattachment_created)rO r6 CmfEventdo_eventr r r r _do_event_save s z CmfAttachment._do_event_savec O s t jj| df|| d S )NZattachment_deleted)r6 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 )Nr5 )setr
get_ownersr r5 addlist)r r r ZownersZ parent_ownerr r r r s zCmfAttachment.get_ownersc s4 t j||}t| jd}|| W 5 Q R X d S )Nrb )r copyri rg r )r r r Zobj_copyfr 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_idr7 rT zproject.logic_prefix)Zsimpler` ZapiFzproject.servicedeskZCmfProjectRole)Zid_onlyN)r r Zdisable_permissionsZacl_admin_modeZcurrent_personrZ Zcurrent_person__member_ofcmfutilZ
get_obj_by_idr7 Z api_scoper rT Z logic_prefixZ
user_localr
class_nameZin_project_roler6 Z
CmfAccessListZsubject_full_group_list)clsZ
attachment_idZowner_idZcheck_for_userZcheck_for_user__member_ofrl Zsubjectr r r check_visibility s8
z CmfAttachment.check_visibilityz @minutely)r\ r] Zschedulec C s t jjj } | jtjs0| jj tjddd | j
tjD ]}z8| dd }t
|}|| t | j| W q> tk
r } z&t tjd| d| tjd W 5 d }~X Y q>X q>d S )NT)r~ r} /u4 Не удалось обработать файлы z: )level)r6 rF rG rH rI ioZ io_existsr@ Z
DIRTY_PATHZio_mkdirZ
io_iterdirsplitr Zget_model_by_idZprocess_dirty_attachZ
cmf_commitZ io_rmtree ExceptionZ cmf_rollbackr debugZloggingZERROR)rL Z
dirty_pathZobj_idZmodeler r r cron_check_dirty s
z CmfAttachment.cron_check_dirty)N)NF)NN);__name__
__module__ __qualname__Zcmfr6 Z CmfEntityZ ui_meta_skipr r Z api_methodsr r r r intr r
ru r
r r3 r9 r= rE rM rN rU rW staticmethodZcmf_deferred_jobrY propertyrn rV rg bytesr r r r boolr rJ r r r r r r r r r r r r classmethodr r
__classcell__r r r r r s
$
C
) r ) Zmathrr pathlibr Z cmf.includeZcmf.utilr Z&common.models.cmf_plugin_convert_mixinr Zmodules.files.fieldsr r r r r r s