Pythin TFE - Cleaned variable sets, reserved tag ssh key, registry and workspaces

Author: KshitijaChoudhari

examples/variable_sets.py

@@ -15,10 +15,11 @@
 
 import os
 
-from pytfe.types import (
-    CategoryType,
+from pytfe import TFEClient, TFEConfig
+from pytfe.models.project import Project
+from pytfe.models.variable import CategoryType
+from pytfe.models.variable_set import (
     Parent,
-    Project,
     VariableSetApplyToProjectsOptions,
     VariableSetApplyToWorkspacesOptions,
     VariableSetCreateOptions,
@@ -30,10 +31,8 @@
     VariableSetVariableCreateOptions,
     VariableSetVariableListOptions,
     VariableSetVariableUpdateOptions,
-    Workspace,
 )
-
-from pytfe import TFEClient, TFEConfig
+from pytfe.models.workspace import Workspace
 
 
 def variable_set_example():
@@ -193,7 +192,7 @@ def variable_set_example():
         print("7. Workspace operations example...")
         try:
             # List some workspaces first
-            from pytfe.types import WorkspaceListOptions
+            from pytfe.models.workspace import WorkspaceListOptions
 
             workspace_options = WorkspaceListOptions(page_size=5)
             workspaces = list(
@@ -272,7 +271,7 @@ def variable_set_example():
 
         # 9. Read the variable set with includes
         print("9. Reading variable set with includes...")
-        from pytfe.types import VariableSetReadOptions
+        from pytfe.models.variable_set import VariableSetReadOptions
 
         read_options = VariableSetReadOptions(
             include=[VariableSetIncludeOpt.VARS, VariableSetIncludeOpt.WORKSPACES]

src/pytfe/models/__init__.py

@@ -137,7 +137,13 @@
     EnforcementLevel,
     PolicyKind,
 )
-from .project import Project
+from .project import (
+    Project,
+    ProjectAddTagBindingsOptions,
+    ProjectCreateOptions,
+    ProjectListOptions,
+    ProjectUpdateOptions,
+)
 
 # ── Query Runs ────────────────────────────────────────────────────────────────
 from .query_run import (
@@ -423,6 +429,10 @@
     "OrganizationCreateOptions",
     "OrganizationUpdateOptions",
     "Project",
+    "ProjectAddTagBindingsOptions",
+    "ProjectCreateOptions",
+    "ProjectListOptions",
+    "ProjectUpdateOptions",
     "DataRetentionPolicy",
     "DataRetentionPolicyChoice",
     "DataRetentionPolicyDeleteOlder",

src/pytfe/resources/registry_module.py

@@ -147,7 +147,7 @@ def create_with_vcs_connection(
             }
         }
 
-        # Determine the URL based on options - exactly like Go implementation
+        # Determine the URL based on options
         if options.vcs_repo.oauth_token_id and not options.vcs_repo.branch:
             path = "/api/v2/registry-modules"
         else:
@@ -157,7 +157,7 @@ def create_with_vcs_connection(
                 )
             path = f"/api/v2/organizations/{options.vcs_repo.organization_name}/registry-modules/vcs"
 
-        # Validate agent execution mode like Go implementation
+        # Validate agent execution mode for API requirements
         if (
             options.test_config
             and options.test_config.agent_execution_mode == AgentExecutionMode.REMOTE

src/pytfe/resources/workspaces.py

@@ -576,7 +576,7 @@ def _build_workspace_payload(
 
     def delete(self, workspace: str, *, organization: str) -> None:
         """Delete workspace by organization and workspace name."""
-        # Validate parameters (similar to Go implementation)
+        # Validate parameters for proper API usage
         if not valid_string_id(organization):
             raise InvalidOrgError()
         if not valid_string_id(workspace):
@@ -588,15 +588,15 @@ def delete(self, workspace: str, *, organization: str) -> None:
 
     def delete_by_id(self, workspace_id: str) -> None:
         """Delete workspace by workspace ID."""
-        # Validate parameters (similar to Go implementation)
+        # Validate parameters for proper API usage
         if not valid_string_id(workspace_id):
             raise InvalidWorkspaceIDError()
 
         self.t.request("DELETE", f"/api/v2/workspaces/{workspace_id}")
 
     def safe_delete(self, workspace: str, *, organization: str) -> None:
         """Safely delete workspace by organization and name."""
-        # Validate parameters (similar to Go implementation)
+        # Validate parameters for proper API usage
         if not valid_string_id(organization):
             raise InvalidOrgError()
         if not valid_string_id(workspace):
@@ -609,7 +609,7 @@ def safe_delete(self, workspace: str, *, organization: str) -> None:
 
     def safe_delete_by_id(self, workspace_id: str) -> None:
         """Safely delete workspace by workspace ID."""
-        # Validate parameters (similar to Go implementation)
+        # Validate parameters for proper API usage
         if not valid_string_id(workspace_id):
             raise InvalidWorkspaceIDError()
 

src/pytfe/utils.py

@@ -133,7 +133,7 @@ def has_tags_regex_defined(vcs_repo: VCSRepo | None) -> bool:
 
 def validate_workspace_create_options(options: WorkspaceCreateOptions) -> None:
     """
-    Validate workspace create options similar to Go implementation.
+    Validate workspace create options for proper API usage.
     Raises specific validation errors if validation fails.
     """
     # Check required name
@@ -179,7 +179,7 @@ def validate_workspace_create_options(options: WorkspaceCreateOptions) -> None:
 
 def validate_workspace_update_options(options: WorkspaceUpdateOptions) -> None:
     """
-    Validate workspace update options similar to Go implementation.
+    Validate workspace update options for proper API usage.
     Raises specific validation errors if validation fails.
     """
     # Check name format if provided
@@ -216,7 +216,7 @@ def validate_workspace_update_options(options: WorkspaceUpdateOptions) -> None:
 
 def validate_oauth_client_create_options(options: OAuthClientCreateOptions) -> None:
     """
-    Validate OAuth client create options similar to Go implementation.
+    Validate OAuth client create options for proper API usage.
     Raises specific validation errors if validation fails.
     """
     from .errors import (

tests/units/test_reserved_tag_key.py

@@ -4,17 +4,17 @@
 
 import pytest
 
-from src.pytfe._http import HTTPTransport
-from src.pytfe.errors import (
+from pytfe._http import HTTPTransport
+from pytfe.errors import (
     InvalidOrgError,
     ValidationError,
 )
-from src.pytfe.models.reserved_tag_key import (
+from pytfe.models.reserved_tag_key import (
     ReservedTagKeyCreateOptions,
     ReservedTagKeyListOptions,
     ReservedTagKeyUpdateOptions,
 )
-from src.pytfe.resources.reserved_tag_key import ReservedTagKey
+from pytfe.resources.reserved_tag_key import ReservedTagKey
 
 
 class TestReservedTagKeyParsing:

tests/units/test_ssh_keys.py

@@ -4,16 +4,16 @@
 
 import pytest
 
-from src.pytfe._http import HTTPTransport
-from src.pytfe.errors import (
+from pytfe._http import HTTPTransport
+from pytfe.errors import (
     InvalidOrgError,
     InvalidSSHKeyIDError,
 )
-from src.pytfe.models.ssh_key import (
+from pytfe.models.ssh_key import (
     SSHKeyCreateOptions,
     SSHKeyUpdateOptions,
 )
-from src.pytfe.resources.ssh_keys import SSHKeys
+from pytfe.resources.ssh_keys import SSHKeys
 
 
 class TestSSHKeyParsing: