a
    ;Xh                     @   sl   d dl mZ ddlmZmZ d dlmZ d dlZeeG dd dej	Z
eeG dd	 d	ej	ZdS )
    )admin   )RepiqueLogFinalizacaoRepique)HttpResponseNc                   @   s^   e Zd ZdZdZdZg dZdddiffZdd	gZe	j
d
ddd Ze	j
dddd ZdS )RepiqueAdmin)	semeaduracadastro_lotedata_repiquequantidade_repicadafuncionario
finalizadoregistrado_por)r   r
   r   )Zsemeadura__numero_mudaZcadastro_lote__numero_loteZfuncionario__nome_completo)r   r	   r   r   u   Informações do Repiquefields)r   r	   r
   r   r   r   r   marcar_como_finalizadoexportar_csvzMarcar como finalizado)descriptionc                 C   s"   |j dd}| || d d S )NT)r   z$ repiques marcados como finalizados.)updatemessage_user)selfrequestquerysetupdated r   6/var/www/viveiro_mudafortebrasil/apps/repique/admin.pyr      s    z#RepiqueAdmin.marcar_como_finalizadozExportar repiques para CSVc                    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=repiques.csvzContent-Dispositionc                 S   s   g | ]
}|j qS r   )name.0fieldr   r   r   
<listcomp>(       z-RepiqueAdmin.exportar_csv.<locals>.<listcomp>c                    s   g | ]}t  |jqS r   )getattrr   r   objr   r   r    *   r!   )r   csvwriterwriterowr   _metar   )r   r   r   responser&   r   r#   r   r   #   s    

zRepiqueAdmin.exportar_csvN)__name__
__module____qualname__list_displaylist_filtersearch_fieldsautocomplete_fields	fieldsetsactionsr   actionr   r   r   r   r   r   r      s   	


r   c                   @   sJ   e Zd ZdZdZdZg dZdd Zde_dd	 Z	dddZ
dddZd
S )LogFinalizacaoRepiqueAdmin)r   numero_lote	data_lotedata_exclusaohora_exclusaousuario_exclusaonome_usuario_exclusaomotivo_formatado)r7   motivo_exclusao)r5   r:   r<   )	r   r5   r6   r7   r8   r9   r:   r<   motivo_exclusao_outrosc                 C   s(   |j dkr"|jr"|j  d|j S |j S )NZOutrosz - )r<   r=   )r   r$   r   r   r   r;   ?   s    z+LogFinalizacaoRepiqueAdmin.motivo_formatadoZMotivoc                 C   s   dS NFr   )r   r   r   r   r   has_add_permissionE   s    z-LogFinalizacaoRepiqueAdmin.has_add_permissionNc                 C   s   dS r>   r   r   r   r$   r   r   r   has_change_permissionH   s    z0LogFinalizacaoRepiqueAdmin.has_change_permissionc                 C   s   dS r>   r   r@   r   r   r   has_delete_permissionK   s    z0LogFinalizacaoRepiqueAdmin.has_delete_permission)N)N)r*   r+   r,   r-   r.   r/   readonly_fieldsr;   short_descriptionr?   rA   rB   r   r   r   r   r4   .   s   
r4   )Zdjango.contribr   modelsr   r   django.httpr   r%   register
ModelAdminr   r4   r   r   r   r   <module>   s   &