diff --git a/dryscrape/xvfb.py b/dryscrape/xvfb.py index 6b27ca1..00f7b87 100644 --- a/dryscrape/xvfb.py +++ b/dryscrape/xvfb.py @@ -3,15 +3,15 @@ _xvfb = None + def start_xvfb(): from xvfbwrapper import Xvfb global _xvfb - if "DISPLAY" in os.environ: - del os.environ["DISPLAY"] _xvfb = Xvfb() _xvfb.start() atexit.register(_xvfb.stop) + def stop_xvfb(): global _xvfb _xvfb.stop()