| Server IP : 109.234.164.53 / Your IP : 216.73.216.110 Web Server : Apache System : Linux cervelle.o2switch.net 4.18.0-553.32.1.lve.el8.x86_64 #1 SMP Thu Dec 19 13:14:03 UTC 2024 x86_64 User : computer3 ( 1098) PHP Version : 7.1.33 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /opt/alt/python27/lib/python2.7/site-packages/pyvirtualdisplay/ |
Upload File : |
from pyvirtualdisplay.display import Display
from PIL import Image
from PIL import ImageChops
import logging
import pyscreenshot
import time
log = logging.getLogger(__name__)
# class DisplayError(Exception):
# pass
class DisplayTimeoutError(Exception):
pass
class SmartDisplay(Display):
pyscreenshot_backend = None
pyscreenshot_childprocess = True
def autocrop(self, im):
'''Crop borders off an image.
:param im: Source image.
:param bgcolor: Background color, using either a color tuple or a color name (1.1.4 only).
:return: An image without borders, or None if there's no actual content in the image.
'''
if im.mode != "RGB":
im = im.convert("RGB")
bg = Image.new("RGB", im.size, self.bgcolor)
diff = ImageChops.difference(im, bg)
bbox = diff.getbbox()
if bbox:
return im.crop(bbox)
return None # no contents
def grab(self, autocrop=True):
try:
# first try newer pyscreenshot version
img = pyscreenshot.grab(
childprocess=self.pyscreenshot_childprocess,
backend=self.pyscreenshot_backend,
)
except TypeError:
# try older pyscreenshot version
img = pyscreenshot.grab()
if autocrop:
img = self.autocrop(img)
return img
def waitgrab(self, timeout=60, autocrop=True, cb_imgcheck=None):
'''start process and create screenshot.
Repeat screenshot until it is not empty and
cb_imgcheck callback function returns True
for current screenshot.
:param autocrop: True -> crop screenshot
:param timeout: int
:param cb_imgcheck: None or callback for testing img,
True = accept img,
False = reject img
'''
t = 0
sleep_time = 0.3 # for fast windows
repeat_time = 1
while 1:
log.debug('sleeping %s secs' % str(sleep_time))
time.sleep(sleep_time)
t += sleep_time
img = self.grab(autocrop=autocrop)
if img:
if not cb_imgcheck:
break
if cb_imgcheck(img):
break
sleep_time = repeat_time
repeat_time += 1 # progressive
if t > timeout:
msg = 'Timeout! elapsed time:%s timeout:%s ' % (t, timeout)
raise DisplayTimeoutError(msg)
break
log.debug('screenshot is empty, next try..')
assert img
# if not img:
# log.debug('screenshot is empty!')
return img