From 45998ec0888d9546867807cc0a3d767c8c2952f6 Mon Sep 17 00:00:00 2001
From: g1879 <g1879@qq.com>
Date: Tue, 24 Nov 2020 00:19:33 +0800
Subject: [PATCH] =?UTF-8?q?DriverElement=E5=A2=9E=E5=8A=A0remove=5Fattr()?=
 =?UTF-8?q?=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DrissionPage/driver_element.py | 14 ++++++++++++--
 README.en.md                   | 13 +++++++++++++
 README.zh-cn.md                | 13 +++++++++++++
 3 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/DrissionPage/driver_element.py b/DrissionPage/driver_element.py
index c659876..681a7a2 100644
--- a/DrissionPage/driver_element.py
+++ b/DrissionPage/driver_element.py
@@ -407,10 +407,20 @@ class DriverElement(DrissionElement):
         try:
             self.run_script(f"arguments[0].{attr} = '{value}';")
             return True
-        except Exception as e:
-            print(e)
+        except:
             return False
 
+    def remove_attr(self, attr: str) -> bool:
+        """设置元素属性          \n
+        :param attr: 属性名
+        :return: 是否设置成功
+        """
+        try:
+            self.run_script(f'arguments[0].removeAttribute("{attr}");')
+            return True
+        except:
+            raise False
+
     def drag(self, x: int, y: int, speed: int = 40, shake: bool = True) -> bool:
         """拖拽当前元素到相对位置                   \n
         :param x: x变化值
diff --git a/README.en.md b/README.en.md
index f8f6896..3cd51b3 100644
--- a/README.en.md
+++ b/README.en.md
@@ -689,6 +689,7 @@ element.clear()  # Clear the element
 element.screenshot(path, filename)  # Take a screenshot of the element
 element.select(text)  # Select the drop- down list based on the text
 element.set_attr(attr, value)  # Set element attribute value
+element.remove_attr(attr)  # remove a element attribute
 element.drag(x, y, speed, shake)  # Drag the relative distance of the element, you can set the speed and whether to shake randomly
 element.drag_to(ele_or_loc, speed, shake)  # Drag the element to another element or a certain coordinate, you can set the speed and whether to shake randomly
 element.hover()  # Hover the mouse over the element
@@ -2095,6 +2096,18 @@ Returns: bool  - whether it was successful
 
 
 
+### remove_attr()
+
+Remove element attributes.
+
+Parameter Description:
+
+- attr: str     -parameter name
+
+Returns: bool  - whether it was successful
+
+
+
 ### drag()
 
 Drag the current element a certain distance, and return whether the drag is successful.
diff --git a/README.zh-cn.md b/README.zh-cn.md
index 0710bde..99ac7f2 100644
--- a/README.zh-cn.md
+++ b/README.zh-cn.md
@@ -689,6 +689,7 @@ element.clear()  # 清空元素
 element.screenshot(path, filename)  # 对元素截图
 element.select(text)  # 根据文本选择下拉列表
 element.set_attr(attr, value)  # 设置元素属性值
+element.remove_attr(attr)  # 删除属性
 element.drag(x, y, speed, shake)  # 拖动元素相对距离，可设置速度和是否随机抖动
 element.drag_to(ele_or_loc, speed, shake)  # 拖动元素到另一个元素或某个坐标，可设置速度和是否随机抖动
 element.hover()  # 在元素上悬停鼠标
@@ -2075,6 +2076,18 @@ driver 模式的元素对象，包装了一个 WebElement 对象，并封装了
 
 
 
+### remove_attr()
+
+删除元素属性。
+
+参数说明：
+
+- attr: str     - 参数名
+
+返回： bool  -是否成功
+
+
+
 ### drag()
 
 拖拽当前元素一段距离，返回是否拖拽成功。