mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-05-06 20:09:06 +08:00
23 lines
709 B
Python
23 lines
709 B
Python
import base64
|
||
from modules.s3 import s3_client
|
||
from urllib.parse import urlparse
|
||
|
||
|
||
def get_photo_base64(s3_url) -> str:
|
||
try:
|
||
s3_url = s3_url
|
||
parsed = urlparse(s3_url)
|
||
path_parts = parsed.path.lstrip('/').split('/', 1)
|
||
|
||
if len(path_parts) != 2:
|
||
raise ValueError("Не удалось извлечь bucket и key из URL.")
|
||
|
||
bucket_name, s3_key = path_parts
|
||
|
||
response = s3_client.get_object(Bucket=bucket_name, Key=s3_key)
|
||
image_data = response['Body'].read()
|
||
return base64.b64encode(image_data).decode('utf-8')
|
||
except Exception as e:
|
||
print(f"Ошибка при получении файла: {e}")
|
||
return ""
|