Grab class names from body and head

DeckThemes · Sep 26, 2023 · 41f6da7 · 41f6da7
1 parent 54599b3
commit 41f6da7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/css_browserhook.py b/css_browserhook.py
@@ -20,7 +20,7 @@ def __init__(self, browserHook, sessionId : str, targetInfo : dict):
         asyncio.create_task(self._init())
 
     async def _init(self):
-        res = await self.evaluate_js("(function(){ return {\"title\": document.title, \"classes\": Array.from(document.documentElement.classList)} })()")
+        res = await self.evaluate_js("(function(){ return {\"title\": document.title, \"classes\": Array.from(document.documentElement.classList).concat(Array.from(document.body.classList)).concat(Array.from(document.head.classList))} })()")
 
         if res != None:
             self.title = res["title"]