almost finish migrate new circumstances
This commit is contained in:
222
rog/services/s3_service.py
Normal file
222
rog/services/s3_service.py
Normal file
@ -0,0 +1,222 @@
|
||||
"""
|
||||
S3 Service for managing uploads and standard images
|
||||
"""
|
||||
import boto3
|
||||
import uuid
|
||||
import base64
|
||||
from datetime import datetime
|
||||
from django.conf import settings
|
||||
from django.core.files.uploadedfile import InMemoryUploadedFile
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class S3Service:
|
||||
"""AWS S3 service for handling image uploads and management"""
|
||||
|
||||
def __init__(self):
|
||||
self.s3_client = boto3.client(
|
||||
's3',
|
||||
aws_access_key_id=settings.AWS_ACCESS_KEY_ID,
|
||||
aws_secret_access_key=settings.AWS_SECRET_ACCESS_KEY,
|
||||
region_name=settings.AWS_S3_REGION_NAME
|
||||
)
|
||||
self.bucket_name = settings.AWS_STORAGE_BUCKET_NAME
|
||||
self.custom_domain = settings.AWS_S3_CUSTOM_DOMAIN
|
||||
|
||||
def upload_checkin_image(self, image_file, event_code, team_code, cp_number, is_goal=False):
|
||||
"""
|
||||
チェックイン画像またはゴール画像をS3にアップロード
|
||||
|
||||
Args:
|
||||
image_file: アップロードする画像ファイル
|
||||
event_code: イベントコード
|
||||
team_code: チームコード
|
||||
cp_number: チェックポイント番号
|
||||
is_goal: ゴール画像かどうか（デフォルト: False）
|
||||
"""
|
||||
try:
|
||||
# ファイル名を生成（UUID + タイムスタンプ）
|
||||
file_extension = image_file.name.split('.')[-1] if '.' in image_file.name else 'jpg'
|
||||
filename = f"{uuid.uuid4()}-{datetime.now().strftime('%Y-%m-%dT%H-%M-%S')}.{file_extension}"
|
||||
|
||||
# S3キーを生成（イベント/チーム/ファイル名）
|
||||
# ゴール画像の場合は専用フォルダに保存
|
||||
if is_goal:
|
||||
s3_key = f"{event_code}/goals/{team_code}/{filename}"
|
||||
else:
|
||||
s3_key = f"{event_code}/{team_code}/{filename}"
|
||||
|
||||
# メタデータをBase64エンコードして設定（S3メタデータはASCIIのみ対応）
|
||||
metadata = {
|
||||
'event_b64': base64.b64encode(event_code.encode('utf-8')).decode('ascii'),
|
||||
'team_b64': base64.b64encode(team_code.encode('utf-8')).decode('ascii'),
|
||||
'cp_number': str(cp_number),
|
||||
'uploaded_at': datetime.now().strftime('%Y-%m-%dT%H-%M-%S'),
|
||||
'image_type': 'goal' if is_goal else 'checkin'
|
||||
}
|
||||
|
||||
# S3にアップロード
|
||||
self.s3_client.upload_fileobj(
|
||||
image_file,
|
||||
self.bucket_name,
|
||||
s3_key,
|
||||
ExtraArgs={
|
||||
'ContentType': f'image/{file_extension}',
|
||||
'Metadata': metadata
|
||||
}
|
||||
)
|
||||
|
||||
# S3 URLを生成
|
||||
s3_url = f"https://{self.bucket_name}.s3.{settings.AWS_S3_REGION_NAME}.amazonaws.com/{s3_key}"
|
||||
|
||||
logger.info(f"{'Goal' if is_goal else 'Checkin'} image uploaded to S3: {s3_url}")
|
||||
return s3_url
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to upload image to S3: {e}")
|
||||
raise
|
||||
|
||||
def upload_standard_image(self, image_file, event_code, image_type):
|
||||
"""
|
||||
規定画像をS3にアップロード
|
||||
"""
|
||||
try:
|
||||
# ファイル拡張子を取得
|
||||
file_extension = image_file.name.split('.')[-1] if '.' in image_file.name else 'jpg'
|
||||
|
||||
# S3キーを生成（イベント/standards/タイプ.拡張子）
|
||||
s3_key = f"{event_code}/standards/{image_type}.{file_extension}"
|
||||
|
||||
# メタデータをBase64エンコードして設定
|
||||
metadata = {
|
||||
'event_b64': base64.b64encode(event_code.encode('utf-8')).decode('ascii'),
|
||||
'image_type': image_type,
|
||||
'uploaded_at': datetime.now().strftime('%Y-%m-%dT%H-%M-%S')
|
||||
}
|
||||
|
||||
# S3にアップロード
|
||||
self.s3_client.upload_fileobj(
|
||||
image_file,
|
||||
self.bucket_name,
|
||||
s3_key,
|
||||
ExtraArgs={
|
||||
'ContentType': f'image/{file_extension}',
|
||||
'Metadata': metadata
|
||||
}
|
||||
)
|
||||
|
||||
# S3 URLを生成
|
||||
s3_url = f"https://{self.bucket_name}.s3.{settings.AWS_S3_REGION_NAME}.amazonaws.com/{s3_key}"
|
||||
|
||||
logger.info(f"Standard image uploaded to S3: {s3_url}")
|
||||
return s3_url
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to upload standard image to S3: {e}")
|
||||
raise
|
||||
|
||||
def get_standard_image_url(self, event_code, image_type):
|
||||
"""
|
||||
Get URL for standard image
|
||||
|
||||
Args:
|
||||
event_code: Event code
|
||||
image_type: Type of image (goal, start, checkpoint, etc.)
|
||||
|
||||
Returns:
|
||||
str: S3 URL of standard image or None if not found
|
||||
"""
|
||||
# Try common image extensions
|
||||
extensions = ['.jpg', '.jpeg', '.png', '.gif']
|
||||
|
||||
for ext in extensions:
|
||||
s3_key = f"{event_code}/standards/{image_type}{ext}"
|
||||
try:
|
||||
# Check if object exists
|
||||
self.s3_client.head_object(Bucket=self.bucket_name, Key=s3_key)
|
||||
return f"https://{self.custom_domain}/{s3_key}"
|
||||
except self.s3_client.exceptions.NoSuchKey:
|
||||
continue
|
||||
except Exception as e:
|
||||
logger.error(f"Error checking standard image: {e}")
|
||||
continue
|
||||
|
||||
return None
|
||||
|
||||
def delete_image(self, s3_url):
|
||||
"""
|
||||
Delete image from S3
|
||||
|
||||
Args:
|
||||
s3_url: Full S3 URL of the image
|
||||
|
||||
Returns:
|
||||
bool: True if deleted successfully
|
||||
"""
|
||||
try:
|
||||
# Extract S3 key from URL
|
||||
s3_key = self._extract_s3_key_from_url(s3_url)
|
||||
if not s3_key:
|
||||
logger.error(f"Invalid S3 URL: {s3_url}")
|
||||
return False
|
||||
|
||||
# Delete from S3
|
||||
self.s3_client.delete_object(Bucket=self.bucket_name, Key=s3_key)
|
||||
logger.info(f"Image deleted from S3: {s3_url}")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to delete image from S3: {e}")
|
||||
return False
|
||||
|
||||
def list_event_images(self, event_code, limit=100):
|
||||
"""
|
||||
List all images for an event
|
||||
|
||||
Args:
|
||||
event_code: Event code
|
||||
limit: Maximum number of images to return
|
||||
|
||||
Returns:
|
||||
list: List of S3 URLs
|
||||
"""
|
||||
try:
|
||||
response = self.s3_client.list_objects_v2(
|
||||
Bucket=self.bucket_name,
|
||||
Prefix=f"{event_code}/",
|
||||
MaxKeys=limit
|
||||
)
|
||||
|
||||
urls = []
|
||||
if 'Contents' in response:
|
||||
for obj in response['Contents']:
|
||||
url = f"https://{self.custom_domain}/{obj['Key']}"
|
||||
urls.append(url)
|
||||
|
||||
return urls
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to list event images: {e}")
|
||||
return []
|
||||
|
||||
def _get_file_extension(self, filename):
|
||||
"""Get file extension from filename"""
|
||||
if '.' in filename:
|
||||
return '.' + filename.split('.')[-1].lower()
|
||||
return '.jpg' # default extension
|
||||
|
||||
def _get_timestamp(self):
|
||||
"""Get current timestamp string"""
|
||||
from datetime import datetime
|
||||
return datetime.now().strftime("%Y-%m-%dT%H-%M-%S")
|
||||
|
||||
def _extract_s3_key_from_url(self, s3_url):
|
||||
"""Extract S3 key from full S3 URL"""
|
||||
try:
|
||||
# Remove domain part to get key
|
||||
if self.custom_domain in s3_url:
|
||||
return s3_url.split(self.custom_domain + '/')[-1]
|
||||
return None
|
||||
except Exception:
|
||||
return None
|
||||
Reference in New Issue
Block a user