From be272e02b371d69a9a0eec3d2ed2f653c68793e1 Mon Sep 17 00:00:00 2001 From: Damien Arnodo Date: Sat, 31 Jan 2026 16:35:29 +0000 Subject: [PATCH] feat: add script to setup Prefect Blocks from env vars --- setup_blocks.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 setup_blocks.py diff --git a/setup_blocks.py b/setup_blocks.py new file mode 100644 index 0000000..b80a529 --- /dev/null +++ b/setup_blocks.py @@ -0,0 +1,30 @@ +""" +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()