doc_gen_cli collects and stores snippets in a second file

aws_doc_sdk_examples_tools/doc_gen_cli.py

@@ -7,7 +7,6 @@
 import logging
 
 from .doc_gen import DocGen, DocGenEncoder
-from .entities import EntityErrors
 
 logging.basicConfig(level=logging.INFO)
 
@@ -28,6 +27,12 @@ def main():
         type=str,
         help="Output a JSON version of the computed DocGen with some properties stripped out. Includes any errors.",
     )
+    parser.add_argument(
+        "--write-snippets",
+        default="doc_gen_snippets.json",
+        type=str,
+        help="Output a JSON version of the computed DocGen with only snippets and snippet files. Separates snippet content from metadata content.",
+    )
 
     parser.add_argument(
         "--strict",
@@ -61,6 +66,20 @@ def main():
     with open(args.write_json, "w") as out:
         out.write(serialized)
 
+    if args.write_snippets:
+        for root in args.from_root:
+            merged_doc_gen.collect_snippets(Path(root))
+
+        serialized_snippets = json.dumps(
+            {
+                "snippets": merged_doc_gen.snippets,
+                "snippet_files": merged_doc_gen.snippet_files,
+            },
+            cls=DocGenEncoder,
+        )
+        with open(args.write_snippets, "w") as out:
+            out.write(serialized_snippets)
+
 
 if __name__ == "__main__":
     main()

aws_doc_sdk_examples_tools/doc_gen_cli_test.py

@@ -76,6 +76,7 @@ def test_doc_gen_strict_option(strict, should_raise, patched_environment):
     mock_args = Namespace(
         from_root=["/mock/path"],
         write_json="mock_output.json",
+        write_snippets="",
         strict=strict,
         skip_entity_expansion=False,
     )
@@ -94,6 +95,7 @@ def test_skip_entity_expansion(patched_environment):
     mock_args = Namespace(
         from_root=["/mock/path"],
         write_json="mock_output.json",
+        write_snippets="",
         strict=False,
         skip_entity_expansion=True,
     )
@@ -111,6 +113,7 @@ def test_default_entity_expansion(patched_environment):
     mock_args = Namespace(
         from_root=["/mock/path"],
         write_json="mock_output.json",
+        write_snippets="",
         strict=False,
         skip_entity_expansion=False,
     )