def add_image(driver: WebDriver): """截图到报告 :param driver: :return: """ if "NATIVE_APP" in driver.context: AddImage(driver.get_screenshot_as_base64()) else: context = driver.context driver.switch_to.context(None) try: AddImage(driver.get_screenshot_as_base64()) except WebDriverException as e: logger().info(f"截图失败:\n{e}") finally: driver.switch_to.context(context)