def take_screenshot(url: str, id: str='world'):
"""
Takes a screenshot of DOM element given its id.
:param url: URL of webpage to open headlessly.
:param id: ID of DOM element.
:return: Image object.
"""
from PIL import Image
driver = get_webdriver()
driver.get(url)
svg = driver.find_element_by_id(id)
location = svg.location
size = svg.size
png = driver.get_screenshot_as_png()
driver.close()
image = Image.open(io.BytesIO(png))
left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']
image = image.crop((left, top, right, bottom))
return image
def test_html_viewer():
# Integration test for visualization service
num_nodes = 3
num_items = 10
options = textworld.GameOptions()
options.seeds = 1234
options.nb_rooms = num_nodes
options.nb_objects = num_items
options.quest_length = 3
options.grammar.theme = "house"
options.grammar.include_adj = True
game = textworld.generator.make_game(options)
game_name = "test_html_viewer_wrapper"
with make_temp_directory(prefix=game_name) as tmpdir:
game_file = compile_game(game, path=tmpdir)
env = textworld.start(game_file)
env = HtmlViewer(env, open_automatically=False, port=8080)
env.reset() # Cause rendering to occur.
# options.binary_location = "/bin/chromium"
driver = get_webdriver()
driver.get("http://127.0.0.1:8080")
nodes = driver.find_elements_by_class_name("node")
assert len(nodes) == num_nodes
items = driver.find_elements_by_class_name("item")
objects = [obj for obj in game.world.objects if obj.type != "I"]
assert len(items) == len(objects)
env.close()
driver.close()
def test_html_viewer():
# Integration test for visualization service
num_nodes = 3
num_items = 10
g_rng.set_seed(1234)
grammar_flags = {"theme": "house", "include_adj": True}
game = textworld.generator.make_game(world_size=num_nodes,
nb_objects=num_items,
quest_length=3,
grammar_flags=grammar_flags)
game_name = "test_html_viewer_wrapper"
with make_temp_directory(prefix=game_name) as tmpdir:
game_file = compile_game(game, game_name, games_folder=tmpdir)
env = textworld.start(game_file)
env = HtmlViewer(env, open_automatically=False, port=8080)
env.reset() # Cause rendering to occur.
# options.binary_location = "/bin/chromium"
driver = get_webdriver()
driver.get("http://127.0.0.1:8080")
nodes = driver.find_elements_by_class_name("node")
assert len(nodes) == num_nodes
items = driver.find_elements_by_class_name("item")
objects = [obj for obj in game.world.objects if obj.type != "I"]
assert len(items) == len(objects)
env.close()
driver.close()
def take_screenshot(url: str, id: str = 'world'):
"""
Takes a screenshot of DOM element given its id.
:param url: URL of webpage to open headlessly.
:param id: ID of DOM element.
:return: Image object.
"""
from PIL import Image
driver = get_webdriver()
if is_msys():
if (os.environ["TEMP"] == "/tmp"):
url = url.replace("file://", "file://%s" % msys_path())
else:
url = url.replace("/c/", "c:/")
url = url.replace("/d/", "d:/")
elif is_cygwin():
if (os.environ["TEMP"] == "/tmp"):
url = url.replace("file://", "file://%s" % msys_path())
else:
url = url.replace("/cygdrive/c/", "c:/")
url = url.replace("/cygdrive/d/", "d:/")
driver.get(url)
svg = driver.find_element_by_id(id)
location = svg.location
size = svg.size
png = driver.get_screenshot_as_png()
driver.close()
image = Image.open(io.BytesIO(png))
left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']
image = image.crop((left, top, right, bottom))
return image