如何安装 xvfb 并使用 python 和 selenium 运行？

Question

我正在尝试使用 Xvfb 运行无头浏览器。到目前为止我遵循的过程
1.安装xvfbsudo apt-get install xvfb
2.创建了virtualenv， 3. 安装 xvfbwrapper
4.运行以下代码

import unittest

from selenium import webdriver
from xvfbwrapper import Xvfb


class TestPages(unittest.TestCase):

    def setUp(self):
        self.xvfb = Xvfb(width=1280, height=720)
        self.addCleanup(self.xvfb.stop)
        self.xvfb.start()
        self.browser = webdriver.Firefox()
        self.addCleanup(self.browser.quit)

    def testUbuntuHomepage(self):
        self.browser.get('http://www.ubuntu.com')
        self.assertIn('Ubuntu', self.browser.title)

    def testGoogleHomepage(self):
        self.browser.get('http://www.google.com')
        self.assertIn('Google', self.browser.title)


if __name__ == '__main__':
    unittest.main(verbosity=2)

但我收到以下错误，即使我尝试使用 sudo 安装它但没有效果。

Traceback (most recent call last):
  File "xvfbwrapper.py", line 4, in <module>
    from xvfbwrapper import Xvfb
  File "/home/ubuntu/unclescrooz/src/robinhood/xvfbwrapper.py", line 4, in <module>
    from xvfbwrapper import Xvfb
ImportError: cannot import name Xvfb

pyvirtualdisplay 也有同样的问题

Traceback (most recent call last):
  File "pyvirtualdisplay.py", line 1, in <module>
    from pyvirtualdisplay import Display
  File "/m4k/projects/scrapper/stock/robinhood/pyvirtualdisplay.py", line 1, in <module>
    from pyvirtualdisplay import Display
ImportError: cannot import name Display

如下代码

from pyvirtualdisplay import Display
from selenium import webdriver

display = Display(visible=0, size=(800, 600))
display.start()

browser = webdriver.Firefox()
browser.get('http://www.google.com')
print browser.title
browser.quit()

display.stop()

Answer 1

将您的文件 xvfbwrapper.py 和 pyvirtualdisplay.py 重命名为您导入的模块名称以外的名称。