a
    NhE                     @   s   d dl mZmZ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mZ d dlmZ d d	lmZ d
d Zdd Zdd Zdd ZdS )    )renderredirectget_object_or_404)modelformset_factory)messages)	timedelta   )RegistroPaperHorarioParada)RegistroPaperFormHorarioParadaForm)get_user_info)datetimec                 C   s   | j jstdS t| j j}tj }| j	d}| j	d}|r|rz4t
|d }t
|d }|j||fd}W n ty   Y n0 |d}t| d||dS )	Nindexiniciofimz%Y-%m-%d)Zdata__rangez-datazpaperpot/lista_paperpot.html)	user_info	registros)useris_authenticatedr   r   idr	   objectsallGETgetr   strptimedatefilter
ValueErrororder_byr   )requestr   r   r   r   data_iniciodata_fim r#   +/var/www/sistema_ama/apps/paperpot/views.pylista_paperpot
   s$    

r%   c                 C   s  | j jstdS t| j j}tttddd}| jdkrt	| j
}|| j
tj d}| r| r|jdd}|jr|jd	 nd
|_|jr|j|j nd
|_d
}d
}|D ]V}|jr|jds|jd}	|jd}
|	r|
r|
|	 }|| d 7 }|d7 }q|d
kr |j| d nd
|_|jp0d|_|  |jrR|jd d	 nd
}t|d|_|D ]6}|jrf|jdsf|jdd}||_|  qft| d tdS t| d nt	 }|tj d}t| d|||dS )Nr   r   T)formextra
can_deletePOST)querysetF)commit(   r   DELETEZhorario_inicioZhorario_paradai      g
ףp=
?   zRegistro salvo com sucesso!r%   u    Corrija os erros no formulário.z"paperpot/criar_registro_paper.html)r&   formsetr   )r   r   r   r   r   r   r
   r   methodr   r)   r   noneis_validsavesubstrato_baldesubstrato_litrosquantidade_paper_cheioproducao_por_litrocleaned_datar   total_secondsZproducao_por_papel
observacaoroundbasocote_kgregistror   successerrorr   )r    r   ZHorarioParadaFormSetr&   r1   r?   Ztotal_horasZnum_paradasfr   paradadeltaZbasocoteZhorarior#   r#   r$   criar_registro_paper%   s^    

rE   c           
   	   C   s   | j jstdS t| j j}tt|d}tjj	|d}| j
dkrt| j|d}| r|  tjj	|d  t| jdd}t|D ]B}| jd| }| jd	| }	|r|	rtjj|||	d
 qtdS n
t|d}t| d|||| |dS )Nr   pk)r?   r)   )instancetotal_horariosr   Zhora_inicio_Zhora_parada_)r?   hora_iniciohora_paradar%   z#paperpot/editar_registro_paper.html)r&   r?   horariosrI   r   )r   r   r   r   r   r   r	   r
   r   r   r2   r   r)   r4   r5   deleteintr   rangecreater   count)
r    rG   r   r?   rL   r&   totalir   rC   r#   r#   r$   editar_registro_paperg   s:    


rT   c                 C   s8   | j jstdS tt|d}|  t| d tdS )Nr   rF   u   Registro excluído com sucesso!r%   )r   r   r   r   r	   rM   r   r@   )r    rG   r?   r#   r#   r$   excluir_registro_paper   s    rU   N)django.shortcutsr   r   r   Zdjango.formsr   django.contribr   r   r   modelsr	   r
   formsr   r   apps.dashboard.utilsr   r%   rE   rT   rU   r#   r#   r#   r$   <module>   s   B*