o hA @sddlZddlZddlZddlZddlZddlZddlmZmZm Z ddl m Z m Z ddl mZddZddZd d Zd ed ed efddZddZddZddZddZddZdddZddZdS) N)FastMail MessageSchemaConnectionConfig) HTTPExceptionstatus) JSONResponsecCsdtjtjtjddS)N)k)joinrandomchoicesstringascii_uppercasedigitsrr6/var/www/html/storm_article_api/dependencies/helper.pygenerate_random_text srcCstdd|}|S)Nz [^a-zA-Z0-9]_)resub) input_stringsanitized_stringrrrreplace_special_characterssrcCsdd|dgD}|S)NcSsg|]}|dqS)messager).0errorrrr sz%get_error_message..errors)get)responsemessagesrrrget_error_messagesr" recipientsubjectbodyc sttdtdtddtddddd}t||g|d d }t|}z ||IdHWdStyE}ztd t|d d}~ww) NSES_ACCESS_KEY_IDSES_SECRET_ACCESS_KEYSES_SENDER_EMAILi SES_AWS_HOSTFT) MAIL_USERNAME MAIL_PASSWORD MAIL_FROM MAIL_PORT MAIL_SERVER MAIL_STARTTLS MAIL_SSL_TLSUSE_CREDENTIALShtml)r$ recipientsr%subtypei) status_codedetail) rosgetenvrr send_message Exceptionrstr)r#r$r%confrfmerrr send_emails0 r?cCs6t| }t|WdS1swYdS)aj Reads a JSON file and returns its content as a Python dictionary or list, depending on the JSON structure. Args: file_path (str): The path to the JSON file to be read. Returns: dict or list: The content of the JSON file. The type depends on the structure of the JSON file (object or array at the root). N)openjsonload file_pathfrrrread_json_file0s $rFcCs4t| }|WdS1swYdS)N)r@readrCrrr read_txt_file?s $rHcCstd}|d|}|S)Nz]:\s+"(.*?)"\s+httpz]: http)rcompiler)textregexrrrparseDs  rLcCsR|durdSi}|dD]\}}||d|d|d|dd||<q|S)Nurl_to_unified_index url_to_infotitlesnippets)urlrOrP)items)search_results citation_dictrQindexrrr+_construct_citation_dict_from_search_resultIs rVcCstdd|S)Nz\s+ )rrstrip)rrrrnormalize_spacesUsrYcCs|stj}t|||ddS)N)rrr5content)rHTTP_400_BAD_REQUESTr)rr5rrr errorResponseXsr^cCsttjtj||ddS)N)rrdatar[)rr HTTP_200_OK)rr_rrrsuccessResponsecsra)rZ)rAr7r rrbase64 fastapi_mailrrrfastapirrfastapi.responsesrrrr"r;r?rFrHrLrVrYr^rarrrrs(