B
    +,_                 @   s`   d dl mZ d dlmZ d dlT d dlmZ d dlmZm	Z	 G dd deZ
G dd	 d	eZd
S )    )request)Resource)*)RealDictCursor)jwt_requiredget_jwt_identityc               @   s   e Zd Zedd ZdS )Cancelamentosc          
   C   s  t j}tdtfdtfdtff|}|r:ddt| idfS t }|jtd}y2|	t
|d |d t |d |d |d f W nB tk
r   tddid||S  tk
r   tdd	id
||S X g }t }x$|D ]}|| ||d  qW |s
tddid||S t|}|d  dkrt|dkry|	t|f |  W nF tk
rp   tddid||S  tk
r   tdd	id
||S X t|d||S )Nenviadosdt_inidt_fimcausezBad Request (arguments: i  )cursor_factoryzmissing parameterzinternal errori  id_check_listmessagezNenhum registro encontrado   FALSEr   )r   argsvalidate_argsvalidate_booleanvalidate_datestrcreate_conncursorr   executesql_cancelamentosr   KeyErrorclose_resources	Exceptionlistappendtupleupperlensql_sent_cancelamentoscommit)selfquery_string	invalidosZconn_cancelamentosr   listaZlist_idsrecord r*   =C:\Sistemas\Destra\wsDestra\resources\access_cancellations.pyget   sp    

 zCancelamentos.getN)__name__
__module____qualname__r   r,   r*   r*   r*   r+   r      s   r   c               @   s   e Zd Zedd ZdS )MarcarCancelamentosc             C   s   t  }tj}t|d }|jtd}y(|t|t f |	  t
|j}W nB tk
rn   tddid||S  tk
r   tddid||S X td|id	||S )
Nids_list)r   r   zmissing parameteri  zinternal errori  zrecord(s) affectedr   )r   r   jsonr    r   r   r   sql_check_cancelamentosr   r$   r   rowcountr   r   r   )r%   Zconn_marcar_cancelamentos	dict_bodyidsr   resultr*   r*   r+   putS   s,    zMarcarCancelamentos.putN)r-   r.   r/   r   r8   r*   r*   r*   r+   r0   R   s   r0   N)flaskr   flask_restfulr   resources.utilspsycopg2.extrasr   flask_jwt_extendedr   r   r   r0   r*   r*   r*   r+   <module>   s   K