Function setup_environment

examples/memory/dapr_session_example.py:482–551

Create Redis/PostgreSQL component files and start containers if available.

(components_dir: str = "./components", overwrite: bool = False)

def setup_environment(components_dir: str = "./components", overwrite: bool = False) -> None:
    """Create Redis/PostgreSQL component files and start containers if available."""
    components_path = Path(components_dir)
    components_path.mkdir(parents=True, exist_ok=True)

    redis_component = """
apiVersion: dapr.io/v1alpha1
kind: Component
metadata:
  name: statestore-redis
spec:
  type: state.redis
  version: v1
  metadata:
  - name: redisHost
    value: localhost:6379
  - name: redisPassword
    value: ""
""".lstrip()

    postgres_component = """
apiVersion: dapr.io/v1alpha1
kind: Component
metadata:
  name: statestore-postgres
spec:
  type: state.postgresql
  version: v2
  metadata:
  - name: connectionString
    value: "host=localhost user=postgres password=postgres dbname=dapr port=5432"
""".lstrip()

    default_component = """
apiVersion: dapr.io/v1alpha1
kind: Component
metadata:
  name: statestore
spec:
  type: state.redis
  version: v1
  metadata:
  - name: redisHost
    value: localhost:6379
  - name: redisPassword
    value: ""
""".lstrip()

    _write_text_file(components_path / "statestore-redis.yaml", redis_component, overwrite)
    _write_text_file(components_path / "statestore-postgres.yaml", postgres_component, overwrite)
    _write_text_file(components_path / "statestore.yaml", default_component, overwrite)

    print(f"Components written under: {components_path.resolve()}")

    _ensure_container("dapr_redis", ["-p", "6379:6379", "redis:7-alpine"])
    _ensure_container(
        "dapr_postgres",
        [