From f2547421ad96837f4ae876eef9cfee62748c6144 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B8=A9=E9=93=AD?= <3348431908@qq.com>
Date: Wed, 25 Sep 2024 11:48:55 +0000
Subject: [PATCH] =?UTF-8?q?update=20DrissionPage/=5Felements/chromium=5Fel?=
 =?UTF-8?q?ement.py.=20=E5=AE=8C=E5=96=84=E8=8E=B7=E5=8F=96xpath=E8=B7=AF?=
 =?UTF-8?q?=E5=BE=84=E9=80=BB=E8=BE=91=EF=BC=8C=E4=BF=9D=E8=AF=81=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E7=9A=84xpath=E8=B7=AF=E5=BE=84=E6=98=AF=E5=94=AF?=
 =?UTF-8?q?=E4=B8=80=E7=9A=84=20=E4=BC=98=E5=8C=96=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E7=9A=84css=E8=B7=AF=E5=BE=84=EF=BC=8C=E5=8E=BB=E9=99=A4?=
 =?UTF-8?q?=E5=A4=9A=E4=BD=99=E7=9A=84=E4=BC=AA=E7=B1=BB=E9=80=89=E6=8B=A9?=
 =?UTF-8?q?=E5=99=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 温铭 <3348431908@qq.com>
---
 DrissionPage/_elements/chromium_element.py | 30 +++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/DrissionPage/_elements/chromium_element.py b/DrissionPage/_elements/chromium_element.py
index 4e9c1dc..f10d0bb 100644
--- a/DrissionPage/_elements/chromium_element.py
+++ b/DrissionPage/_elements/chromium_element.py
@@ -640,8 +640,21 @@ class ChromiumElement(DrissionElement):
             txt1 = 'let tag = el.nodeName.toLowerCase();'
             txt3 = ''' && sib.nodeName.toLowerCase()==tag'''
             txt4 = '''
-            if(nth>1){path = '/' + tag + '[' + nth + ']' + path;}
-                    else{path = '/' + tag + path;}'''
+            let sib1 = el, chi = 0;
+        while (sib1) {
+            if (sib1.nodeType === Node.ELEMENT_NODE && sib1.nodeName.toLowerCase() === tag) {
+                chi += 1;
+            }
+            sib1 = sib1.nextSibling;
+        }
+        if (nth > 1) {
+            path = '/' + tag + '[' + nth + ']' + path;
+        } else if (chi > 1) {
+            path = '/' + tag + '[1]' + path;
+        } else {
+            path = '/' + tag + path;
+        }
+        '''
             txt5 = '''return path;'''
 
         elif mode == 'css':
@@ -651,7 +664,18 @@ class ChromiumElement(DrissionElement):
             break;}
             '''
             txt3 = ''
-            txt4 = '''path = '>' + el.tagName.toLowerCase() + ":nth-child(" + nth + ")" + path;'''
+            txt4 = '''let tag = el.nodeName.toLowerCase(), sib1 = el, chi = 0;
+        while (sib1) {
+            if (sib1.nodeType === Node.ELEMENT_NODE && sib1.nodeName.toLowerCase() === tag) {
+                chi += 1;
+            }
+            sib1 = sib1.nextSibling;
+        }
+        if (nth > 1 || chi > 1) {
+            path = '>' + el.tagName.toLowerCase() + ":nth-child(" + nth + ")" + path;
+        } else {
+            path = '>' + el.tagName.toLowerCase() + path;
+        }'''
             txt5 = '''return path.substr(1);'''
 
         else: