o i@s>ddlZddlZddlmZmZddlmZGdddZdS)N)datetimetimezone) SimpleCookiec@sFeZdZddZdedefddZdefddZd ed efd d Zd S) FgtsClientcCs.ddddddddd d d d d dd|_d|_dS)Nzfgtsdigital.sistema.gov.brz"Linux"zpt-BR,pt;q=0.9z!application/json, text/plain, */*z("Chromium";v="143", "Not A(Brand";v="24"zeMozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36z?0z same-originZcorsemptyz2https://fgtsdigital.sistema.gov.br/portal/servicoszgzip, deflate, brzu=1, iz keep-alive)HostzSec-Ch-Ua-PlatformzAccept-LanguageAcceptz Sec-Ch-Uaz User-AgentzSec-Ch-Ua-MobilezSec-Fetch-SitezSec-Fetch-ModezSec-Fetch-DestZRefererzAccept-EncodingZPriority Connectionz6https://fgtsdigital.sistema.gov.br/proxy-api/v2/perfil)headersbase_url)selfr /app/services/fgts_client.py__init__s  zFgtsClient.__init__ cookie_strreturncCs"t}||dd|DS)uEHelper para converter string de cookie (k=v; k2=v2) para dicionário.cSsi|]\}}||jqSr )value).0kvr r r sz8FgtsClient._parse_cookie_str_to_dict..)rloaditems)r rcookier r r_parse_cookie_str_to_dicts z$FgtsClient._parse_cookie_str_to_dictcCsddd|DS)z;Converte o CookieJar do requests de volta para string pura.z; cSsg|] }|jd|jqS)=)namer)rrr r r #sz2FgtsClient._dict_to_cookie_str..)join)r Z cookie_jarr r r_dict_to_cookie_str!szFgtsClient._dict_to_cookie_strcompany_cookie_decrypted customer_cnpjc Cs||}d|d}zWtj|j|j||ddd}||jd}|s4z |}|d}WnY|s:tdt j |d did }|d } | sNtd t j | t jd } ||j} | | dWStjyw} z td| | d} ~ wty} z td| | d} ~ ww)u Executa o GET /perfil emulando a troca de empresa para cliente. Retorna o novo cookie completo e a data de expiração. 2)Z tipoPerfilZniPerfilF)r paramscookiesverifytimeoutZ fgtsd_tokenZ accessTokenu<Token JWT (fgtsd_token) não encontrado na resposta do FGTS.verify_signature)optionsexpu#Campo 'exp' não encontrado no JWT.)tz)r expires_onuErro de conexão com FGTS: Nz#Erro ao processar troca de perfil: )rrequestsgetr r Zraise_for_statusr%json Exceptionjwtdecoder fromtimestamprutcrZRequestExceptionprint) r r r!Z cookies_dictr$responseZnew_jwtZ json_respdecoded exp_timestampr,Zfull_new_cookieer r r swap_context%sR    zFgtsClient.swap_contextN) __name__ __module__ __qualname__rstrdictrrr:r r r rrs r)r-r1rrZ http.cookiesrrr r r rs