52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
import boto3
|
|
from botocore.exceptions import ClientError
|
|
from typing import Optional
|
|
import uuid
|
|
from app.core.config import settings
|
|
|
|
|
|
class S3Service:
|
|
def __init__(self):
|
|
self.s3_client = boto3.client(
|
|
's3',
|
|
endpoint_url=settings.s3_endpoint_url,
|
|
aws_access_key_id=settings.s3_access_key_id,
|
|
aws_secret_access_key=settings.s3_secret_access_key,
|
|
region_name=settings.s3_region
|
|
)
|
|
self.bucket_name = settings.s3_bucket_name
|
|
|
|
async def upload_file(self, file_content: bytes, file_name: str, content_type: str) -> Optional[str]:
|
|
try:
|
|
file_key = f"{uuid.uuid4()}_{file_name}"
|
|
|
|
self.s3_client.put_object(
|
|
Bucket=self.bucket_name,
|
|
Key=file_key,
|
|
Body=file_content,
|
|
ContentType=content_type
|
|
)
|
|
|
|
file_url = f"{settings.s3_endpoint_url}/{self.bucket_name}/{file_key}"
|
|
return file_url
|
|
|
|
except ClientError as e:
|
|
print(f"Error uploading file to S3: {e}")
|
|
return None
|
|
|
|
async def delete_file(self, file_url: str) -> bool:
|
|
try:
|
|
file_key = file_url.split('/')[-1]
|
|
|
|
self.s3_client.delete_object(
|
|
Bucket=self.bucket_name,
|
|
Key=file_key
|
|
)
|
|
return True
|
|
|
|
except ClientError as e:
|
|
print(f"Error deleting file from S3: {e}")
|
|
return False
|
|
|
|
|
|
s3_service = S3Service() |