import logging from pyvirtualdisplay.abstractdisplay import AbstractDisplay log = logging.getLogger(__name__) PROGRAM = "Xvfb" class XvfbDisplay(AbstractDisplay): """ Xvfb wrapper Xvfb is an X server that can run on machines with no display hardware and no physical input devices. It emulates a dumb framebuffer using virtual memory. """ def __init__( self, size=(1024, 768), color_depth=24, bgcolor="black", use_xauth=False, fbdir=None, dpi=None, retries=10, extra_args=[], manage_global_env=True, ): """ :param bgcolor: 'black' or 'white' :param fbdir: If non-null, the virtual screen is memory-mapped to a file in the given directory ('-fbdir' option) :param dpi: screen resolution in dots per inch if not None """ self._screen = 0 self._size = size self._color_depth = color_depth self._bgcolor = bgcolor self._fbdir = fbdir self._dpi = dpi AbstractDisplay.__init__( self, PROGRAM, use_xauth=use_xauth, retries=retries, extra_args=extra_args, manage_global_env=manage_global_env, ) def _check_flags(self, helptext): pass def _cmd(self): cmd = [ dict(black="-br", white="-wr")[self._bgcolor], "-nolisten", "tcp", "-screen", str(self._screen), "x".join(map(str, list(self._size) + [self._color_depth])), ] if self._fbdir: cmd += ["-fbdir", self._fbdir] if self._dpi is not None: cmd += ["-dpi", str(self._dpi)] if self._has_displayfd: cmd += ["-displayfd", str(self._pipe_wfd)] else: cmd += [self.new_display_var] return [PROGRAM] + cmd