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()