jPddlZddlmZmZddlmZmZmZddlmZeje Z GddZ e Z y)N)datetime timedelta)BlobSasPermissionsBlobServiceClientgenerate_blob_sas)settingsc ^eZdZdZdZdedededefdZdededefdZ d deded e defd Z y ) AzureStorageServicec tjtj|_d|_y#t $r+}tjd|dd|_Yd}~yd}~wwxYw)Nzknowledge-basez0Failed to initialize Azure Blob Storage client: Texc_info) rfrom_connection_stringrAZURE_STORAGE_CONNECTION_STRINGblob_service_clientcontainer_name Exceptionloggererror)selfes 9/var/www/html/strategist-ai/app/services/infra/storage.py__init__zAzureStorageService.__init__ s] ,'8'O'OPXPxPx'yD $"2D  , LLKA3OZ^L _'+D $ $ ,s/2 A&!A!!A&c4|jsy |jj|j}|js%|jj |jy#t $r$}t jd|dYd}~yd}~wwxYw)NFTz'Error ensuring Azure container exists: r )rget_container_clientrexistscreate_containerrrr)rcontainer_clientrs r_ensure_containerz%AzureStorageService._ensure_containers'' #77LLTM`M`a #**,((99$:M:MN  LLB1#FQUL V sAA** B3BB tenant_id file_name content_bytesreturncB|jsy d|d|}|jj|j|}|j |dt j d|d|y#t$r$}t jd |d Yd }~yd }~wwxYw) zT Uploads a file to Azure Blob Storage under a tenant-specific path. Ftenant//files/ containerblobT) overwritezSuccessfully uploaded z to Azure for tenant z Failed to upload file to Azure: r N) rrget_blob_clientr upload_blobrinforr)rrr r! blob_path blob_clientrs r upload_filezAzureStorageService.upload_files%%' !)GI;?I22BBTM`M`gpBqK  # #MT # B KK0 ;PQZP[\ ]  LL;A3?$L O sAA11 B:BBcT|jsy d|d|}|jj|j|}|jr+|j t j d|d|y#t$r$}t jd|d Yd }~yd }~wwxYw) z9 Deletes a file from Azure Blob Storage. Fr$r%r&zSuccessfully deleted z from Azure for tenant Tz"Failed to delete file from Azure: r N) rr*rr delete_blobrr,rr)rrr r-r.rs r delete_filezAzureStorageService.delete_file2s'' !)GI;?I22BBTM`M`gpBqK!!#'') 3I;>UV_U`ab  LL=aSADL Q sA*A:: B'B""B'hoursc B|jsy d|d|}|jj|j|}|jstj d|yt |jj|j||jjjtdtjt|z }|jd |S#t$r'}tj!d |d |d Yd}~yd}~wwxYw)aA temporary read-only link to a stored file, or '' if it cannot be made. The container is private, so visitors cannot be handed a plain blob URL. A SAS token grants read access to this one blob and nothing else, and stops working once it expires. r$r%r&zNo stored file at T)read)r3) account_namer blob_name account_key permissionexpiry?z$Could not build a download link for z: r N)rr*rrrwarningrr7 credentialr9rrutcnowrurlrr)rrr r3r-r.tokenrs r download_urlz AzureStorageService.download_urlEs'' !)GI;?I22BB--IC?K%%'!3I;?@%!55BB#22# 44??KK-48(95+AA E"oo&aw/ /  LL? {"QCP[_L ` sAC.'BC.. D7DDN)) __name__ __module__ __qualname__rrstrbytesboolr/r2intrBrr r si, SSSW&SST&cc#srLr )loggingrrazure.storage.blobrrrapp.core.configr getLoggerrDrr storage_servicerKrLrrRs=(WW$   8 $WWv&'rL