lint issues fixed

Author: aayushsingh2502

examples/query_run.py

@@ -15,7 +15,7 @@
 
 Usage:
     python query_run.py
-    
+
 Note: Query Runs require Terraform ~>1.14 which includes the 'terraform query' command.
       These tests may fail with error status since the feature is not fully available yet.
 """
@@ -36,7 +36,7 @@ def get_client_and_workspace():
     client = TFEClient(TFEConfig.from_env())
     organization = os.getenv("TFE_ORG", "aayush-test")
     workspace_name = "query-test"  # Default workspace for testing
-    
+
     # Get workspace
     workspace = client.workspaces.read(workspace_name, organization=organization)
     return client, workspace
@@ -45,27 +45,27 @@ def get_client_and_workspace():
 def test_list():
     """Test 1: List query runs in a workspace."""
     print("=== Test 1: List Query Runs ===")
-    
+
     client, workspace = get_client_and_workspace()
-    
+
     try:
         # Simple list
         query_runs = list(client.query_runs.list(workspace.id))
         print(f"Found {len(query_runs)} query runs in workspace '{workspace.name}'")
-        
+
         for i, qr in enumerate(query_runs[:5], 1):
             print(f"  {i}. {qr.id}")
             print(f"     Status: {qr.status}")
             print(f"     Created: {qr.created_at}")
             print()
-        
+
         # List with options
         options = QueryRunListOptions(page_size=5)
         limited_runs = list(client.query_runs.list(workspace.id, options))
         print(f"Retrieved {len(limited_runs)} query runs (page_size=5)")
-        
+
         return query_runs
-        
+
     except Exception as e:
         print(f"Error: {e}")
         return []
@@ -74,34 +74,34 @@ def test_list():
 def test_create():
     """Test 2: Create a new query run."""
     print("\n=== Test 2: Create Query Run ===")
-    
+
     client, workspace = get_client_and_workspace()
-    
+
     try:
         # Get the latest configuration version
         config_versions = list(client.configuration_versions.list(workspace.id))
         if not config_versions:
             print("ERROR: No configuration versions found in workspace")
             return None
-        
+
         config_version = config_versions[0]
         print(f"Using configuration version: {config_version.id}")
-        
+
         # Create query run
         options = QueryRunCreateOptions(
             source=QueryRunSource.API,
             workspace_id=workspace.id,
             configuration_version_id=config_version.id,
         )
-        
+
         query_run = client.query_runs.create(options)
         print(f"Created query run: {query_run.id}")
         print(f"  Status: {query_run.status}")
         print(f"  Source: {query_run.source}")
         print(f"  Created: {query_run.created_at}")
-        
+
         return query_run
-        
+
     except Exception as e:
         print(f"Error: {e}")
         return None
@@ -110,9 +110,9 @@ def test_create():
 def test_read(query_run_id=None):
     """Test 3: Read a specific query run."""
     print("\n=== Test 3: Read Query Run ===")
-    
+
     client, workspace = get_client_and_workspace()
-    
+
     try:
         # If no query_run_id provided, get the first one from the list
         if not query_run_id:
@@ -122,16 +122,16 @@ def test_read(query_run_id=None):
                 return None
             query_run_id = query_runs[0].id
             print(f"Using first query run from list: {query_run_id}")
-        
+
         # Read the query run
         query_run = client.query_runs.read(query_run_id)
         print(f"Read query run: {query_run.id}")
         print(f"  Status: {query_run.status}")
         print(f"  Source: {query_run.source}")
         print(f"  Created: {query_run.created_at}")
-        
+
         if query_run.status_timestamps:
-            print(f"  Status Timestamps:")
+            print("  Status Timestamps:")
             if query_run.status_timestamps.queued_at:
                 print(f"    Queued: {query_run.status_timestamps.queued_at}")
             if query_run.status_timestamps.running_at:
@@ -140,9 +140,9 @@ def test_read(query_run_id=None):
                 print(f"    Finished: {query_run.status_timestamps.finished_at}")
             if query_run.status_timestamps.errored_at:
                 print(f"    Errored: {query_run.status_timestamps.errored_at}")
-        
+
         return query_run
-        
+
     except Exception as e:
         print(f"Error: {e}")
         return None
@@ -151,9 +151,9 @@ def test_read(query_run_id=None):
 def test_logs(query_run_id=None):
     """Test 4: Retrieve logs for a query run."""
     print("\n=== Test 4: Get Query Run Logs ===")
-    
+
     client, workspace = get_client_and_workspace()
-    
+
     try:
         # If no query_run_id provided, get the first one from the list
         if not query_run_id:
@@ -163,33 +163,33 @@ def test_logs(query_run_id=None):
                 return None
             query_run_id = query_runs[0].id
             print(f"Using first query run from list: {query_run_id}")
-        
+
         # Get logs
         logs = client.query_runs.logs(query_run_id)
         log_content = logs.read().decode("utf-8")
-        
+
         print(f"Retrieved logs for query run: {query_run_id}")
         print(f"  Log size: {len(log_content)} bytes")
-        print(f"\n--- Log Preview (first 500 chars) ---")
+        print("\n--- Log Preview (first 500 chars) ---")
         print(log_content[:500])
         if len(log_content) > 500:
             print(f"\n... ({len(log_content) - 500} more characters)")
         print("--- End of Log Preview ---")
-        
+
         return log_content
-        
+
     except Exception as e:
         print(f"Error: {e}")
-        print(f"  Note: Logs may not be available if the query run hasn't started yet")
+        print("  Note: Logs may not be available if the query run hasn't started yet")
         return None
 
 
 def test_cancel(query_run_id=None):
     """Test 5: Cancel a query run."""
     print("\n=== Test 5: Cancel Query Run ===")
-    
+
     client, workspace = get_client_and_workspace()
-    
+
     try:
         # If no query_run_id provided, create a new one
         if not query_run_id:
@@ -200,30 +200,30 @@ def test_cancel(query_run_id=None):
                 return False
             query_run_id = new_run.id
             time.sleep(1)  # Give it a moment to start
-        
+
         # Cancel the query run
         client.query_runs.cancel(query_run_id)
         print(f"Cancel requested for query run: {query_run_id}")
-        
+
         # Verify cancellation
         time.sleep(2)
         query_run = client.query_runs.read(query_run_id)
         print(f"  Status after cancel: {query_run.status}")
-        
+
         return True
-        
+
     except Exception as e:
         print(f"Error: {e}")
-        print(f"  Note: Query run may not be in a cancelable state")
+        print("  Note: Query run may not be in a cancelable state")
         return False
 
 
 def test_force_cancel(query_run_id=None):
     """Test 6: Force cancel a query run."""
     print("\n=== Test 6: Force Cancel Query Run ===")
-    
+
     client, workspace = get_client_and_workspace()
-    
+
     try:
         # If no query_run_id provided, create a new one
         if not query_run_id:
@@ -234,21 +234,21 @@ def test_force_cancel(query_run_id=None):
                 return False
             query_run_id = new_run.id
             time.sleep(1)  # Give it a moment to start
-        
+
         # Force cancel the query run
         client.query_runs.force_cancel(query_run_id)
         print(f"Force cancel requested for query run: {query_run_id}")
-        
+
         # Verify force cancellation
         time.sleep(2)
         query_run = client.query_runs.read(query_run_id)
         print(f"  Status after force cancel: {query_run.status}")
-        
+
         return True
-        
+
     except Exception as e:
         print(f"Error: {e}")
-        print(f"  Note: Query run may not be in a force-cancelable state")
+        print("  Note: Query run may not be in a force-cancelable state")
         return False
 
 
@@ -262,26 +262,26 @@ def main():
     print("NOTE: Query Runs require Terraform 1.10+ with 'terraform query' command.")
     print("      Most query runs will error since this feature is not yet available.")
     print("=" * 80)
-    
+
     # Test 1: List query runs
     query_runs = test_list()
-    
+
     # Test 2: Create a query run
     new_query_run = test_create()
-    
+
     # Test 3: Read a query run
     if query_runs:
         test_read(query_runs[0].id)
     elif new_query_run:
         test_read(new_query_run.id)
-    
+
     # Test 4: Get logs (use first query run from list)
     if query_runs:
         test_logs(query_runs[0].id)
-    
+
     # Test 5: Cancel a query run (creates new one)
     test_cancel()
-    
+
     # Test 6: Force cancel a query run (creates new one)
     test_force_cancel()
 

src/pytfe/models/__init__.py

@@ -434,10 +434,12 @@
     "RegistryProviderReadOptions",
     # Query runs
     "QueryRun",
+    "QueryRunActions",
     "QueryRunCancelOptions",
     "QueryRunCreateOptions",
     "QueryRunForceCancelOptions",
     "QueryRunIncludeOpt",
+    "QueryRunList",
     "QueryRunListOptions",
     "QueryRunReadOptions",
     "QueryRunSource",

src/pytfe/models/query_run.py

@@ -2,7 +2,6 @@
 
 from datetime import datetime
 from enum import Enum
-from typing import Any
 
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -54,7 +53,9 @@ class QueryRunStatusTimestamps(BaseModel):
         None, alias="running-at", description="When the query run started running"
     )
     finished_at: datetime | None = Field(
-        None, alias="finished-at", description="When the query run finished successfully"
+        None,
+        alias="finished-at",
+        description="When the query run finished successfully",
     )
     errored_at: datetime | None = Field(
         None, alias="errored-at", description="When the query run encountered an error"
@@ -90,9 +91,7 @@ class QueryRun(BaseModel):
     updated_at: datetime | None = Field(
         None, alias="updated-at", description="The time this query run was last updated"
     )
-    source: QueryRunSource | str = Field(
-        ..., description="The source of the query run"
-    )
+    source: QueryRunSource | str = Field(..., description="The source of the query run")
     status: QueryRunStatus = Field(
         ..., description="The current status of the query run"
     )
@@ -127,14 +126,14 @@ class QueryRunCreateOptions(BaseModel):
 
     model_config = ConfigDict(populate_by_name=True)
 
-    source: QueryRunSource | str = Field(
-        ..., description="The source of the query run"
-    )
+    source: QueryRunSource | str = Field(..., description="The source of the query run")
     variables: list[QueryRunVariable] | None = Field(
         None, description="Run-specific variable values"
     )
     workspace_id: str = Field(
-        ..., alias="workspace-id", description="The workspace ID to run the query against"
+        ...,
+        alias="workspace-id",
+        description="The workspace ID to run the query against",
     )
     configuration_version_id: str | None = Field(
         None,