o
    i	                     @   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
 d dlmZ d dlmZ edd	gd
ZejdddddefddZejddddee
fdefddZejddddee
fdefddZdS )    )	APIRouterDependsHTTPException)
DictCursor)Optional)datetime)verify_api_key)CompanyController)Requestz
/companieszAPI Companies)prefixtags    z Registra ou atualiza uma empresa)status_codesummarypayloadc                 C   s&   t  | j}|d u rtddd|S Ni  u   Empresa não encontrada)r   detail)r	   storecookier   )r   result r   /app/routers/company.pycreate_or_update_company   s   r   z/{cnpj}   zBusca uma empresa por CNPJcnpjc                 C   $   t  | }|d u rtddd|S r   )r	   showr   r   _r   r   r   r   get_session_by_cnpj_api      r    zRemove uma empresa por CNPJc                 C   r   r   )r	   deleter   r   r   r   r   delete_session_by_cnpj_api"   r!   r#   N)fastapir   r   r   psycopg2.extrasr   typingr   r   utilsr   Zcontrollers.companyr	   Zmodels.companyr
   routerpostr   getstrr    r"   r#   r   r   r   r   <module>   s"    