lint issue fixed

Author: aayushsingh2502

src/tfe/resources/registry_module.py

@@ -21,12 +21,12 @@
     RegistryModuleListOptions,
     RegistryModulePermissions,
     RegistryModuleUpdateOptions,
+    RegistryModuleVCSRepo,
     RegistryModuleVersion,
     RegistryModuleVersionStatuses,
     RegistryName,
     TerraformRegistryModule,
     TestConfig,
-    VCSRepo,
 )
 from ..utils import valid_string, valid_string_id, valid_version
 from ._base import _Service
@@ -453,7 +453,7 @@ def _parse_registry_module(self, data: dict[str, Any]) -> RegistryModule:
         vcs_repo = None
         if "vcs-repo" in attributes:
             vcs_data = attributes["vcs-repo"]
-            vcs_repo = VCSRepo(
+            vcs_repo = RegistryModuleVCSRepo(
                 branch=vcs_data.get("branch"),
                 display_identifier=vcs_data.get("display-identifier"),
                 identifier=vcs_data.get("identifier"),

src/tfe/resources/variable.py

@@ -9,7 +9,7 @@
     ERR_REQUIRED_CATEGORY,
     ERR_REQUIRED_KEY,
 )
-from ..types import (
+from ..types import (  # type: ignore[attr-defined]
     Variable,
     VariableCreateOptions,
     VariableListOptions,

src/tfe/resources/workspaces.py

@@ -16,7 +16,7 @@
     WorkspaceMinimumLimitError,
     WorkspaceRequiredError,
 )
-from ..types import (
+from ..types import (  # type: ignore[attr-defined]
     DataRetentionPolicy,
     DataRetentionPolicyChoice,
     DataRetentionPolicyDeleteOlder,
@@ -69,7 +69,7 @@ def _em_safe(v: Any) -> ExecutionMode | None:
     # Only accept strings; map to enum if known, else None
     if not isinstance(v, str):
         return None
-    return ExecutionMode._value2member_map_.get(v)  # type: ignore[return-value]
+    return ExecutionMode._value2member_map_.get(v)
 
 
 def _ws_from(d: dict[str, Any], org: str | None = None) -> Workspace:
@@ -1174,4 +1174,4 @@ def readme(self, workspace_id: str) -> str | None:
             if inc.get("type") == "workspace-readme" and inc.get("id") == readme_id:
                 return (inc.get("attributes") or {}).get("raw-markdown")
 
-        return None
+        return None

src/tfe/types/__init__.py

@@ -24,6 +24,7 @@
     RegistryModulePermissions,
     RegistryModuleStatus,
     RegistryModuleUpdateOptions,
+    RegistryModuleVCSRepo,
     RegistryModuleVCSRepoOptions,
     RegistryModuleVCSRepoUpdateOptions,
     RegistryModuleVersion,
@@ -32,8 +33,8 @@
     RegistryName,
     Resource,
     Root,
+    TerraformRegistryModule,
     TestConfig,
-    VCSRepo,
 )
 
 # Define what should be available when importing with *
@@ -57,6 +58,7 @@
     "RegistryModulePermissions",
     "RegistryModuleStatus",
     "RegistryModuleUpdateOptions",
+    "RegistryModuleVCSRepo",
     "RegistryModuleVCSRepoOptions",
     "RegistryModuleVCSRepoUpdateOptions",
     "RegistryModuleVersion",
@@ -66,7 +68,7 @@
     "Resource",
     "Root",
     "TestConfig",
-    "VCSRepo",
+    "TerraformRegistryModule",
     # Main types from types.py (will be dynamically added below)
     "Capacity",
     "DataRetentionPolicy",
@@ -76,8 +78,10 @@
     "DataRetentionPolicyDontDelete",
     "DataRetentionPolicyDontDeleteSetOptions",
     "DataRetentionPolicySetOptions",
+    "EffectiveTagBinding",
     "Entitlements",
     "ExecutionMode",
+    "LockedByChoice",
     "Organization",
     "OrganizationCreateOptions",
     "OrganizationUpdateOptions",
@@ -87,7 +91,37 @@
     "Run",
     "RunQueue",
     "RunStatus",
+    "Tag",
+    "TagBinding",
+    "TagList",
+    "Variable",
+    "VariableCreateOptions",
+    "VariableListOptions",
+    "VariableUpdateOptions",
+    "VCSRepo",
     "Workspace",
+    "WorkspaceActions",
+    "WorkspaceAddRemoteStateConsumersOptions",
+    "WorkspaceAddTagBindingsOptions",
+    "WorkspaceAddTagsOptions",
+    "WorkspaceAssignSSHKeyOptions",
+    "WorkspaceCreateOptions",
+    "WorkspaceIncludeOpt",
+    "WorkspaceList",
+    "WorkspaceListOptions",
+    "WorkspaceListRemoteStateConsumersOptions",
+    "WorkspaceLockOptions",
+    "WorkspaceOutputs",
+    "WorkspacePermissions",
+    "WorkspaceReadOptions",
+    "WorkspaceRemoveRemoteStateConsumersOptions",
+    "WorkspaceRemoveTagsOptions",
+    "WorkspaceRemoveVCSConnectionOptions",
+    "WorkspaceSettingOverwrites",
+    "WorkspaceSource",
+    "WorkspaceTagListOptions",
+    "WorkspaceUpdateOptions",
+    "WorkspaceUpdateRemoteStateConsumersOptions",
 ]
 
 # Load the main types.py file that's at the same level as this types/ directory

src/tfe/types/registry_module_types.py

@@ -87,8 +87,8 @@ class RegistryModulePermissions(BaseModel):
     can_retry: bool
 
 
-class VCSRepo(BaseModel):
-    """VCS repository configuration."""
+class RegistryModuleVCSRepo(BaseModel):
+    """VCS repository configuration for registry modules."""
 
     branch: str | None = None
     display_identifier: str | None = None
@@ -101,6 +101,7 @@ class VCSRepo(BaseModel):
     tags: bool | None = None
     source_directory: str | None = None
     tag_prefix: str | None = None
+    organization_name: str | None = None
 
 
 class TestConfig(BaseModel):
@@ -132,7 +133,7 @@ class RegistryModule(BaseModel):
     publishing_mechanism: PublishingMechanism | None = None
     status: RegistryModuleStatus | None = None
     test_config: TestConfig | None = None
-    vcs_repo: VCSRepo | None = None
+    vcs_repo: RegistryModuleVCSRepo | None = None
     version_statuses: list[RegistryModuleVersionStatuses] = Field(default_factory=list)
     created_at: str | None = None
     updated_at: str | None = None

src/tfe/utils.py

@@ -15,7 +15,11 @@
     UnsupportedBothTriggerPatternsAndPrefixesError,
     UnsupportedOperationsError,
 )
-from .types import VCSRepo, WorkspaceCreateOptions, WorkspaceUpdateOptions
+from .types import (  # type: ignore[attr-defined]
+    VCSRepo,
+    WorkspaceCreateOptions,
+    WorkspaceUpdateOptions,
+)
 
 _STRING_ID_PATTERN = re.compile(r"^[a-zA-Z0-9][a-zA-Z0-9_-]{2,}$")
 _VERSION_PATTERN = re.compile(
@@ -50,7 +54,8 @@ def valid_string_id(v: str | None) -> bool:
 def valid_version(v: str | None) -> bool:
     """Validate semantic version string."""
     return v is not None and _VERSION_PATTERN.match(str(v)) is not None
-  
+
+
 def is_valid_workspace_name(name: str | None) -> bool:
     """
     Check if a workspace name is valid.