在屏幕截图名称中附加当前 url

【问题标题】：Append current url in screenshot name在屏幕截图名称中附加当前 url
【发布时间】：2014-08-16 10:50:59
【问题描述】：

我需要在屏幕截图文件名中添加当前 url。

代码如下：

从硒导入网络驱动程序

driver = webdriver.Firefox()
driver.get("http://www.google.com")
driver.save_screenshot("image.png") # save a screenshot to disk
print driver.current_url
driver.quit()

如何将 image.png 替换为 driver.current_url？

【问题讨论】：

标签： python selenium-webdriver python-unittest

【解决方案1】：

对于您问题中给出的具体示例：

driver.save_screenshot(driver.current_url.replace('://','_'))

更一般地说，由于您的问题源于特殊字符，您可以执行以下操作：

chars = [':','/','%'] # add more if needed
image_name = driver.current_url
for char in chars:
    image_name = image_name.replace(char,'_')
driver.save_screenshot(image_name)

【讨论】：

谢谢！！工作没有问题