DOP-3948

mut/index/SnootyManifest.py

@@ -12,6 +12,23 @@
 logger = logging.getLogger(__name__)
 
 
+class Facet:
+    def __init__(self, category: str, value: str, sub_facets: [List[object]]) -> None:
+        self.category = category
+        self.value = value
+        self.sub_facets = None
+
+        for sub_facet in sub_facets:
+            self.sub_facets = self.sub_facets or []
+            self.sub_facets.append(
+                Facet(
+                    category=sub_facet["category"],
+                    value=sub_facet["value"],
+                    sub_facets=sub_facet["sub_facets"] or [],
+                )
+            )
+
+
 class ManifestEntry(TypedDict):
     slug: str
     title: Optional[str]
@@ -20,14 +37,13 @@ class ManifestEntry(TypedDict):
     code: dict
     preview: Optional[str]
     tags: List[str]
-    facets: Optional[Dict[str, Any]]
+    facets: Optional[Dict[str, List[str]]]
 
 
 class Document:
     """Return indexing data from a page's AST for search purposes."""
 
     def __init__(self, data) -> None:
-
         self.tree = data[0]
 
         self.robots, self.keywords, self.description = self.find_metadata()
@@ -125,13 +141,41 @@ def derive_preview(self) -> Optional[str]:
             return None
 
     def derive_facets(self) -> Optional[Dict[str, Any]]:
-        logger.debug("finding facets")
+        """
+        Format facets for ManifestEntry from bson entry tree['facets'] if it exists
+        """
+        document_facets: Dict[str, List[str]] = {}
+        HIERARCHY_KEY = ">"
+
+        # recursive function to look within facets.sub_facets<facets[]>
+        def insert_key_values(facet: Facet, prefix: str = ""):
+            key = prefix + facet.category
+            document_facets[key] = document_facets.get(key, [])
+            document_facets[key].append(facet.value)
+            if not facet.sub_facets:
+                return
+            for sub_facet in facet.sub_facets:
+                insert_key_values(
+                    sub_facet, key + HIERARCHY_KEY + facet.value + HIERARCHY_KEY
+                )
+
+        def create_facet(facet_entry: object):
+            facet = Facet(
+                category=facet_entry["category"],
+                value=facet_entry["value"],
+                sub_facets=facet_entry["sub_facets"] or [],
+            )
+            insert_key_values(facet)
+
         try:
             if self.tree["facets"]:
-                return self.tree["facets"]
+                for facet_entry in self.tree["facets"]:
+                    create_facet(facet_entry)
         except KeyError:
             return None
 
+        return document_facets
+
     def find_metadata(self):
         logger.debug("Finding metadata")
         robots: str = True

mut/index/test_SnootyManifest.py

@@ -147,5 +147,10 @@ def test_generate_manifest() -> None:
 def test_derive_facets() -> None:
     # Test facets derived from page
     document = setup_doc(ROOT_PATH, "facet_example.bson")
-    expected = {"genres": [{"name": "reference"}]}
+    expected = {
+        "programming_language": ["cpp"],
+        "genre": ["reference"],
+        "target_product": ["atlas"],
+        "target_product>atlas>sub_product": ["search"],
+    }
     assert document["facets"] == expected