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)