Improve IAM policy title abbreviation to prevent duplicates

- Enhance title_abbrev generation by appending source context (e.g. "from AWS Account Management")
- Add test coverage for the updated title abbreviation logic
- Add test resources for IAMPolicyGrammar configuration

This reduces duplicate title abbreviations that occur when multiple policies have similar names.

Author: cpyle0819

aws_doc_sdk_examples_tools/agent/bin/main.py

@@ -25,11 +25,8 @@
 
 
 def format_duration(seconds: float) -> str:
-    """Format duration in seconds to hours:minutes:seconds format."""
     td = timedelta(seconds=seconds)
-    hours, remainder = divmod(td.total_seconds(), 3600)
-    minutes, seconds = divmod(remainder, 60)
-    return f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d}"
+    return str(td).zfill(8)
 
 
 @app.command()

aws_doc_sdk_examples_tools/agent/process_ailly_files.py

@@ -81,7 +81,6 @@ def parse_ailly_file(
             if key in result:
                 result[key] = f"{prefix}{result[key]}"
 
-        result["title_abbrev"] = result["title"]
         result["id"] = Path(file_path).name.split(".md.ailly.md")[0]
         result["_source_file"] = file_path
 

aws_doc_sdk_examples_tools/agent/update_doc_gen.py

@@ -8,6 +8,8 @@
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+IAM_LANGUAGE = "IAMPolicyGrammar"
+
 
 def examples_from_updates(updates_path: Path) -> Iterable[Example]:
     """
@@ -44,7 +46,9 @@ def update_examples(doc_gen: DocGen, examples: Iterable[Example]) -> None:
     for example in examples:
         if doc_gen_example := doc_gen.examples.get(example.id):
             doc_gen_example.title = example.title
-            doc_gen_example.title_abbrev = example.title_abbrev
+            # This reduces the number of duplicate title_abbrev that occur due to similar policies
+            source = doc_gen_example.languages[IAM_LANGUAGE].versions[0].source.title
+            doc_gen_example.title_abbrev = f"{example.title_abbrev} (from {source})"
             doc_gen_example.synopsis = example.synopsis
         else:
             logger.warning(f"Could not find example with id: {example.id}")

aws_doc_sdk_examples_tools/agent/update_doc_gen_test.py

@@ -0,0 +1,41 @@
+import pytest
+from pathlib import Path
+
+from aws_doc_sdk_examples_tools.doc_gen import DocGen
+from aws_doc_sdk_examples_tools.metadata import Example
+from aws_doc_sdk_examples_tools.agent.update_doc_gen import update_examples
+
+
+@pytest.fixture
+def doc_gen_tributary():
+    """
+    Fixture that returns a DocGen instance using the doc_gen_tributary_test as root.
+    """
+    tributary_root = (
+        Path(__file__).parent.parent / "test_resources" / "doc_gen_tributary_test"
+    )
+    doc_gen = DocGen.from_root(tributary_root)
+    doc_gen.collect_snippets()
+    return doc_gen
+
+
+def smoke_test_doc_gen(doc_gen_tributary: DocGen):
+    assert isinstance(doc_gen_tributary, DocGen)
+
+
+def test_update_examples_title_abbrev(doc_gen_tributary: DocGen):
+    """Test that title_abbrev is updated correctly with service_main suffix."""
+    # Create an example with a title_abbrev to update
+    update_example = Example(
+        id="iam_policies_example",
+        file=None,
+        languages={},
+        title_abbrev="Updated Title Abbrev",
+    )
+
+    # Update the examples
+    update_examples(doc_gen_tributary, [update_example])
+
+    # Verify title_abbrev was updated with the service_main suffix
+    updated_example = doc_gen_tributary.examples["iam_policies_example"]
+    assert updated_example.title_abbrev == "Updated Title Abbrev (from AWS Account Management)"

aws_doc_sdk_examples_tools/test_resources/doc_gen_tributary_test/.doc_gen/config/sdks.yaml

@@ -11,3 +11,16 @@ IAMPolicy:
         name: "&SAZR;"
         link_template: "SomeTemplate"
   guide: "&guide-iam-user;"
+IAMPolicyGrammar:
+    property: json
+    syntax: json
+    sdk:
+      1:
+        long: "IAM policy"
+        short: "IAM policy"
+        guide: "IAM/latest/UserGuide/introduction.html"
+        api_ref:
+          uid: "IAMPolicy"
+          name: "&SAZR;"
+          link_template: "SomeTemplate"
+    guide: "&guide-iam-user;"

aws_doc_sdk_examples_tools/test_resources/doc_gen_tributary_test/.doc_gen/metadata/AccountControlApiDoc_metadata.yaml

@@ -0,0 +1,21 @@
+iam_policies_example:
+  category: IAMPolicy
+  languages:
+    IAMPolicyGrammar:
+      versions:
+      - authors:
+        - alias: amazonian@amazon.com
+          name: Sir Peccy
+        excerpts:
+          - description: test
+        owner: AWS/Documentation/Accounts Management Docs
+        sdk_version: 1
+        source:
+          title: AWS Account Management
+          url: https://code.amazon.com/packages/AccountControlApiDoc
+  services:
+    iam: {}
+  synopsis: This identity-based policy allows the attached identity to retrieve the
+    billing alternate contact information for a specific account within an organization.
+  title: Allow retrieval of a specific alternate contact type for an account
+  title_abbrev: Allow retrieval of a specific alternate contact type for an account