�
�Uc@sYdZd
dl
Z
d
dlZd
dlZd
dlZd
dlZddd��YZdS(s?wrapper for running display inside X virtual framebuffer (Xvfb)i����NtXvfbcBsYeZ
dd
dd�Zd�Zd�Zd�Zd�Zd�Zd	�Zd
�Z	RS(i i�icKs�|
|_||_
||_d
dd|j|j
|jfg|_x4|j�D]&\}}|jd||g|_qPWd|_dtjkr�tjdj	d�
d|_
n	d|_
dS(	Ns-screent
0s%dx%dx%ds-%stDISPLAYt
:i
i(twidththeightt
colordepthtxvfb_cmdtitemstNonetproctostenvirontsplittold_display_num(tselfRRRtkwargstkeytvalue((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyt__init__s
	
	
		

	

 c


Cs|j�
|S(
N(
tstart(
R((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyt	__enter__'s


c
Cs|j�
dS(
N(
tstop(Rtexc_typetexc_valtexc_tb((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyt__exit__+s
c
Cs�|j�|_
d
d|j
g|j|_tj|jdttj�
dttj�
�
|_t	j
d�

|jj�}
|
dkr�|j
|j
�

n|j
|j�

d|_dGHdS(NRs:%dtstdouttstderrg�������?sError: Xvfb did not start(tsearch_for_free_displaytvdisplay_numRt
subprocesstPopentopenRtdevnullR
ttimetsleeptpollR	t_redirect_displayR(Rtret_code((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyR.s








	
c

CsI|j|j
�

|jdk	rE|jj�
|jj�
d|_ndS(
N(R&RR
R	tkilltwait(
R((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyR?s







c
Cs�g|j�D],}
t
|
jd
�
djd�
d�
^q
}d}t|�
rmt|t|�
d�}n|}tj�
|tjdd�7}|S(Nt
Xi
t
-ii�id(t_lock_filestintR
tlentmaxtrandomtseedtrandint(Rt
xtlstmin_display_numtdisplay_num((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyRFs
?






c
Cs~d
}
d}tj
tj|
�
|�}g|D]}tjj|
|�^q.}g|D]}tjj|�
rV|^qV}|S(Ns/tmps.X*-lock(tfnmatchtfilterRtlistdirtpathtjointisfile(RttmpdirtpatterntnamestchildR4t
p((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyR,Qs



(
+
cCsd
|
tj
d<dS(Ns:%sR(RR(RR6((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyR&Ys
(
t__name__t
__module__RRRRRRR,R&(((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyRs						((t__doc__RR7R0RR#R(((s//usr/lib/python2.7/site-packages/xvfbwrapper.pyt<module>s