k)iF dZddlZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZmZmZmZmZmZmZdd lm Z dd l!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4ejje6Z7de de8de#fdZ9de dede8dee#fdZ:de de;de"fdZffdZ? d0de de8defd Z@de8ddfd!ZAd"eBde8ddfd#ZCd"eBddfd$ZDdeddfd%ZEde d&ede8de#fd'ZFde d(eede8dee#fd)ZGde de8dee"fd*ZHde de;de;dee"fd+ZIde de8de fd,ZJd-e ddfd.ZKde d-e deBfd/ZLy)1z File services module for handling file operations. This module provides services for file upload, download, listing, and deletion operations with support for both local filesystem and CDN storage. N)DictListOptionalTupleAny)IntegrityError)Session)select)status)Response)StreamingResponse)get_active_cdn)get_approximate_filesizeremove_file_singleupload_file_multipleupload_file_singledownload_file_from_s3download_file_from_system_pathget_image_resolution)File) FileSchemaFileResponseSchema) APIException) DBSortTypes FileTypes)MAX_UPLOAD_LIMITMAX_UPLOAD_SIZE_MBALLOWED_FILE_CONTENT_TYPESMAX_UPLOAD_RESOLUTION$THUMBNAIL_ALLOWED_FILE_CONTENT_TYPESTHUMBNAIL_MAX_FILE_SIZE_BYTES)testINVOICE_UNIQUE_IDRECEIPT_UNIQUE_IDCONTRACT_UNIQUE_IDdb file_typereturncKt|||jj}t|||dk(r t ||t ||||j d}t||||d{}|S7w)a Upload a single file and create database entry. Args: db: Database session. file: FastAPI UploadFile object. file_type: Type of file being uploaded. created_by: User who is uploading the file. merchant: Merchant context for file organization. Returns: FileResponseSchema with file details. Raises: APIException: If validation fails or upload errors occur. thumbnail/)r& file_datar' created_byN)_validate_file_type_and_contentfileread_validate_file_content _validate_thumbnail_requirementsr merchant_id_create_file_database_entry)r&r/r'r-merchant file_contentuploaded_file_datanew_files Z/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/src/apps/files/file_services.pyupload_single_filer:2s.$D)499>>#L<3K(|<+ D,8#7#7"8 :1 $ H OsA1A<3A:4A<filescKt|tkDr)ttitj dtdt |t|||jd}t||||d{}|S7w)a Upload multiple files and create database entries. Args: db: Database session. files: List of FastAPI UploadFile objects. file_type: Type of files being uploaded. created_by: User who is uploading the files. merchant: Merchant context for file organization. Returns: List of FileResponseSchema with file details. Raises: APIException: If validation fails or upload errors occur. z A maximum of z files can be uploaded at oncemoduleerror status_codemessager+)r& files_datar'r-N) lenrr__name__r HTTP_400_BAD_REQUEST_validate_multiple_file_typesrr3&_create_multiple_file_database_entries)r&r;r'r-r5uploaded_files_data new_filess r9upload_multiple_filesrJ_s. 5z$$33#$4#55ST   "%(. Eh**+1-= & I sA3A>5A<6A>file_idcKttjtj|k(}|j |}|j }|s!t titjdtj|Sw)a& Retrieve a file by its ID. Args: db: Database session. file_id: ID of the file to retrieve. created_by: User requesting the file. Returns: FileSchema with file details. Raises: APIException: If file not found. No such file foundr=) r rwhereidexecutescalar_one_or_nonerrDr HTTP_404_NOT_FOUNDrmodel_validate)r&rKstmtresultfile_objs r9get_file_by_idrWs{ $<   7 DZZ F((*H 11(     $ $X ..sB B idsskiplimitclK|rt|||d{St||||d{S77w)a List files for a user with optional filtering by IDs. Args: db: Database session. ids: Comma-separated string of file IDs to filter by. skip: Number of records to skip for pagination. limit: Maximum number of records to return. created_by: User requesting the files. Returns: List of FileSchema objects. Raises: APIException: If validation fails or files not found. N)_get_files_by_ids_get_paginated_files)r&rXrYrZr-s r9list_user_filesr^s:. &r3 ;;;)"dE:FFF<Fs404244cZKttjtj|k(tj|jk(}|j |}|j }|s!ttitjd |j||jt|jddiS#t$rQ}|j!t"j%d|d|ttitj&dd }~wwxYww) aE Delete a file by its ID. Args: db: Database session. file_id: ID of the file to delete. created_by: User requesting the deletion. Returns: Dictionary indicating successful deletion. Raises: APIException: If file not found or deletion fails. rMr=pathdeletedTzError deleting file z: zFailed to delete fileN)r rrNrO created_by_idrPrQrrDr rRdeletecommitr upload_path Exceptionrollbackloggerr?HTTP_500_INTERNAL_SERVER_ERROR)r&rKr-rTrUrVes r9delete_file_by_idrls $<   7 jmm+ DZZ F((*H 11(     (   4 454     +G9Bqc:;==+    s+BD+:C D+ D(A D##D((D+rac|s!ttitjdt |||}t ||t ||}t||jS)a; Download a file by its path. Args: db: Database session. path: Path to the file. current_user: Currently authenticated user (optional). Returns: Response with file content. Raises: APIException: If file not found or access denied. zPath is requiredr=)content media_type) rrDr rE_find_file_by_path_validate_file_access_download_file_contentr mime)r&ra current_userrVr6s r9download_file_by_pathrus]& 33&   ""dL9H(L1)"h7L LX]] CCc |jjdd}|tvr:ttit j d|ddjty)z2Validate file type and basic content requirements.r+zCannot upload 'z#' files. Allowed content types are , r=N content_typesplitrrrDr rEjoin)r/r'r{s r9r.r.#sf$$**3/2L5533%l^3VW[W`W`a{W|V}~   6rvr6ct|dk(r!ttitjdt|t dzkDr)ttitjdt dy)z2Validate file content size and basic requirements.rz*The file you are trying to upload is emptyr=i@B#File exceeds maximum size limit of mbN)rCrrDr rEr)r6r's r9r1r1/sr <A33@    <.89339:L9MRP   :rvc $|jtvr/ttitj d|jdt |tdzkDr)ttitj dtd t|}|rU|dtdkDs|dtdkDr6ttitj d tdd tdd y y #t$r"ttitj d wxYw)z3Validate specific requirements for thumbnail files.zInvalid file type 'z '. Only JPG and PNG are allowed.r=irzKB.rrxz-Image resolution exceeds the maximum allowed xz pixels.z!Failed to parse image resolution.N) r{r rrDr rErCr!rr ValueError)r/r6 resolutions r9r2r2Bs8  DD33)$*;*;)<<\]    <9D@A339:W9XX[\    ),7  qM1!4 4 qM1!4 4"77GH]^_H`Gaabcxyzc{b|}EF   5   337    s A!C$$+Dc |D]b}|jjdd}|tvs*ttit j d|ddjty)z-Validate file types for multiple file upload.r+rxzCannot upload z$ files. Allowed file extensions are ryr=Nrz)r;r/ extensions r9rFrFisp%%++C03 6 6"77( 3WX\XaXab|X}W~  rvr,c nK t|}t|d|dd|dz|d||j|d|r |jnd }|j||j |j |t j|}|d |_|S#t$r2|jttitjd t$rN}|jt j#d |ttitj$d d}~wwxYww)z+Create a single file entry in the database.r& file_name original_name.formatr{raNnamerrsr'rcracdn_idfilesize#Attempting to create duplicate filer=zError creating file entry: zFailed to create file entry)rrrOaddrerefreshrrSrrrhrrDr rErgrir?rj)r&r,r'r- active_cdnr8rUrks r9r4r4vs'% #r* ;'#O49X;N:O7PP>*$--6"$.:==D  x  8#228<#J/   339      21#67==1    s+D5BB"!D5"AD2$A D--D22D5rBc K t|}g}|D]R}t|d|dd|dz|d||j|d|r |jnd }|j|T|j ||j g}|D]b} |j | tj| } t|j| j|| _ |j| d|S#t$r2|jtt it"j$d t&$rN} |jt(j+d | tt it"j,d d} ~ wwxYww)z-Create multiple file entries in the database.rrrrrr{raNrrr=z&Error creating multiple file entries: zFailed to create file entries)rrrOappendadd_allrerrrSrrfrarrrhrrDr rErgrir?rj) r&rBr'r-r files_listr,r8 response_listrVrUrks r9rGrGs- #r*  # (I{+'8Qy?R>S;TT~.#(mmv&(2z}}H   h ' ( :  " )H JJx '66x@F6))*8==/:JFO   (  )   339      =aSAB==3    s+FC0C54F5AF7A FFFczK |jdDcgc] }t|}}t|dkDr!ttit j dttjtjj|tj|jk(}|j|}|jj}|Dcgc]}t!j"|c}Scc}wcc}w#t$$r"ttit j dwxYww)z#Get files filtered by specific IDs., z.A maximum of 10 files can be requested at oncer=zInvalid file ID format)r|intrCrrDr rEr rrNrOin_rcrPscalarsallrrSr) r&rXr-fidfile_idsrTrUr;rVs r9r\r\s (+ #7CH77 x=2 "77H  d|!! GGKK !   *-- / D! $$&DIJ ))(3JJ!8 K  33,    s9D;D DB;D $DD D; D +D88D;cKttjtj|jk(j tj jj|j|}|j|}|jj}|Dcgc]}tj|c}Scc}ww)zGet paginated files for a user.)r rrNrcrOorder_by created_atdescoffsetrZrPrrrrS)r&rYrZr-rTrUr;rVs r9r]r]s $<   jmm+ ht##%&vvd|EE%L ZZ F NN  "E@E FHJ % %h / FF FsB6C8CCc t|}|}|rB|j|j|jfD]}|s||vs |j |d}|j dd}t t ttfDcgc]}t||c}}|rS|s!ttitjdttj!tj"|k(} n/ttj!tj|k(} |j%| } | j'} | s!ttitj(d| Scc}w)z-Find file by path with proper access control.rr+/Authentication required to access this resourcer=rM)rhostrootrareplaceanyr#r$r%r"rrDr HTTP_403_FORBIDDENr rrN storage_keyrPrQrR) r&rartrsanitized_pathprefixfile_storage_keypatternis_private_filerTrUrVs r9rprpsL2&JN! I DF&N2!/!7!7!C D &--c26)+<>PQ  w'O "55I   d|!!$"2"26F"FGd|!!$))~"=> ZZ F((*H 11(    O;s=E!rVcd|jr$|s!ttitjdyy)z!Validate user access to the file.rr=N)rrrDr r)rVrts r9rqrq2s4L11E   %1rvcPt|}|rB|jjdk(r%t|j|j }n$t |j|j }|rt|dkr!ttitjd|S)z#Download file content from storage.rs3r`rxz4The resource you are trying to access is unavailabler=) rlabellowerrrfrarrCrrDr rR)r&rVrr6s r9rrrr=s2&Jj&&,,.$6,**+HMM?; 6**+HMM?;  3|,q011J    rv)N)M__doc__loggingostypingrrrrrsqlalchemy.excrsqlalchemy.ormr sqlalchemyr starletter starlette.responsesr fastapi.responsesr src.apps.base.servicesrsrc.apps.files.helper.iorrrrrrrsrc.apps.files.models.filer"src.apps.files.schemas.file_commonrrsrc.core.exceptionsrsrc.core.utils.enumsrrsrc.core.utils.constantsrrrrr r!src.apps.base.utils.regexpr"r#r$r% getLoggerrDristrr:rJrrWr^boolrlrur.bytesr1r2rFr4rGr\r]rprqrrrvr9rs 33)"(/1,M,7   8 $***  *Z,, ,,   ,^/W/s/z/@GG G G  G  * G:0 0 #0 d3PT9o0 l D D  D DJ  S  T    3 4 &$ $ 4$ N  $ , , , ,  , ^4 4 T 4 4   4 n  c $zBR : G7 G# Gc GRVWaRb G*7*#**Z D 4 w$5rv