TF- 30336 - Feature providing registry module support to sdk

src/tfe/client.py

@@ -4,6 +4,7 @@
 from .config import TFEConfig
 from .resources.organizations import Organizations
 from .resources.projects import Projects
+from .resources.registry_module import RegistryModules
 from .resources.variable import Variables
 from .resources.workspaces import Workspaces
 
@@ -29,6 +30,7 @@ def __init__(self, config: TFEConfig | None = None):
         self.projects = Projects(self._transport)
         self.variables = Variables(self._transport)
         self.workspaces = Workspaces(self._transport)
+        self.registry_modules = RegistryModules(self._transport)
 
     def close(self) -> None:
         pass

src/tfe/errors.py

@@ -57,6 +57,26 @@ class RequiredFieldMissing(TFEError): ...
 ERR_REQUIRED_NAME = "name is required"
 ERR_INVALID_ORG = "invalid organization name"
 ERR_REQUIRED_EMAIL = "email is required"
+
+# Registry Module Error Constants
+ERR_REQUIRED_PROVIDER = "provider is required"
+ERR_INVALID_PROVIDER = "invalid value for provider"
+ERR_REQUIRED_VERSION = "version is required"
+ERR_INVALID_VERSION = "invalid value for version"
+ERR_REQUIRED_NAMESPACE = "namespace is required"
+ERR_INVALID_REGISTRY_NAME = "invalid registry name"
+ERR_UNSUPPORTED_BOTH_NAMESPACE_AND_PRIVATE_REGISTRY_NAME = (
+    "namespace cannot be used with private registry"
+)
+ERR_REQUIRED_VCS_REPO = "VCS repo is required"
+ERR_REQUIRED_BRANCH_WHEN_TESTS_ENABLED = "branch is required when tests are enabled"
+ERR_BRANCH_MUST_BE_EMPTY_WHEN_TAGS_ENABLED = (
+    "branch must be empty when tags are enabled"
+)
+ERR_AGENT_POOL_NOT_REQUIRED_FOR_REMOTE_EXECUTION = (
+    "agent pool not required for remote execution"
+)
+ERR_INVALID_MODULE_ID = "invalid module ID"
 # Workspaces
 ERR_INVALID_WORKSPACE_ID = "invalid workspace ID"
 ERR_INVALID_VARIABLE_ID = "invalid variable ID"

src/tfe/models/__init__.py

@@ -0,0 +1,137 @@
+"""Types package for TFE client."""
+
+# Import all types from the main types module by using importlib to avoid circular imports
+import importlib.util
+import os
+
+# Re-export all registry module types
+from .registry_module_types import (
+    AgentExecutionMode,
+    Commit,
+    CommitList,
+    Input,
+    Output,
+    ProviderDependency,
+    PublishingMechanism,
+    RegistryModule,
+    RegistryModuleCreateOptions,
+    RegistryModuleCreateVersionOptions,
+    RegistryModuleCreateWithVCSConnectionOptions,
+    RegistryModuleID,
+    RegistryModuleList,
+    RegistryModuleListIncludeOpt,
+    RegistryModuleListOptions,
+    RegistryModulePermissions,
+    RegistryModuleStatus,
+    RegistryModuleUpdateOptions,
+    RegistryModuleVCSRepo,
+    RegistryModuleVCSRepoOptions,
+    RegistryModuleVCSRepoUpdateOptions,
+    RegistryModuleVersion,
+    RegistryModuleVersionStatus,
+    RegistryModuleVersionStatuses,
+    RegistryName,
+    Resource,
+    Root,
+    TerraformRegistryModule,
+    TestConfig,
+)
+
+# Define what should be available when importing with *
+__all__ = [
+    # Registry module types
+    "AgentExecutionMode",
+    "Commit",
+    "CommitList",
+    "Input",
+    "Output",
+    "ProviderDependency",
+    "PublishingMechanism",
+    "RegistryModule",
+    "RegistryModuleCreateOptions",
+    "RegistryModuleCreateVersionOptions",
+    "RegistryModuleCreateWithVCSConnectionOptions",
+    "RegistryModuleID",
+    "RegistryModuleList",
+    "RegistryModuleListIncludeOpt",
+    "RegistryModuleListOptions",
+    "RegistryModulePermissions",
+    "RegistryModuleStatus",
+    "RegistryModuleUpdateOptions",
+    "RegistryModuleVCSRepo",
+    "RegistryModuleVCSRepoOptions",
+    "RegistryModuleVCSRepoUpdateOptions",
+    "RegistryModuleVersion",
+    "RegistryModuleVersionStatus",
+    "RegistryModuleVersionStatuses",
+    "RegistryName",
+    "Resource",
+    "Root",
+    "TestConfig",
+    "TerraformRegistryModule",
+    # Main types from types.py (will be dynamically added below)
+    "Capacity",
+    "DataRetentionPolicy",
+    "DataRetentionPolicyChoice",
+    "DataRetentionPolicyDeleteOlder",
+    "DataRetentionPolicyDeleteOlderSetOptions",
+    "DataRetentionPolicyDontDelete",
+    "DataRetentionPolicyDontDeleteSetOptions",
+    "DataRetentionPolicySetOptions",
+    "EffectiveTagBinding",
+    "Entitlements",
+    "ExecutionMode",
+    "LockedByChoice",
+    "Organization",
+    "OrganizationCreateOptions",
+    "OrganizationUpdateOptions",
+    "Pagination",
+    "Project",
+    "ReadRunQueueOptions",
+    "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
+types_py_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "types.py")
+spec = importlib.util.spec_from_file_location("main_types", types_py_path)
+if spec is not None and spec.loader is not None:
+    main_types = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(main_types)
+
+    # Re-export all main types
+    for name in dir(main_types):
+        if not name.startswith("_"):
+            globals()[name] = getattr(main_types, name)