python-utilities-tfc 0.2.0

Utility functions to work with Terraform Cloud and manage Terraform state.

terraform_state_utils.py

A Python utility module for advanced management and automation of Terraform Cloud workspaces and state files.

Features

• State File Filtering:
  Clean and filter Terraform state files by resource/module prefixes.

• Terraform Cloud API Automation:

  • Download/upload state files.
  • Lock/unlock workspaces.
  • Trigger, monitor, and manage runs (cancel, discard, apply).
  • Manage workspace variables.
  • Retrieve run outputs and plan summaries.

• Logging:
  Consistent, timestamped logging for all operations.

Usage

1. Setup

Install dependencies:

pip install requests

Import the module in your Python script:

from utilities import terraform_state_utils

2. Example Operations

Clean a State File

terraform_state_utils.clean_state_file_by_prefixes(
    input_path="input.tfstate",
    output_path="cleaned.tfstate",
    keep_prefixes=["module1"],
    remove_prefixes=["module2"]
)

Download Workspace State

terraform_state_utils.download_workspace_state(
    org_name="my-org",
    workspace_name="my-workspace",
    token="TERRAFORM_API_TOKEN"
)

Trigger and Monitor a Run

run_id = terraform_state_utils.trigger_run(
    api_token="TERRAFORM_API_TOKEN",
    organization="my-org",
    workspace_id="ws-xxxx",
    auto_apply=True
)
terraform_state_utils.monitor_run(
    api_token="TERRAFORM_API_TOKEN",
    organization="my-org",
    run_id=run_id
)

Get Run Outputs

outputs = terraform_state_utils.get_run_outputs(
    api_token="TERRAFORM_API_TOKEN",
    run_id="run-xxxx"
)
print(outputs)

Functions Overview

• setup_logging(): Configure logging.
• clean_state_file_by_prefixes(): Filter resources in a state file.
• download_workspace_state(): Download the latest state from Terraform Cloud.
• push_state_to_terraform_cloud(): Upload a state file to Terraform Cloud.
• lock_workspace(), unlock_workspace(): Lock/unlock a workspace.
• trigger_run(), monitor_run(), cancel_run(), discard_run(), apply_run(): Manage runs.
• get_run_outputs(), get_plan_summary(): Retrieve outputs and plan details.
• get_variable_id(), update_variable(): Manage workspace variables.

Requirements

• Python 3.11+
• requests library

Notes

• All API operations require a valid Terraform Cloud API token.
• Use caution when discarding or canceling runs, as this may affect your infrastructure state.

License

MIT License