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()