fix(problemset): 修复元素的查找

XYShaoKang · Sep 20, 2023 · 253ec7d · 253ec7d
1 parent 128e855
commit 253ec7d
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/content/pages/problemset/Rank.tsx b/src/content/pages/problemset/Rank.tsx
@@ -98,10 +98,15 @@ const Rank: FC<{ enable: boolean }> = ({ enable }) => {
     dispatch(enableProblemRating(currentPage!))
   }
   async function handleSetTableEl() {
-    const tableEl = await findElementByXPath('//div[@role="table"]')
+    const tableEl = await findElementByXPath(
+      {
+        xpath: '//div[@role="table"]',
+        nodeType: 'UNORDERED_NODE_ITERATOR_TYPE',
+      },
+      els => !!els.filter(el => el.childElementCount > 0)
+    )
     if (!mountState.isMount) return
-
-    setTableEl(tableEl)
+    setTableEl(tableEl.find(el => el.childElementCount > 0))
   }
 
   useEffect(() => {