o i,@sddlmZmZmZddlmZddlmZddlm Z ddl m Z ddl Z ddl m Z mZddlmZe d ZGd d d ZdS) ) HTTPExceptionDependsHeader) DictCursor)get_db_connection)timezone)datetimeN)CompanyResponse)EncryptionServicezAmerica/Sao_Pauloc@s6eZdZdefddZdefddZdefddZd S) CompanyControllercookiec Csztdd|dDd}|stdtj|ddid}|d}|d }|d }t|||gs7td t|ts@td t|t t fsKtd t j |t d}t|}t} z:| } d} | | ||||t t f| Wdn1s~wYtdd||d|ddW| WS| wttjfy} ztdt| dd} ~ wty} z td| tdddd} ~ ww)Ncss(|]}|dr|ddVqdS)z fgtsd_token==N) startswithsplit).0sr/app/controllers/company.py s&z*CompanyController.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; successzEmpresa registrada com sucesso.createdrnameaction expires_onstatusmessagedatai status_codedetailzErro inesperado no registro: iz$Ocorreu um erro interno no servidor.)nextr ValueErrorjwtdecodegetall isinstancestrintfloatr fromtimestamp SAO_PAULO_TZr encrypt_datarcursorexecutenowcommitr close PyJWTErrorr Exceptionprint) selfr fgtsd_token_str decoded_tokenrr exp_timestamp exp_datetimeencrypted_cookieconncurqueryerrrstoresR          zCompanyController.storercCst}zC|jtd}|d|f|}Wdn1s wY|s.W|dSt|}t|d|d<t di|W|S|w)Ncursor_factoryz(SELECT * FROM companies WHERE cnpj = %s;r r) rr7rr8fetchoner;dictr decrypt_datar r?rrErFcompanyrrrshowFs  zCompanyController.showc Cst}zK|jtd6}|d|f|}|stddd|tdd|d|d d |d d d WdW|S1sCwYW|dS|w)NrJz2DELETE FROM companies WHERE cnpj = %s RETURNING *;iu%Empresa não encontrada para remover.r'rzEmpresa removida com sucesso.rr deletedr"rr#) rr7rr8rLrr:r r;rOrrrdeleteVs*  zCompanyController.deleteN)__name__ __module__ __qualname__r1rIrQrSrrrrr s4r )fastapirrrpsycopg2.extrasrutilsrpytzrrr,models.companyr r services.encryptionr r5r rrrrs