feat: add script to setup Prefect Blocks from env vars
This commit is contained in:
30
setup_blocks.py
Normal file
30
setup_blocks.py
Normal file
@@ -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()
|
||||||
Reference in New Issue
Block a user