Examples working fine

Author: NimishaShrivastava-dev

examples/organization_token.py

@@ -21,13 +21,10 @@
 - Appropriate permissions to manage organization tokens
 """
 
-import os
-import sys
 from datetime import datetime, timedelta
 
 # Add the src directory to the path
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
-
+##sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
 from pytfe import TFEClient, TFEConfig
 from pytfe.models import (
     OrganizationTokenCreateOptions,
@@ -37,6 +34,28 @@
 )
 
 
+def redact_token(token_value: str | None) -> str:
+    """Redact token value for safe display."""
+    if not token_value:
+        return "None"
+    if len(token_value) <= 8:
+        return f"{'*' * len(token_value)}"
+    # Show first 3 and last 3 characters
+    return f"{token_value[:3]}...{token_value[-3:]}".replace(
+        token_value[3:-3], "*" * (len(token_value) - 6)
+    )
+
+
+def redact_id(id_value: str | None) -> str:
+    """Redact ID for safe display."""
+    if not id_value:
+        return "None"
+    if len(id_value) <= 6:
+        return f"{'*' * len(id_value)}"
+    # Show first 3 and last 3 characters
+    return f"{id_value[:3]}...{id_value[-3:]}"
+
+
 def main():
     """Execute organization token operation examples."""
 
@@ -46,8 +65,7 @@ def main():
 
     # Initialize the TFE client
     client = TFEClient(TFEConfig.from_env())
-    organization_name = "my-org"  # Change to your organization name
-
+    organization_name = "prab-sandbox01"
     # =====================================================
     # 1. CREATE ORGANIZATION TOKEN (BASIC)
     # =====================================================
@@ -58,16 +76,16 @@ def main():
         token = client.organization_tokens.create(organization_name)
 
         print("Token created successfully!")
-        print(f"  Token ID: {token.id}")
+        print(f"  Token ID: {redact_id(token.id)}")
         print(f"  Created At: {token.created_at}")
         print(f"  Description: {token.description}")
-        print(f"  Token Value: {token.token}")
+        print(f"  Token Value: {redact_token(token.token)}")
         if token.expired_at:
             print(f"  Expires At: {token.expired_at}")
         print()
 
     except Exception as e:
-        print(f"✗ Error: {e}")
+        print(f" Error: {e}")
         print()
 
     # =====================================================
@@ -86,7 +104,7 @@ def main():
         )
 
         print("Token created with options successfully!")
-        print(f"  Token ID: {token.id}")
+        print(f"  Token ID: {redact_id(token.id)}")
         print(f"  Created At: {token.created_at}")
         if token.expired_at:
             print(f"  Expires At: {token.expired_at}")
@@ -108,12 +126,12 @@ def main():
         )
 
         print(" Audit-trails token created successfully!")
-        print(f"  Token ID: {token.id}")
-        print(f"  Token Value: {token.token}")
+        print(f"  Token ID: {redact_id(token.id)}")
+        print(f"  Token Value: {redact_token(token.token)}")
         print()
 
     except Exception as e:
-        print(f"✗ Error: {e}")
+        print(f"Error: {e}")
         print()
 
     # =====================================================
@@ -124,7 +142,7 @@ def main():
         token = client.organization_tokens.read(organization_name)
 
         print("Token read successfully!")
-        print(f"  Token ID: {token.id}")
+        print(f"  Token ID: {redact_id(token.id)}")
         print(f"  Created At: {token.created_at}")
         print(f"  Description: {token.description}")
         if token.last_used_at:
@@ -134,7 +152,7 @@ def main():
         print()
 
     except Exception as e:
-        print(f"✗ Error: {e}")
+        print(f" Error: {e}")
         print()
 
     # =====================================================
@@ -147,12 +165,12 @@ def main():
         token = client.organization_tokens.read_with_options(organization_name, options)
 
         print(" Audit-trails token read successfully!")
-        print(f"  Token ID: {token.id}")
-        print(f"  Token Value: {token.token}")
+        print(f"  Token ID: {redact_id(token.id)}")
+        print(f"  Token Value: {redact_token(token.token)}")
         print()
 
     except Exception as e:
-        print(f"✗ Error: {e}")
+        print(f" Error: {e}")
         print()
 
     # =====================================================

src/pytfe/models/__init__.py

@@ -93,6 +93,15 @@
     OrganizationMembershipStatus,
     OrgMembershipIncludeOpt,
 )
+
+# ── Organization Token ────────────────────────────────────────────────────────
+from .organization_token import (
+    OrganizationToken,
+    OrganizationTokenCreateOptions,
+    OrganizationTokenDeleteOptions,
+    OrganizationTokenReadOptions,
+    TokenType,
+)
 from .policy import (
     Policy,
     PolicyCreateOptions,
@@ -287,7 +296,6 @@
 from .ssh_key import (
     SSHKey,
     SSHKeyCreateOptions,
-    SSHKeyList,
     SSHKeyListOptions,
     SSHKeyUpdateOptions,
 )
@@ -382,7 +390,6 @@
     # SSH keys
     "SSHKey",
     "SSHKeyCreateOptions",
-    "SSHKeyList",
     "SSHKeyListOptions",
     "SSHKeyUpdateOptions",
     # Reserved tag keys
@@ -486,6 +493,12 @@
     "OrganizationMembershipReadOptions",
     "OrganizationMembershipStatus",
     "OrgMembershipIncludeOpt",
+    # Organization tokens
+    "OrganizationToken",
+    "OrganizationTokenCreateOptions",
+    "OrganizationTokenDeleteOptions",
+    "OrganizationTokenReadOptions",
+    "TokenType",
     "OrganizationAccess",
     "Team",
     "TeamPermissions",

src/pytfe/models/organization_token.py

@@ -23,9 +23,9 @@ class OrganizationToken(BaseModel):
 
     id: str = Field(..., description="Organization token ID")
     created_at: datetime = Field(..., description="Creation timestamp")
-    description: str = Field("", description="Token description")
+    description: str | None = Field(None, description="Token description")
     last_used_at: datetime | None = Field(None, description="Last usage timestamp")
-    token: str = Field("", description="The actual token value")
+    token: str | None = Field(None, description="The actual token value")
     expired_at: datetime | None = Field(None, description="Token expiration timestamp")
     created_by: Any | None = Field(
         None, description="The entity that created this token"

src/pytfe/resources/organization_token.py

@@ -30,7 +30,7 @@ def create(self, organization: str) -> OrganizationToken:
         Raises:
             ValueError: If the organization name is invalid
         """
-        return self.create_with_options(organization, None)
+        return self.create_with_options(organization)
 
     def create_with_options(
         self,
@@ -57,7 +57,7 @@ def create_with_options(
         # Build request body
         body: dict[str, Any] = {
             "data": {
-                "type": "authentication-tokens",
+                "type": "authentication-token",
                 "attributes": {},
             }
         }