U
    {hQ                     @   s   d dl Zd dlmZ d dlmZ d dlmZ d dlm	Z
 d dlmZ ddlmZ dd	lmZmZ dd
lmZ G dd deZG dd deZG dd de
Z	G dd deZG dd deZdS )    N)AddView)CreateFromUploadView)DeleteUploadView)
DeleteView)EditView   get_document_model)get_document_formget_document_multi_form)permission_policyc                       sl   e Zd ZeZdZdZdZdZdZdZ	dZ
dZdZd	Zd
Zdd Zdd Zdd Zdd Z fddZ  ZS )r   zwagtaildocs/multiple/add.htmlwagtaildocs:edit_multiplewagtaildocs:delete_multipledocdoc_id2wagtaildocs:create_multiple_from_uploaded_document"wagtaildocs:delete_upload_multipleuploaded-documentuploaded_documentuploaded_file_idc                 C   s   t  S Nr   self r   h/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/documents/views/multiple.py	get_model"   s    zAddView.get_modelc                 C   s
   t | jS r   )r
   modelr   r   r   r   get_upload_form_class%   s    zAddView.get_upload_form_classc                 C   s
   t | jS r   r   r   r   r   r   r   get_edit_form_class(   s    zAddView.get_edit_form_classc                 C   s*   |j dd}| jj|_|  |   |S )NF)commit)saverequestuseruploaded_by_user_set_document_file_metadata)r   formr   r   r   r   save_object+   s
    
zAddView.save_objectc                    s*   t  jf |}|d| jjd ji |S )NZmax_title_lengthtitle)superget_context_dataupdater&   fields
max_length)r   kwargscontext	__class__r   r   r*   3   s     zAddView.get_context_data)__name__
__module____qualname__r   Ztemplate_nameedit_object_url_namedelete_object_url_nameedit_object_form_prefixcontext_object_namecontext_object_id_nameedit_upload_url_namedelete_upload_url_nameedit_upload_form_prefixcontext_upload_nameZcontext_upload_id_namer   r   r   r'   r*   __classcell__r   r   r0   r   r      s"   r   c                   @   s8   e Zd ZeZdZdZdZdZdZdZ	dd Z
dd Zd	S )
r   r   r   r   r   c                 C   s   t  S r   r   r   r   r   r   r   H   s    zEditView.get_modelc                 C   s
   t | jS r   r   r   r   r   r   r   K   s    zEditView.get_edit_form_classN)r2   r3   r4   r   pk_url_kwargr7   r8   r9   r5   r6   r   r   r   r   r   r   r   ?   s   r   c                   @   s    e Zd ZeZdZdZdd ZdS )r   r   c                 C   s   t  S r   r   r   r   r   r   r   T   s    zDeleteView.get_modelN)r2   r3   r4   r   r?   r9   r   r   r   r   r   r   O   s   r   c                   @   s<   e Zd ZdZdZdZdZdZdZdd Z	d	d
 Z
dd ZdS )CreateFromUploadedDocumentViewr   r   r   r   r   r   c                 C   s   t  S r   r   r   r   r   r   r   `   s    z(CreateFromUploadedDocumentView.get_modelc                 C   s
   t | jS r   r   r   r   r   r   r   c   s    z2CreateFromUploadedDocumentView.get_edit_form_classc                 C   sJ   | j jjtj| jjj| jjjdd | jj	| j _
| j   |  d S )NF)r!   )objectfiler!   ospathbasenameuploadnamer"   r#   r$   r%   )r   r&   r   r   r   r'   f   s      
z*CreateFromUploadedDocumentView.save_objectN)r2   r3   r4   r:   r;   upload_pk_url_kwargr<   r9   r=   r   r   r'   r   r   r   r   r@   X   s   r@   c                   @   s   e Zd ZdZdd ZdS )r   r   c                 C   s   t  S r   r   r   r   r   r   r   x   s    zDeleteUploadView.get_modelN)r2   r3   r4   rH   r   r   r   r   r   r   u   s   r   )os.pathrC   Z+wagtail.admin.views.generic.multiple_uploadr   ZBaseAddViewr   ZBaseCreateFromUploadViewr   ZBaseDeleteUploadViewr   ZBaseDeleteViewr   ZBaseEditView r	   Zformsr
   r   Zpermissionsr   r@   r   r   r   r   <module>   s   -	