Selenium，从图像对象中检索url的问题

Question

我正在使用我在旧工具（HP QTP）中进行测试的应用程序，现在我需要将其转换为selenium（python webdriver），我遇到了特定部分的问题，在我收到的测试流程中带有值的图像，（数据仅用于测试时是静态的），所以我知道应该为响应发送什么值。 在HP QTP中，我通过'if conditional'和loop（下面的VB代码）解决了它：

For Each elem As HtmlElement In WebBrowser1.Document.GetElementsByTagName("img")
     If elem.GetAttribute("src") = "http://URL/image_1.png" Then
     WebBrowser1.Document.GetElementById("response").SetAttribute("value", "70")
     End If
     If elem.GetAttribute("src") = "http://URL/image_2.png" Then
     WebBrowser1.Document.GetElementById("response").SetAttribute("value", "80")
     End If

我不知道它应该在python中看起来如何。我没有在Python中找到“elem.GetAttribute（”“）”的等效函数。 不幸的是我缺乏python的经验，所以我正在为这个问题寻找解决方案或提示。

Answer 1

python API中有一个get_attribute方法。您需要找到该元素并能够执行您在vb中执行的相同操作。请参阅this，here是一个如何操作的示例。