Comment appliquer une boucle à travailler Python Selenium Script?

Question

J'essaie de comprendre comment appliquer une boucle for à ce script et j'ai beaucoup de problèmes. Je veux parcourir une liste de sous-domaines qui sont stockés au format csv (c'est-à-dire: une colonne avec 20 sous-domaines) et imprimer le code html pour chacun. Ils ont tous le même SourceDomain. Merci!

#Python 2.6 
from selenium import selenium 
import unittest, time, re, csv, logging 

class Untitled(unittest.TestCase): 
    def setUp(self): 
     self.verificationErrors = [] 
     self.selenium = selenium("localhost", 4444, "*firefox", "http://www.SourceDomain.com") 
     self.selenium.start() 

    def test_untitled(self): 
     sel = self.selenium 
     sel.open("/dns/www.subdomains.com.html") 
     sel.wait_for_page_to_load("30000") 
     html = sel.get_html_source() 
     print html 

    def tearDown(self): 
     self.selenium.stop() 
     self.assertEqual([], self.verificationErrors) 

if __name__ == "__main__": 
    unittest.main() 

Answer 1

#Python 2.6 
from selenium import selenium 
import unittest, time, re, csv, logging 

class Untitled(unittest.TestCase): 
    def setUp(self): 
     self.verificationErrors = [] 
     self.selenium = selenium("localhost", 4444, "*firefox", "http://www.SourceDomain.com") 
     self.selenium.start() 

    def test_untitled(self): 
     sel = self.selenium 
     spamReader = csv.reader(open('your_file.csv')) 
     for row in spamReader: 
      sel.open(row[0]) 
      sel.wait_for_page_to_load("30000") 
      print sel.get_html_source() 

    def tearDown(self): 
     self.selenium.stop() 
     self.assertEqual([], self.verificationErrors) 

if __name__ == "__main__": 
    unittest.main() 

BTW, notez qu'il n'y a pas besoin de placer ce script enveloppé dans un testcase unittest. Mieux encore, vous n'avez pas besoin de sélénium pour une tâche aussi simple (au moins à première vue).

Essayez ceci:

import urllib2, csv 

def fetchsource(url): 
    page = urllib2.urlopen(url) 
    source = page.read() 
    return source 

fooReader = csv.reader(open('your_file.csv')) 
for url in fooReader: 
    print fetchsource(url)