Merge pull request #209 from canonical/wd-20353

feat: Add function that allows injecting custom css

Author: petesfrench

CHANGELOG.md

@@ -1,3 +1,7 @@
+### 6.2.0 [28-04-2025]
+**Added** _inject_custom_css def
+A function that finds css directives (`[style=CLASSNAME]`) in the soup and applies to them to the next found element.
+
 ### 6.1.1 [12-03-2025]
 **Updated** EngagePages class
 Pass values for the provided keys, even if the values are empty or null, as they can be a filter themselves.

canonicalwebteam/discourse/parsers/base_parser.py

@@ -536,6 +536,7 @@ def _process_topic_soup(self, soup):
         soup = self._replace_polls(soup)
         soup = self._remove_trailing_numbers_from_headings(soup)
         soup = self._add_anchor_links(soup)
+        soup = self._inject_custom_css(soup)
 
         return soup
 
@@ -859,3 +860,29 @@ def _add_anchor_links(self, soup):
                 heading.append(anchor)
 
         return soup
+
+    def _inject_custom_css(self, soup):
+        """
+        Given HTML soup, finds style identifiers and applies the given
+        class to the element directly next.
+
+        Example:
+        [style=p-table--wide-table] applies the class 'p-table--wide-table'
+        """
+        custom_css_directives = soup.find_all(
+            string=re.compile(r"\[style=(.*?)\]")
+        )
+
+        for css_directive in custom_css_directives:
+            match = re.search(r"\[style=(.*?)\]", css_directive)
+            if match:
+                style_class = match.group(1)
+                next_element = css_directive.find_next_sibling()
+                if next_element:
+                    if "class" in next_element.attrs:
+                        next_element["class"].append(style_class)
+                    else:
+                        next_element["class"] = [style_class]
+            css_directive.extract()
+
+        return soup

setup.py

@@ -4,7 +4,7 @@
 
 setup(
     name="canonicalwebteam.discourse",
-    version="6.1.1",
+    version="6.2.0",
     author="Canonical webteam",
     author_email="webteam@canonical.com",
     url="https://github.com/canonical/canonicalwebteam.discourse",