40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""
|
|
Setup Prefect Blocks from environment variables.
|
|
"""
|
|
|
|
import os
|
|
from prefect_aws import AwsCredentials
|
|
from prefect.blocks.system import Secret
|
|
|
|
|
|
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")
|
|
|
|
# PostgreSQL password pour netbox
|
|
pg_password = os.environ.get("PG_PASSWORD")
|
|
if pg_password:
|
|
Secret(value=pg_password).save("netbox-db-password", overwrite=True)
|
|
print("✅ Block 'netbox-db-password' created/updated")
|
|
else:
|
|
print("⚠️ Skipping netbox-db-password: PG_PASSWORD not set")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
setup_blocks()
|