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