def __init__(self):
if self.driver == None:
debug("WebDriver -> 初始化driver: %s" % DRIVER)
if DRIVER == 'Chrom':
self.driver = webdriver.Chrome()
else:
self.driver = webdriver.Firefox()
else:
debug("WebDriver -> %s driver已经实例化。" % DRIVER)
def open_and_check(self):
debug("PageObject -> %s打开并检测制定url页面")
self.driver.get(self.url)
return self.check_if_page_opened()
def __init__(self):
self.driver = driver
debug("PageObject -> %s 获取当前driver的实例：" % str(self.__class__))
def get_driver(self):
debug("WebDriver -> %s 获取并返回当前driver 实例。")
return self.driver
# -- encoding:utf-8 --
from utils.abs import Singleton
from config import DRIVER
from selenium import webdriver
from utils.logs import debug
@Singleton
class WebDriver():
driver = None
def __init__(self):
if self.driver == None:
debug("WebDriver -> 初始化driver: %s" % DRIVER)
if DRIVER == 'Chrom':
self.driver = webdriver.Chrome()
else:
self.driver = webdriver.Firefox()
else:
debug("WebDriver -> %s driver已经实例化。" % DRIVER)
def get_driver(self):
debug("WebDriver -> %s 获取并返回当前driver 实例。")
return self.driver
driver = WebDriver().get_driver()
debug("webdriver.py -> %s 实例化并返回当前driver 实例" % str(driver))
fwd_pos = env.front_pos
fwd_cell = env.grid.get(*fwd_pos)
# turn left
if action == "a":
obs, reward, done, _ = env.step(0)
# move forward
elif action == "s":
obs, reward, done, _ = env.step(2)
# turn right
else:
obs, reward, done, _ = env.step(1)
# add negative reward when agent go into lava
if fwd_cell is not None and fwd_cell.type == "lava" and action == "s":
reward = -1
env.render()
# print(obs)
print("\n" + "-" * 50, env.step_count, "-" * 50)
debug(f"Obs:\n{obs['image']}")
debug(f"Obs shape: {obs['image'].shape}")
debug(f"Direction: {obs['direction']}")
info(f"Mission: {obs['mission']}")
success(f"Reward: {reward}")
if done:
print("Completed with final reward of:", reward)
env.reset()
env.render()
def open_and_check(self, url):
debug("PageObject -> %s 打开并检测指定url页面")
self.driver.get(url)
def __init__(self):
self.driver = driver
debug("PageObject -> %s 获取当前driver的实例" % driver)