o i@sddlmZmZddlmZmZddlmZddlm Z ddl m Z m Z ddl Z ddlmZmZddlmZmZdd lmZdd lmZdd lmZdd lmZmZed ZeddZeejjd<GdddZej ejj!d<dS))Request HTTPException)RedirectResponse JSONResponse)Jinja2Templates) DictCursor)ListTupleN)get_db_connectionto_sao_paulo_time)register_session_logic decrypt_data)timezone)datetime)EncryptionService)rResponsezAmerica/Sao_Paulo templates) directoryr c@seZdZeddededefddZededeeeeffdd Z edefd d Z ded efd dZ dedede fddZ dedefddZdS)PanelControllersuccessrequestmessagecategorycCs,d|jvr g|jd<|jd||fdS)u'Adiciona uma mensagem flash à sessão. _messagesN)sessionappend)rrrr/app/controllers/panel.pyflashs  zPanelController.flashreturncCsd|jvr |jdSgS)u/Retorna e remove as mensagens flash da sessão.r)rpop)rrrrget_flashed_messages sz$PanelController.get_flashed_messagescCszt}z4|jtd}|d|}td||dWdW|S1s,wYW|dS|w)u:Retorna todas as sessões ordenadas por data de criação.cursor_factoryz1SELECT * FROM companies ORDER BY created_at DESC;zdashboard.html)rsessionsN)r cursorrexecutefetchallrTemplateResponseclose)rconncurr$rrrget_all_sessions%s  z PanelController.get_all_sessionscookiec Csztdd|dDd}|stdtj|ddid}|d}|d }|d }t|||gs7td t|ts@td t|t t fsKtd t j |t d}t|} t} zC| /} d} | | ||| |t t f| ||ddtdddWdW| WS1swYW| WdS| wttjfy} ztdt| dd} ~ wty} z td| tdddd} ~ ww)Ncss(|]}|dr|ddVqdS)z fgtsd_token==N) startswithsplit).0srrr 3s&z(PanelController.store..z; u8O 'fgtsd_token' não foi encontrado na string de cookie.verify_signatureF)optionscnpj cnpj_nomeexpu(JWT inválido. Faltam campos essenciais.u!CNPJ deve ser uma string válida.u5Timestamp de expiração deve ser um número válido.)tza INSERT INTO companies (cnpj, name, cookie, expires_on, status, updated_at) VALUES (%s, %s, %s, %s, 'active', %s) ON CONFLICT (cnpj) DO UPDATE SET name = EXCLUDED.name, cookie = EXCLUDED.cookie, expires_on = EXCLUDED.expires_on, status = 'active', updated_at = EXCLUDED.updated_at, details = NULL; u*Sessão registrada/atualizada com sucesso!r/panel/url status_codei)r?detailzErro inesperado no registro: z$Ocorreu um erro interno no servidor.)nextr1 ValueErrorjwtdecodegetall isinstancestrintfloatr fromtimestamp SAO_PAULO_TZr encrypt_datar r%r&nowcommitrrr) PyJWTErrorr Exceptionprint)selfrr-fgtsd_token_str decoded_tokenr7name exp_timestamp exp_datetimeencrypted_cookier*r+queryerrrstore1sH          zPanelController.storer7c Cst}z`z=|/}|d|f||jdkr&||d|ddn ||d|ddWd n1s;wYWnty]}z||d |d WYd }~nd }~wwW|n|wtd d dS)uRemove uma sessão pelo CNPJ.z&DELETE FROM companies WHERE cnpj = %s;ruSessão do CNPJ z removida com sucesso.ru'Nenhuma sessão encontrada para o CNPJ .warningNuErro ao remover sessão: dangerr;r<r=) r r%r&rProwcountrrRr)r)rTrr7r*r+r\rrrdelete_session]s$    zPanelController.delete_sessionc Cst}zfzD|jtd}|d|f|}Wdn1s!wY|s6tddiddWW|St|d}td |id WW|Styi}ztdt |id dWYd}~W|Sd}~ww|w) z>Busca um cookie pelo CNPJ, descriptografa e retorna como JSON.r"z-SELECT cookie FROM companies WHERE cnpj = %s;NerroruCNPJ não encontradoi)contentr?r-decrypted_cookie)rdrA) r r%rr&fetchonerr)r rRrI)rTr7r*r+ session_datarer\rrrget_decrypted_cookiers$       z$PanelController.get_decrypted_cookieN)r)__name__ __module__ __qualname__ staticmethodrrIrrr r!r,r]rrbrrhrrrrrs  ,rr!)"fastapirrfastapi.responsesrrfastapi.templatingrpsycopg2.extrasrtypingrr rDutilsr r bootstrapr r pytzrrZservices.encryptionrZ models.panelrrMrenvfiltersrr!globalsrrrrs"       n