（我快到了 - 我想）用Selenium循环遍历子域名列表

Question

我想从一个基本的网址开始，用selenium这个工具循环遍历一个短的子域名列表，这个列表是以csv格式存储的（也就是说，里面有一列20个子域名），然后打印出每个子域名的html内容。不过我在这个过程中遇到了一些困难。谢谢！

from selenium import selenium
import unittest, time, re, csv, logging

subds = csv.reader(open('listofsubdomains.txt', 'rb'))
for subd in subds:
        try:
            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(subd[0])
                    html = sel.get_html_source()
                    print html

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

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

        except Exception, e:
            print>>sys.stderr, "Url % not processed: error (%s) % (url, e)"

Answer 1

你在类里面一次又一次地定义同一个函数。其实在 unittest.main() 开始之前，整个类已经创建好了，所以在这个类里只会保留一个测试方法。