Add an optional parameter to hide elements from get_elements function

PyWizards · Jun 16, 2020 · bcc6ae6 · bcc6ae6
1 parent 2dbef0a
commit bcc6ae6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Screenshot/Screenshot_Clipping.py b/Screenshot/Screenshot_Clipping.py
@@ -116,7 +116,7 @@ def full_Screenshot(self, driver: WebDriver, save_path: str = '', image_name: st
             stitched_image.save(save_path)
             return save_path
 
-    def get_element(self, driver: WebDriver, element: WebElement, save_location: str) -> str:
+    def get_element(self, driver: WebDriver, element: WebElement, save_location: str, to_hide: list = None) -> str:
         """
          Usage:
              Capture Element screenshot as a image
@@ -129,7 +129,7 @@ def get_element(self, driver: WebDriver, element: WebElement, save_location: str
          Raises:
              N/A
          """
-        image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png')
+        image = self.full_Screenshot(driver, save_path=save_location, image_name='clipping_shot.png', elements=to_hide)
         location = element.location
         size = element.size
         x = location['x']