o gZ/ @sFdZddlZddlZddlZddlZddlmZddlmZddlm Z m Z ddl Z ddl Z ddl Z ddlZddlZddlmZddlmZmZmZmZmZddlmZmZmZmZmZdd lmZdd l m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'ddl(m)Z)eddgdddiigdZ*e*+dde,fddZ-e*+dde,fddZ.e*/dedededfde,d e e,d!e e0fd"d#Z1e*+d$ dAde,fd%d&Z2e*+d'd(e,fd)d*Z3e*+d+d,e,fd-d.Z4e*+d/d0d1Z5e*+d2de,fd3d4Z6e*+d5dAde,fd6d7Z7e*j8d8d9d:gd;edde,fd?d@Z9dS)Bz%Fast API module for general utilitiesN)datetime)BytesIO)IOOptional) ClientError) APIRouterBody HTTPExceptionQuery UploadFile) FileResponse JSONResponseRedirectResponseResponseStreamingResponse)get_brand_from_url)PublicComparables) validate_url)fetch_brand_details)get_google_fonts_list)S3BucketStoragez /utilities utilities descriptionz Not found)prefixtags responses dependenciesz /validate_urlurlcCs"t|}|r ddiStddd)zValidate the input urlmessagez URL is validrzURL is not valid status_codedetail)rr )rvalidate_responser$1/var/www/html/XCapMarket/controllers/utilities.pyvalidate)s  r&z/validate/tickertickercCs t|}|stdddSdS)zValidate the input tickerrzTicker is not validr )rget_ticker_datar )r' ticker_datar$r$r%validate_ticker3s  r+z /upload_file/. s3_file_pathsource_file_paths3_filec sz&tjd}|dd}t|d}|j|||d}|r%tdd|idWSWdStyP}z|jd d d krBtd d |d|td d|d|d}~wt yg}z tddd|id|d}~wty~}z tddd|id|d}~ww)a Uploads a file to the specified location in the S3 bucket and returns a JSON response indicating the result. Args: s3_file_path (str): The path to upload the file to in the S3 bucket. source_file_path (str): The path of the file to upload from the local system. Returns: JSONResponse: A JSON response indicating the success or failure of the upload operation. xcap_s3_storager. bucket_name)r,r-r.r(rr!contentErrorCode NoSuchKeyException occurred: r Error: N) osenvirongetrupload_file_to_s3_bucketr rresponser ValueError) r,r-r.r/s3_storage_objresulteveher$r$r%r?=s:    r?z /get_file/c sz&tjd}t|d}|j|d}t|d}|r%tdd|idWSWdStyP}z|j dd d krBt d d |d |t d d|d |d}~wt yg}z t d dd|id |d}~ww)a3 Retrieves a file from the specified location in the S3 bucket and returns its content as base64 encoded JSON. Args: s3_file_path (str): The path of the file to retrieve from the S3 bucket. Returns: JSONResponse: A JSON response containing the base64 encoded file content. r/r1)r,zutf-8r(rr3r5r6r7r;r9r r:N) r<r=rget_file_from_s3_bucketbase64 b64encodedecoder rr@r rA)r,r/rB file_contentrDrEr$r$r%rGvs,    rGz/search_files/keywordc sztjd}t|d}|j|d}tdd|idWStyC}z|jddd kr5td d |d |td d |d |d}~wtyZ}z td dd |id |d}~ww)a Search for files containing the specified keyword in the S3 bucket and return a JSON response. Args: keyword (str): The keyword to search for in file names. Returns: JSONResponse: A JSON response containing the list of matching file paths. r/r1)rLr(matching_filesr3r5r6r7r;r9r r:Nr) r<r=rsearch_files_in_bucketr rr@r rA)rLr/rBrMrDrEr$r$r%search_files_in_s3_buckets&    rOzscheduled_task_status/{task_id}task_idcs<ddlm}||}|r||j|jdS||jdS)z,Check the status of the research slide task.r)celery)rPstatusrC)rPrR)tasks.celery_workerrQ AsyncResultreadyrRrC)rPrQtaskr$r$r%research_slide_statuss    rWz /fonts/googlecstIdH}td|dS)z Retrieves a list of all available Google Font families. Caches the response for 7 days to minimize API calls. Returns: JSONResponse: A JSON response containing just the font family names. Nr(r3)rr ) fonts_datar$r$r%get_google_fontss rYz /fetch_brandcs t|IdH}tdd|idS)z Fetch brand details for a given URL using the Brandfetch API. Args: url (str): The URL of the company to fetch brand details for. Returns: JSONResponse: The brand details retrieved from the Brandfetch API. Nr(datar3)rr )rrZr$r$r% fetch_brands  r[z/fetch-and-previewc s|s tdddzTt4IdH7}||4IdH}|t|IdH}WdIdHn 1IdHs:wYWdIdHn 1IdHsOwYt|dddidWStjyx}zt t |dd WYd}~Sd}~ww) Nr;zURL is requiredr application/pdfContent-Dispositionzinline; filename=invoice.pdf) media_typeheadersr8)r4r!) r aiohttp ClientSessionr>raise_for_statusrreadrrrstr)rsessionr@ pdf_contentrDr$r$r% download_pdfs& *(rgz /sign_blobGETHEAD)methodszUnsigned URL or S3 object key)r unsigned_urlc sLzztjd}ddlm}|dr||}|jd}|dr,||}|jd}n|}|s6tddd t|d }| |\}}|d urNt d |id WS|r[|dr[t |dWS|rk|drkt |ddidWSt |dd| ddidWSt y}z tdt|d |d }~wty} z tdd| d | d } ~ ww)a6 Generates a signed URL for an S3 object from an unsigned URL or S3 file key, and returns a response based on the file type. For image files (e.g. .jpg, .png, etc.) the function will return a redirect to the signed URL so that it can be used directly in an tag. For other file types, it returns the signed URL inside a JSON response. Query Parameters: unsigned_url (str): The unsigned URL or S3 object key. Returns: Either a RedirectResponse (for image files) or a JSONResponse with the signed URL. r/r)urlparsehttp/zs3://r;zInvalid unsigned URL provided.r r1N signed_url)r4zimage/)rr\r]zinline; filename=document.pdf)rr_zinline; filename=document.r8zUnexpected error: )r<r= urllib.parserl startswithpathlstripr rgenerate_signed_urlr rsplitrArd Exception) rkr/rlparseds3_key s3_storagero mime_typerDexr$r$r%generate_signed_url_blobsL       r})N):__doc__rHjsonloggingr<riortypingrrr`backoffboto3redisrequestsbotocore.exceptionsrfastapirrr r r fastapi.responsesr r rrr(services.company_profile.logo_standaloner%services.stock_info.public_comparablerutilsrutils.brand.brand_fetchrutils.google_fonts_servicerutils.s3_storagerrr>rdr&r+postdictr?rGrOrWrYr[rg api_router}r$r$r$r%s             8 - )