o i @sPddlmZmZmZmZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZmZmZddlmZ edd gd Ze d d Zeejjd <d'dededefddZdede eeeffddZeejjd<eddefddZededfdedefddZ edded efd!d"Z!ed#ded efd$d%Z"d&S)() APIRouterRequestFormDepends)RedirectResponse JSONResponse)Jinja2Templates) DictCursor)ListTuple)register_session_logic decrypt_datato_sao_paulo_time)get_db_connection/panelz API Web Panel)prefixtags templates) directoryrsuccessrequestmessagecategorycCs,d|jvr g|jd<|jd||fdSNZ _messages)sessionappend)rrrr/app/routers/panel.pyflashs  rreturncCsd|jvr |jdSgSr)rpop)rrrrget_flashed_messagessr!cspt}z%|jtd}|d|}Wdn1swYW|n|wtd||dS)NZcursor_factoryz1SELECT * FROM companies ORDER BY created_at DESC;zdashboard.html)rsessions)rcursorr executeZfetchallcloserZTemplateResponse)rconncurr$rrrget_panel_dashboard.s  r*z /register. cookie_stringc svz t|t|ddWn'ty4}zt|dr|jnt|}t|d|dWYd}~nd}~wwtdddS) Nu*Sessão registrada/atualizada com sucesso!rdetailzErro ao registrar: dangerr/url status_code)r r Exceptionhasattrr,strr)rr+er,rrrhandle_panel_registerJs r6z/delete/{cnpj}cnpjc st}z]z;|-}|d|f||jdkr&t|d|ddn t|d|ddWdn1s:wYWnty[}zt|d |d WYd}~nd}~wwW|n|wtd d d S)Nz&DELETE FROM companies WHERE cnpj = %s;ruSessão do CNPJ z removida com sucesso.ru'Nenhuma sessão encontrada para o CNPJ .warninguErro ao remover sessão: r-rr.r/) rr%r&commitZrowcountrr2r'r)rr7r(r)r5rrrhandle_panel_deleteTs&   r;z/get-decrypted-cookie/{cnpj}c st}zozJ|jtd}|d|f|}Wdn1s"wY|s:tddiddWW|r9|SSt|d}td |id WW|rO|SStys}ztdt |id dWYd}~W|rn|SSd}~ww|r{|ww) zn Endpoint interno para o painel. Busca um cookie pelo CNPJ, descriptografa-o e retorna como JSON. r#z-SELECT cookie FROM companies WHERE cnpj = %s;NerroruCNPJ não encontradoi)contentr1cookiedecrypted_cookie)r=i) rr%r r&Zfetchonerr'r r2r4)rr7r(r)Z session_datar?r5rrrget_decrypted_cookiees6       r@N)r)#fastapirrrrfastapi.responsesrrZfastapi.templatingrZpsycopg2.extrasr typingr r Z bootstrapr r rutilsrrouterrenvfiltersr4rr!globalsgetr*postr6r;r@rrrrs2