/usr/lib/python2.7/dist-packages/xvfbwrapper.py is in python-xvfbwrapper 0.2.4-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | #!/usr/bin/env python
#
# * Corey Goldberg, 2012, 2013
#
# * inspired by:
# PyVirtualDisplay: http://pypi.python.org/pypi/PyVirtualDisplay
"""wrapper for running display inside X virtual framebuffer (Xvfb)"""
import os
import fnmatch
import random
import subprocess
import time
class Xvfb:
def __init__(self, width=800, height=680, colordepth=24, **kwargs):
self.width = width
self.height = height
self.colordepth = colordepth
self.xvfb_cmd = [
'-screen', '0', '%dx%dx%d' %
(self.width, self.height, self.colordepth)
]
for key, value in kwargs.items():
self.xvfb_cmd = self.xvfb_cmd + ['-%s' % key, value]
self.proc = None
if 'DISPLAY' in os.environ:
self.old_display_num = os.environ['DISPLAY'].split(':')[1]
else:
self.old_display_num = 0
def __enter__(self):
self.start()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.stop()
def start(self):
self.vdisplay_num = self.search_for_free_display()
self.xvfb_cmd = ['Xvfb', ':%d' % self.vdisplay_num] + self.xvfb_cmd
with open(os.devnull) as devnull:
self.proc = subprocess.Popen(self.xvfb_cmd,
stdout=devnull,
stderr=devnull,
)
time.sleep(0.2) # give Xvfb time to start
ret_code = self.proc.poll()
if ret_code is None:
self._redirect_display(self.vdisplay_num)
else:
self._redirect_display(self.old_display_num)
self.proc = None
print('Error: Xvfb did not start')
def stop(self):
self._redirect_display(self.old_display_num)
if self.proc is not None:
self.proc.kill()
self.proc.wait()
self.proc = None
def search_for_free_display(self):
ls = [int(x.split('X')[1].split('-')[0]) for x in self._lock_files()]
min_display_num = 1000
if len(ls):
display_num = max(min_display_num, max(ls) + 1)
else:
display_num = min_display_num
random.seed()
display_num += random.randint(0, 100)
return display_num
def _lock_files(self):
tmpdir = '/tmp'
pattern = '.X*-lock'
names = fnmatch.filter(os.listdir(tmpdir), pattern)
ls = [os.path.join(tmpdir, child) for child in names]
ls = [p for p in ls if os.path.isfile(p)]
return ls
def _redirect_display(self, display_num):
os.environ['DISPLAY'] = ':%s' % display_num
|