while True:
if len(driver.window_handles) > 2:
driver.switch_to.window(driver.window_handles[2])
break
# driver.switch_to_window()
print("sleep...")
time.sleep(1)
wait = WebDriverWait(driver, 10)
cxn = wait.until(
ec.visibility_of_element_located(
(By.XPATH, "//div[@id=\"Cocos2dGameContainer\"]")))
while True:
img = take_screenshot_as_cv2img(driver)
roundId = Detector.get_round_id(img)
print("round: " + roundId)
if roundId:
folder = "./mydata/rounds/%s" % roundId
if not os.path.isdir(folder):
os.mkdir(folder)
else:
millis = int(round(time.time() * 1000))
file = "%s/%s.png" % (roundId, millis)
cv2.imwrite("%s/%s.jpg" % (folder, millis), img)
print("saved")
else:
cv2.imshow("debug", img)
cv2.waitKey(0)
