a
    Qh
                     @   s   d dl mZ ddlmZmZmZmZ d dlmZ d dl	Z	G dd dej
ZeeG dd	 d	ejZeeG d
d dejZG dd dej
ZeeG dd dejZeeG dd dejZdS )    )admin   )	ProtocoloProtocoloSemeaduraProtocoloFinalizadoProtocoloSemeaduraFinalizado)HttpResponseNc                   @   s   e Zd ZeZdZdS )ProtocoloSemeaduraInliner   N)__name__
__module____qualname__r   modelextra r   r   -/var/www/sistema_ama/apps/protocolos/admin.pyr	   
   s   r	   c                   @   sp   e Zd ZdZdZdZg dZegZdddifddd	ifd
ddiffZ	dgZ
dd Zde_ejdddd ZdS )ProtocoloAdmin)num_protocologet_cliente_nometipo_clientetipo_pagamentodata_criacaodata_saida_protocolo
finalizado)r   r   r   r   r   Zcliente_fisico__nome_completoZcliente_empresa__nome_fantasia)cliente_fisicocliente_empresa	semeadurau   Informações do Clientefields)r   r   r   zDetalhes do Protocolo)r   referente_nfr   
num_pedidodestino_mudaplaca_veiculoresp_recebimentoinscricao_estadualr   r   r   u   Responsáveis e Observações)produtorrenasem_produtorresp_tecnicorenasem_tecnico
observacaoexportar_csvc                 C   s4   |j dkr|jr|jjS |j dkr0|jr0|jjS dS NFE-r   r   nome_completor   nome_fantasiaselfobjr   r   r   r   ,   s
    zProtocoloAdmin.get_cliente_nomeClientezExportar protocolos para CSV)descriptionc                    s^   t dd}d|d< t|}|dd tjjD  |D ]  | fddtjjD  q8|S )Nztext/csv)content_typez#attachment; filename=protocolos.csvzContent-Dispositionc                 S   s   g | ]
}|j qS r   name.0fieldr   r   r   
<listcomp>9       z/ProtocoloAdmin.exportar_csv.<locals>.<listcomp>c                    s   g | ]}t  |jqS r   )getattrr8   r9   r3   r   r   r<   ;   r=   )r   csvwriterwriterowr   _metar   )r2   requestquerysetresponserA   r   r?   r   r)   4   s    

zProtocoloAdmin.exportar_csvN)r
   r   r   list_displaylist_filtersearch_fieldsautocomplete_fieldsr	   inlines	fieldsetsactionsr   short_descriptionr   actionr)   r   r   r   r   r      s(   
r   c                   @   s    e Zd ZdZdZddgZdZdS )ProtocoloSemeaduraAdmin)	protocolor   	categoriaqtd_especie
valor_unitvalor_total
data_venda)Zprotocolo__num_protocolosemeadura__numero_mudarQ   r   rR   rV   N)r
   r   r   rG   rI   rJ   rH   r   r   r   r   rP   ?   s   rP   c                   @   s   e Zd ZeZdZdZdS )"ProtocoloSemeaduraFinalizadoInliner   )r   rR   rS   rT   rU   rV   N)r
   r   r   r   r   r   readonly_fieldsr   r   r   r   rY   J   s   rY   c                   @   sP   e Zd ZdZdZdZdd ejjD Z	e
gZdd Zde_d	d
 ZdddZdS )ProtocoloFinalizadoAdmin)r   r   r   data_finalizacaousuario_finalizacaor   r   )r   r   r\   c                 C   s   g | ]
}|j qS r   r7   r9   r   r   r   r<   X   r=   z#ProtocoloFinalizadoAdmin.<listcomp>c                 C   s4   |j dkr|jr|jjS |j dkr0|jr0|jjS dS r*   r.   r1   r   r   r   r   [   s
    z)ProtocoloFinalizadoAdmin.get_cliente_nomer4   c                 C   s   dS NFr   r2   rD   r   r   r   has_add_permissionc   s    z+ProtocoloFinalizadoAdmin.has_add_permissionNc                 C   s   dS r^   r   r2   rD   r3   r   r   r   has_change_permissionf   s    z.ProtocoloFinalizadoAdmin.has_change_permission)N)r
   r   r   rG   rI   rH   r   rC   r   rZ   rY   rK   r   rN   r`   rb   r   r   r   r   r[   P   s   r[   c                   @   s<   e Zd ZdZdZdZdd ejjD Z	dd Z
dd	d
ZdS )!ProtocoloSemeaduraFinalizadoAdmin)protocolo_finalizador   rR   rS   rT   rU   rV   )Z#protocolo_finalizado__num_protocolorW   rX   c                 C   s   g | ]
}|j qS r   r7   r9   r   r   r   r<   r   r=   z,ProtocoloSemeaduraFinalizadoAdmin.<listcomp>c                 C   s   dS r^   r   r_   r   r   r   r`   t   s    z4ProtocoloSemeaduraFinalizadoAdmin.has_add_permissionNc                 C   s   dS r^   r   ra   r   r   r   rb   w   s    z7ProtocoloSemeaduraFinalizadoAdmin.has_change_permission)N)r
   r   r   rG   rI   rH   r   rC   r   rZ   r`   rb   r   r   r   r   rc   j   s   rc   )Zdjango.contribr   modelsr   r   r   r   django.httpr   r@   TabularInliner	   register
ModelAdminr   rP   rY   r[   rc   r   r   r   r   <module>   s   /
