Damien/prefect-flows-pg-backup
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

fix: use get_s3_client() instead of manual boto3 client creation
All checks were successful
Deploy Prefect Flows / deploy (push) Successful in 51s
Details

Browse Source
This commit is contained in:
Damien Arnodo
2026-02-01 10:33:17 +00:00
parent 15b2adf766
commit d8f1751f1a
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
flows/backup.py
View File

@@ -12,7 +12,6 @@ from tempfile import TemporaryDirectory
from prefect import flow, task, get_run_logger from prefect import flow, task, get_run_logger
from prefect.blocks.system import Secret from prefect.blocks.system import Secret
from prefect_aws import AwsCredentials from prefect_aws import AwsCredentials
import boto3
@task(name="pg_dump") @task(name="pg_dump")
@@ -88,12 +87,8 @@ def upload_to_s3(
"""Upload le fichier vers S3 et retourne l'URI.""" """Upload le fichier vers S3 et retourne l'URI."""
logger = get_run_logger() logger = get_run_logger()
s3_client = boto3.client( # Utiliser get_s3_client() qui gère automatiquement l'endpoint
"s3", s3_client = credentials.get_s3_client()
aws_access_key_id=credentials.aws_access_key_id,
aws_secret_access_key=credentials.aws_secret_access_key.get_secret_value(),
endpoint_url=credentials.aws_endpoint_url,
)
key = f"{prefix}/{file_path.name}" if prefix else file_path.name key = f"{prefix}/{file_path.name}" if prefix else file_path.name