o ?g)@sddlZddlZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z ddl mZddlmZddlmZddlZdd lmZed ZGd d d eZGd ddZdS)N)BytesIO) UploadFile) ClientError)settings) BaseSettings)Field) HTTPException)Optional adimsayarc@sleZdZUedddZeed<edddZeed<edddZeed<edddZ eed<Gdd d Z d S) Settings.AWS_S3_ENDPOINT_URL)env AWS_S3_BUCKETAWS_S3_BUCKET_NAMEAWS_ACCESS_KEY_IDAWS_SECRET_ACCESS_KEYc@seZdZdZdZdZdS)zSettings.Configz.envTignoreN)__name__ __module__ __qualname__env_filecase_sensitiveextrarr"/app/app/services/media_service.pyConfigsrN) rrrrr str__annotations__rrrrrrrrr s r c @seZdZddededededef ddZd ed eefd d Zd dZdeded efddZ defddZ ddeded efddZ de d dfddZ dS) MediaServiceNs3_endpoint_url bucket_name access_key secret_key region_namecCsV||_||_|p |||_td|d|d|jtjd||||jd|_dS)u{ AWS S3 클라이언트를 초기화합니다. :param s3_endpoint_url: S3 엔드포인트 URL (예: "https://s3.eu-central-1.amazonaws.com") :param bucket_name: AWS S3 버킷 이름 :param access_key: AWS_ACCESS_KEY_ID :param secret_key: AWS_SECRET_ACCESS_KEY :param region_name: (선택) 리전 이름. 필요 시 사용. z&MediaService initialized with bucket: z , endpoint: z , region: Zs3) endpoint_urlZaws_access_key_idZaws_secret_access_keyr#N) rr _extract_region_from_endpointr#loggerinfoboto3client s3_client)selfrr r!r"r#rrr__init__s  zMediaService.__init__r$returncCs2ddl}|sdSd}|||}|r|dSdS)u S3 엔드포인트 URL에서 리전 정보를 추출합니다. 예: "https://s3.eu-central-1.amazonaws.com" -> "eu-central-1" rN"s3[.-]([a-z0-9-]+)\.amazonaws\.com)researchgroup)r+r$r0aws_region_patternmatchrrrr%6s  z*MediaService._extract_region_from_endpointc Cs~z |jj|jdWdSty>z|jj|jdtd|jdWYdSty=}z td|d}~www)u (필요 시) 버킷이 존재하지 않으면 새로 생성합니다. AWS S3의 경우 사전에 버킷을 생성하거나, 퍼블릭 접근 정책을 설정하는 것이 좋습니다. )BucketzBucket 'z ' created.zError creating bucket: N)r*Z head_bucketr rZ create_bucketr&r'error)r+errr_create_bucketHs zMediaService._create_bucketfilefilenamec CsVz|j|j|j|td||WSty*}z td|d}~ww)u 업로드된 파일 객체(file)를 S3 버킷에 업로드합니다. :param file: UploadFile 객체 (FastAPI UploadFile) :param filename: S3에 저장될 파일명 :return: 파일 키 (예: filename) zFile uploaded: zError uploading file to S3: N)r*Zupload_fileobjr9r r&r'rr6)r+r9r:errrrr upload_fileXszMediaService.upload_file folder_namec Cs|ds |d7}z|jj|j|tdddtd|d|jdWd Sty@}ztd|d t |d }~ww) u 폴더 생성은 S3에서는 객체 키로 관리되므로, 빈 객체를 업로드하여 폴더를 생성하는 효과를 냅니다. /application/octet-streamr5KeyBody ContentTypezFolder 'z' created in bucket ''zError creating folder 'z': N) endswithr* put_objectr rr&r' Exceptionr6r)r+r=r7rrr create_foldergs  zMediaService.create_folderprofiler<folderc Csz;td|jd||ds|d}n|}|j}td||s/tdd}tj|d }|sBtdd }t |}||}td ||j }|j d t|} td | d | d kr{tdtd|j} | s|dvrd} n|dkrd} n |dkrd} nd} td| td|jd|d| |jj|j|t|| d} td| d|jvr|j} | sd dl} d}| ||j}|r|d} nd} d |jd!| d"|}td#|n9tj|j}|jd$|j }|jr#|jdkr#|j!d}||d|}n|d|}td%|td&||WSt"y`}ztjd't#|d(d)t$d*d+t#|d,d}~wt%y}ztjd-t#|d(d)t$d*d+t#|d,d}~ww).u% 파일을 AWS S3의 특정 폴더에 업로드하고, 업로드된 파일의 전체 URL을 반환합니다. 파일명은 고유한 UUID로 생성됩니다. 예: "https://adimsayar.s3.eu-central-1.amazonaws.com/profile/bb3c2811-a2c1-40f9-b82c-731c2db7ff45.jpg" zUploading image to S3: z to folder: r>zOriginal filename: z0Empty filename detected, generating generic namezunnamed_file.jpgr/z)No file extension detected, assuming .jpg.jpgzGenerated file key: rz File size: z byteszFile content is empty)rLz.jpegz image/jpegz.pngz image/pngz.gifz image/gifr@z!No content type provided, using: zUploading to bucket: z, key: z, content-type: rAzUpload successful: z amazonaws.comNr.zunknown-regionzhttps://z.s3.z.amazonaws.com/z"Using AWS S3 standard URL format: z://z)Using custom storage service URL format: zFinal uploaded image URL: zAWS S3 error uploading image: T)exc_infoizFailed to upload image: ) status_codedetailz"Unexpected error uploading image: )&r&r'r:rFdebugwarningospathsplitextloweruuiduuid4r9readseeklenr6 ValueError content_typer r*rGrrr#r0r1r2urllibparseurlparseschemenetlocrstriprrrrH)r+r<rK full_folderoriginal_filename extensionunique_filenamefile_keyZ file_contentcontent_lengthr\responseZregionr0r3r4full_url parsed_urlZ base_endpoint base_pathr7rrr upload_imageys          zMediaService.upload_imagepolicyc Csft|}z|jj|j|dtd|jdWdSty2}z tdt |d}~ww)uv S3 버킷 정책을 업데이트합니다. :param policy: 정책 dict (AWS S3 정책 형식) )r5Policyz"Bucket policy updated for bucket 'rEzError updating bucket policy: N) jsondumpsr*Zput_bucket_policyr r&r'rHr6r)r+rnZ policy_jsonr7rrrupdate_bucket_policys z!MediaService.update_bucket_policy)N)rJ)rrrrr,r r%r8rr<rIrmdictrrrrrrrs ur)iorVrRloggingrprfastapirr(Zbotocore.exceptionsrapp.core.configrpydantic_settingsrpydanticrr urllib.parser]typingr getLoggerr&r rrrrrs$