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