Selenium，從圖像對象檢索url問題

Question

我正在使用我的舊工具（HP QTP）測試過的應用程序，現在我需要將它轉換爲硒（python webdriver），我遇到了問題特定的部分，在測試流程期間，我接收具有該值的圖像（數據僅用於測試而爲靜態），因此我知道應該發送什麼值作爲響應。 在HP QTP我解決了它的「如果有條件」和環形（以下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中找到「elem.GetAttribute（」「）」的等價函數。 不幸的是，我缺乏Python的經驗，所以我正在尋找解決方案或提示這個問題。

Answer 1

python API中有一個get_attribute方法可用。你需要找到元素，並能夠執行你在vb中做的同樣的動作。請參閱this和here是一個示例。