fix: add missing exports to __all__ and fix linting issues

- Add all new run tasks integration models to __all__ exports
- Fix trailing whitespace issues across multiple files
- Run ruff format to ensure consistent code style
- All 22 unit tests passing
- All linting checks pass

Author: KshitijaChoudhari

src/pytfe/models/__init__.py

@@ -258,23 +258,6 @@
     RunTaskRequest,
     RunTaskRequestCapabilities,
 )
-from .task_result import (
-    TaskEnforcementLevel,
-    TaskResult,
-    TaskResultStatus,
-    TaskResultStatusTimestamps,
-)
-from .task_stages import (
-    Actions,
-    Permissions,
-    Stage,
-    TaskStage,
-    TaskStageListOptions,
-    TaskStageOverrideOptions,
-    TaskStageReadOptions,
-    TaskStageStatus,
-    TaskStageStatusTimestamps,
-)
 from .run_trigger import (
     RunTrigger,
     RunTriggerCreateOptions,
@@ -293,6 +276,23 @@
     SSHKeyListOptions,
     SSHKeyUpdateOptions,
 )
+from .task_result import (
+    TaskEnforcementLevel,
+    TaskResult,
+    TaskResultStatus,
+    TaskResultStatusTimestamps,
+)
+from .task_stages import (
+    Actions,
+    Permissions,
+    Stage,
+    TaskStage,
+    TaskStageListOptions,
+    TaskStageOverrideOptions,
+    TaskStageReadOptions,
+    TaskStageStatus,
+    TaskStageStatusTimestamps,
+)
 
 # Variables
 from .variable import (
@@ -556,6 +556,19 @@
     "RunTaskCreateOptions",
     "RunTaskUpdateOptions",
     "RunTaskReadOptions",
+    "RunTaskRequest",
+    "RunTaskRequestCapabilities",
+    "TaskResult",
+    "TaskResultStatus",
+    "TaskResultStatusTimestamps",
+    "Actions",
+    "Permissions",
+    "TaskStage",
+    "TaskStageListOptions",
+    "TaskStageOverrideOptions",
+    "TaskStageReadOptions",
+    "TaskStageStatus",
+    "TaskStageStatusTimestamps",
     # Run triggers
     "RunTrigger",
     "RunTriggerCreateOptions",

src/pytfe/models/run_task_request.py

@@ -15,19 +15,18 @@ class RunTaskRequestCapabilities(BaseModel):
     """Capabilities that the caller supports."""
 
     outcomes: bool = Field(
-        default=False,
-        description="Whether the run task server supports outcomes"
+        default=False, description="Whether the run task server supports outcomes"
     )
-    
+
     model_config = ConfigDict(populate_by_name=True)
 
 
 class RunTaskRequest(BaseModel):
     """Represents the payload that TFC/TFE sends to a run task's URL.
-    
+
     This is the incoming request that your external run task server receives
     from Terraform Cloud/Enterprise when a run task is triggered.
-    
+
     API Documentation:
         https://developer.hashicorp.com/terraform/enterprise/api-docs/run-tasks/run-tasks-integration#common-properties
     """
@@ -36,45 +35,26 @@ class RunTaskRequest(BaseModel):
         description="Token to use for authentication when sending callback"
     )
     capabilities: RunTaskRequestCapabilities | None = Field(
-        default=None,
-        description="Capabilities that the caller supports"
+        default=None, description="Capabilities that the caller supports"
     )
     configuration_version_download_url: str | None = Field(
-        default=None,
-        description="URL to download the configuration version"
+        default=None, description="URL to download the configuration version"
     )
     configuration_version_id: str | None = Field(
-        default=None,
-        description="ID of the configuration version"
-    )
-    is_speculative: bool = Field(
-        description="Whether this is a speculative run"
-    )
-    organization_name: str = Field(
-        description="Name of the organization"
-    )
-    payload_version: int = Field(
-        description="Version of the payload format"
+        default=None, description="ID of the configuration version"
     )
+    is_speculative: bool = Field(description="Whether this is a speculative run")
+    organization_name: str = Field(description="Name of the organization")
+    payload_version: int = Field(description="Version of the payload format")
     plan_json_api_url: str | None = Field(
         default=None,
-        description="URL to access the plan JSON via API (post_plan, pre_apply, post_apply stages)"
-    )
-    run_app_url: str = Field(
-        description="URL to view the run in TFC/TFE UI"
-    )
-    run_created_at: datetime = Field(
-        description="Timestamp when the run was created"
-    )
-    run_created_by: str = Field(
-        description="Username of the user who created the run"
-    )
-    run_id: str = Field(
-        description="ID of the run"
-    )
-    run_message: str = Field(
-        description="Message associated with the run"
+        description="URL to access the plan JSON via API (post_plan, pre_apply, post_apply stages)",
     )
+    run_app_url: str = Field(description="URL to view the run in TFC/TFE UI")
+    run_created_at: datetime = Field(description="Timestamp when the run was created")
+    run_created_by: str = Field(description="Username of the user who created the run")
+    run_id: str = Field(description="ID of the run")
+    run_message: str = Field(description="Message associated with the run")
     stage: str = Field(
         description="Stage when the run task is executed (pre_plan, post_plan, pre_apply, post_apply)"
     )
@@ -84,37 +64,24 @@ class RunTaskRequest(BaseModel):
     task_result_enforcement_level: str = Field(
         description="Enforcement level for the task result (advisory, mandatory)"
     )
-    task_result_id: str = Field(
-        description="ID of the task result"
-    )
-    vcs_branch: str | None = Field(
-        default=None,
-        description="VCS branch name"
-    )
+    task_result_id: str = Field(description="ID of the task result")
+    vcs_branch: str | None = Field(default=None, description="VCS branch name")
     vcs_commit_url: str | None = Field(
-        default=None,
-        description="URL to the VCS commit"
+        default=None, description="URL to the VCS commit"
     )
     vcs_pull_request_url: str | None = Field(
-        default=None,
-        description="URL to the VCS pull request"
+        default=None, description="URL to the VCS pull request"
     )
     vcs_repo_url: str | None = Field(
-        default=None,
-        description="URL to the VCS repository"
+        default=None, description="URL to the VCS repository"
     )
     workspace_app_url: str = Field(
         description="URL to view the workspace in TFC/TFE UI"
     )
-    workspace_id: str = Field(
-        description="ID of the workspace"
-    )
-    workspace_name: str = Field(
-        description="Name of the workspace"
-    )
+    workspace_id: str = Field(description="ID of the workspace")
+    workspace_name: str = Field(description="Name of the workspace")
     workspace_working_directory: str | None = Field(
-        default=None,
-        description="Working directory for the workspace"
+        default=None, description="Working directory for the workspace"
     )
 
     model_config = ConfigDict(populate_by_name=True)

src/pytfe/models/task_result.py

@@ -17,7 +17,7 @@
 
 class TaskResultStatus(str, Enum):
     """Task result status enum."""
-    
+
     PASSED = "passed"
     FAILED = "failed"
     PENDING = "pending"
@@ -28,30 +28,30 @@ class TaskResultStatus(str, Enum):
 
 class TaskEnforcementLevel(str, Enum):
     """Task enforcement level enum."""
-    
+
     ADVISORY = "advisory"
     MANDATORY = "mandatory"
 
 
 class TaskResultStatusTimestamps(BaseModel):
     """Timestamps recorded for a task result."""
-    
+
     errored_at: datetime | None = Field(default=None, alias="errored-at")
     running_at: datetime | None = Field(default=None, alias="running-at")
     canceled_at: datetime | None = Field(default=None, alias="canceled-at")
     failed_at: datetime | None = Field(default=None, alias="failed-at")
     passed_at: datetime | None = Field(default=None, alias="passed-at")
-    
+
     model_config = ConfigDict(populate_by_name=True)
 
 
 class TaskResult(BaseModel):
     """Represents a HCP Terraform or Terraform Enterprise run task result.
-    
+
     API Documentation:
         https://developer.hashicorp.com/terraform/cloud-docs/api-docs/task-results
     """
-    
+
     id: str
     status: TaskResultStatus
     message: str
@@ -67,8 +67,8 @@ class TaskResult(BaseModel):
         alias="workspace-task-enforcement-level"
     )
     agent_pool_id: str | None = Field(default=None, alias="agent-pool-id")
-    
+
     # Relationships
     task_stage: TaskStage | None = Field(default=None, alias="task-stage")
-    
+
     model_config = ConfigDict(populate_by_name=True)

src/pytfe/models/task_stages.py

@@ -19,7 +19,7 @@
 
 class Stage(str, Enum):
     """Enum representing possible run stages for run tasks."""
-    
+
     PRE_PLAN = "pre-plan"
     POST_PLAN = "post-plan"
     PRE_APPLY = "pre-apply"
@@ -28,7 +28,7 @@ class Stage(str, Enum):
 
 class TaskStageStatus(str, Enum):
     """Enum representing all possible statuses for a task stage."""
-    
+
     PENDING = "pending"
     RUNNING = "running"
     PASSED = "passed"
@@ -41,43 +41,43 @@ class TaskStageStatus(str, Enum):
 
 class Permissions(BaseModel):
     """Permission types for overriding a task stage."""
-    
+
     can_override_policy: bool | None = Field(default=None, alias="can-override-policy")
     can_override_tasks: bool | None = Field(default=None, alias="can-override-tasks")
     can_override: bool | None = Field(default=None, alias="can-override")
-    
+
     model_config = ConfigDict(populate_by_name=True)
 
 
 class Actions(BaseModel):
     """Task stage actions."""
-    
+
     is_overridable: bool | None = Field(default=None, alias="is-overridable")
-    
+
     model_config = ConfigDict(populate_by_name=True)
 
 
 class TaskStageStatusTimestamps(BaseModel):
     """Timestamps recorded for a task stage."""
-    
+
     errored_at: datetime | None = Field(default=None, alias="errored-at")
     running_at: datetime | None = Field(default=None, alias="running-at")
     canceled_at: datetime | None = Field(default=None, alias="canceled-at")
     failed_at: datetime | None = Field(default=None, alias="failed-at")
     passed_at: datetime | None = Field(default=None, alias="passed-at")
-    
+
     model_config = ConfigDict(populate_by_name=True)
 
 
 class TaskStage(BaseModel):
     """Represents a HCP Terraform or Terraform Enterprise run's task stage.
-    
+
     Task stages are where run tasks can occur during a run lifecycle.
-    
+
     API Documentation:
         https://developer.hashicorp.com/terraform/cloud-docs/api-docs/task-stages
     """
-    
+
     id: str
     stage: Stage
     status: TaskStageStatus
@@ -86,34 +86,33 @@ class TaskStage(BaseModel):
     updated_at: datetime = Field(alias="updated-at")
     permissions: Permissions | None = None
     actions: Actions | None = None
-    
+
     # Relationships
     run: Run | None = None
     task_results: list[TaskResult] = Field(default_factory=list, alias="task-results")
     policy_evaluations: list[PolicyEvaluation] = Field(
-        default_factory=list, 
-        alias="policy-evaluations"
+        default_factory=list, alias="policy-evaluations"
     )
-    
+
     model_config = ConfigDict(populate_by_name=True)
 
 
 class TaskStageOverrideOptions(BaseModel):
     """Options for overriding a task stage."""
-    
+
     comment: str | None = None
 
 
 class TaskStageReadOptions(BaseModel):
     """Options for reading a task stage."""
-    
+
     include: list[str] | None = None
 
 
 class TaskStageListOptions(BaseModel):
     """Options for listing task stages."""
-    
+
     page_number: int | None = Field(default=None, alias="page[number]")
     page_size: int | None = Field(default=None, alias="page[size]")
-    
+
     model_config = ConfigDict(populate_by_name=True)