Files
prefect-flows-pg-backup/setup_blocks.py

31 lines
908 B
Python

"""
Setup Prefect Blocks from environment variables.
"""
import os
from prefect_aws import AwsCredentials
def setup_blocks():
"""Crée ou met à jour les Blocks nécessaires."""
# AWS/S3 Credentials pour Garage
aws_access_key = os.environ.get("AWS_ACCESS_KEY_ID")
aws_secret_key = os.environ.get("AWS_SECRET_ACCESS_KEY")
aws_endpoint = os.environ.get("AWS_ENDPOINT_URL")
if all([aws_access_key, aws_secret_key, aws_endpoint]):
creds = AwsCredentials(
aws_access_key_id=aws_access_key,
aws_secret_access_key=aws_secret_key,
aws_endpoint_url=aws_endpoint,
)
creds.save("garage-credentials", overwrite=True)
print("✅ Block 'garage-credentials' created/updated")
else:
print("⚠️ Skipping garage-credentials: missing environment variables")
if __name__ == "__main__":
setup_blocks()