| import os |
| import time |
| from selenium import webdriver |
| import utils as U |
|
|
| class VisualAPI: |
| def __init__(self): |
| self.driver = webdriver.Chrome() |
|
|
| self.driver.set_window_size(320, 512) |
|
|
| def run(self): |
| self.driver.get("http://localhost:9000") |
| IMAGE_DIR = 'Adam/game_image' |
| U.f_mkdir(IMAGE_DIR) |
| if not os.path.exists(IMAGE_DIR): |
| os.makedirs(IMAGE_DIR) |
| print('Visual API Ready') |
| while True: |
| screenshot_path = os.path.join(IMAGE_DIR, 'tmp.png') |
| self.driver.save_screenshot(screenshot_path) |
| time.sleep(10) |
|
|
| def stop(self): |
| self.driver.quit() |
|
|
|
|
| module = VisualAPI() |
| module.run() |
|
|
