def play_game(self, challenge=None):
self.state.update_state("status", "finding_game")
self.logger.info("Finding a game...")
self.selenium.screenshot('log.png')
tier_click = False
while not tier_click:
try:
self.selenium.choose_tier()
self.selenium.screenshot('log.png')
if challenge:
self.selenium.start_challenge_battle(challenge)
else:
self.selenium.start_ladder_battle()
self.selenium.screenshot('log.png')
tier_click = True
except TierException:
self.logger.warning("Unable to click tier. Trying again...")
self.selenium.screenshot('log.png')
self.battle_url = self.selenium.driver.current_url
self.selenium.screenshot('log.png')
self.logger.info("Found game: %s", self.battle_url)
self.state.update_state("status", "in_battle")
self.state.update_state("battle_url", self.battle_url)
#self.update_monitor()
self.selenium.wait_for_move()
#self.selenium.chat("gl hf!")
self.selenium.switch_initial(0, 0)
self.selenium.screenshot('log.png')
gamestate = self.create_initial_gamestate()
gamestate = self.update_latest_turn(gamestate)
log = SimulatorLog.parse(self.selenium.get_log())
over, _ = log.is_over()
while not over:
print(
"=========================================================================================="
)
print("My primary:", gamestate.get_team(0).primary())
print("Their primary:", gamestate.get_team(1).primary())
print("Their moves: ",
gamestate.get_team(1).primary().moveset.moves)
print("Their item: ", gamestate.get_team(1).primary().item)
print("Their ability: ",
gamestate.get_team(1).primary().ability)
move = self.agent.get_action(gamestate, 0)
print("My move:", move)
if move.is_switch():
self.selenium.switch(move.switch_index, move.backup_switch)
else:
print(move.move_index)
self.selenium.move(move.move_index,
move.backup_switch,
mega=move.mega,
volt_turn=move.volt_turn)
gamestate = self.update_latest_turn(gamestate)
gamestate = self.correct_gamestate(gamestate)
self.selenium.screenshot('log.png')
self.state.delete_state("battle_url")
def create_initial_gamestate(self):
self.logger.info("Creating initial gamestate...")
my_pokes = self.my_team.copy()
for i, poke in enumerate(my_pokes.poke_list):
poke_name = poke.name
if poke_name in NAME_CORRECTIONS:
poke_name = NAME_CORRECTIONS[poke_name]
poke.health = poke.final_stats['hp']
poke.alive = True
opp_poke_list = []
log = SimulatorLog.parse(self.selenium.get_log())
for event in log.events:
if event.type == "team" and event.details[
'username'] != self.username:
opp_poke_names = event.details['team']
for name in opp_poke_names:
if not name:
continue
poke_name = correct_name(name)
"Corrected to:", poke_name
if poke_name in self.smogon_data:
moveset = [
m for m in self.smogon_data[poke_name].movesets
if 'Overused' == m['tag'] or 'Underused' == m['tag']
or 'Rarelyused' == m['tag'] or 'Neverused' == m['tag']
or 'Unreleased' == m['tag'] or 'Ubers' == m['tag']
or 'PU' in m['tag']
]
if len(moveset) > 1:
moveset = SmogonMoveset.from_dict(moveset[1])
elif len(moveset) == 1:
moveset = SmogonMoveset.from_dict(moveset[0])
else:
moveset = [
m for m in self.smogon_bw_data[poke_name].movesets
if 'Overused' == m['tag'] or 'Underused' == m['tag']
or 'Rarelyused' == m['tag'] or 'Neverused' == m['tag']
or 'Unreleased' == m['tag'] or 'Ubers' == m['tag']
or 'PU' in m['tag']
]
moveset = SmogonMoveset.from_dict(moveset[0])
elif poke_name not in self.smogon_data and poke_name in self.smogon_bw_data:
moveset = [
m for m in self.smogon_bw_data[poke_name].movesets
if 'Overused' == m['tag'] or 'Underused' == m['tag']
or 'Rarelyused' == m['tag'] or 'Neverused' == m['tag']
or 'Unreleased' == m['tag'] or 'Ubers' == m['tag']
or 'PU' in m['tag']
]
moveset = SmogonMoveset.from_dict(moveset[0])
else:
moveset = SmogonMoveset(
None, None, None, {
'hp': 88,
'patk': 84,
'pdef': 84,
'spatk': 84,
'spdef': 84,
'spe': 84
}, {
'hp': 1.0,
'patk': 1.0,
'pdef': 1.0,
'spatk': 1.0,
'spdef': 1.0,
'spe': 1.0
}, None, 'ou')
moveset.moves = None
if poke_name in self.smogon_data:
typing = self.smogon_data[poke_name].typing
stats = self.smogon_data[poke_name].stats
elif poke_name not in self.smogon_data and poke_name in self.smogon_bw_data:
typing = self.smogon_bw_data[poke_name].typing
stats = self.smogon_bw_data[poke_name].stats
else:
typing = ['Normal']
stats = {
'hp': 80,
'patk': 80,
'pdef': 80,
'spatk': 80,
'spdef': 80,
'spe': 80
}
predictor = create_predictor(self.predictor_name, name,
self.pokedata)
poke = Pokemon(name,
typing,
stats,
moveset,
predictor,
calculate=True)
moves = [x[0] for x in poke.predict_moves([])]
poke.moveset.moves = moves[:4]
poke.health = poke.final_stats['hp']
poke.alive = True
opp_poke_list.append(poke)
my_primary = None
for event in log.events:
if event.type == "switch" and event.player == 0:
for poke in my_pokes.poke_list:
if poke.name == event.poke:
my_primary = my_pokes.poke_list.index(poke)
elif event.type == "switch" and event.player == 1:
for poke in opp_poke_list:
if poke.name == event.poke:
opp_primary = opp_poke_list.index(poke)
assert my_primary != None
self.opp_team = Team(opp_poke_list)
opp_pokes = self.opp_team.copy()
my_pokes.primary_poke = my_primary
opp_pokes.primary_poke = opp_primary
gamestate = GameState([my_pokes, opp_pokes])
return gamestate
def run(self, num_games=1, challenge=None):
self.state.update_state("status", "initializing_battle")
if challenge:
self.logger.info("Set to challenge: %s", challenge)
else:
self.logger.info("Set to play %u games", num_games)
self.init()
self.scores = {'wins': 0, 'losses': 0, 'crashes': 0}
def signal_handler(signal, frame):
self.update_monitor(done=True)
sys.exit(0)
#signal.signal(signal.SIGINT, signal_handler)
for i in range(num_games):
self.simulator.log.reset()
result, error = None, None
try:
self.play_game(challenge=challenge)
except GameOverException:
log = SimulatorLog.parse(self.selenium.get_log())
disconnected = log.disconnected()
if disconnected:
over, _ = log.is_over()
while not over:
time.sleep(5)
over, _ = log.is_over()
_, over_event = log.is_over()
result = over_event.details['username'] == self.username
except UserNotOnlineException:
self.logger.error("User not online: %s", challenge)
self.logger.info("Exiting...")
return
except:
error = traceback.format_exc()
print "Error", error
log = SimulatorLog.parse(self.selenium.get_log())
_, over_event = log.is_over()
if over_event is not None:
result = over_event.details['username'] == self.username
log = self.selenium.get_log()
id = self.selenium.get_battle_id()
battle_url = "http://replay.pokemonshowdown.com/battle-%s" % id
self.logger.info("Finished game! Replay can be found at: %s",
battle_url)
user_folder = Path(".") / self.username
if not user_folder.exists():
user_folder.mkdir()
if not (user_folder / "wins").exists():
(user_folder / "wins").mkdir()
if not (user_folder / "losses").exists():
(user_folder / "losses").mkdir()
if not (user_folder / "crashes").exists():
(user_folder / "crashes").mkdir()
if result == True:
print "---------------"
print "Won the battle! - %s" % battle_url
print "---------------"
self.scores['wins'] += 1
with open(user_folder / "wins" / ("%s.log" % id), 'w') as fp:
fp.write(log)
with open(user_folder / "wins" / ("%s.score" % id), 'w') as fp:
print >> fp, self.simulator.score
print >> fp, self.simulator.total
elif result == False:
print "---------------"
print "Lost the battle! - %s" % battle_url
print "---------------"
self.scores['losses'] += 1
with open(user_folder / "losses" / ("%s.log" % id), 'w') as fp:
fp.write(log)
with open(user_folder / "losses" / ("%s.score" % id),
'w') as fp:
print >> fp, self.simulator.score
print >> fp, self.simulator.total
else:
print "---------------"
print "Crashed! - %s" % id
print "---------------"
self.scores['crashes'] += 1
with open(user_folder / "crashes" / ("%s.log" % id),
'w') as fp:
fp.write(log)
with open(user_folder / "crashes" / ("%s.err" % id),
'w') as fp:
fp.write(error)
with open(user_folder / "crashes" / ("%s.score" % id),
'w') as fp:
print >> fp, self.simulator.score
print >> fp, self.simulator.total
events = SimulatorLog.parse(self.selenium.get_log())
for event in events:
if event.type == "ladder":
if event.details['username'] == self.username:
with open(user_folder / "ladder_ratings.txt",
"a") as fp:
fp.write(event.details['ladder'] + "\n")
self.reset()
self.update_monitor(done=True)
self.selenium.close()
self.logger.info("Done!")
def create_initial_gamestate(self):
self.logger.info("Creating initial gamestate...")
my_pokes = self.my_team.copy()
for i, poke in enumerate(my_pokes.poke_list):
poke_name = poke.name
if poke_name in NAME_CORRECTIONS:
poke_name = NAME_CORRECTIONS[poke_name]
poke.health = poke.final_stats['hp']
poke.alive = True
opp_poke_list = []
log = SimulatorLog.parse(self.selenium.get_log())
for event in log.events:
if event.type == "team" and event.details['username'] != self.username:
opp_poke_names = event.details['team']
for name in opp_poke_names:
if not name:
continue
poke_name = correct_name(name)
"Corrected to:", poke_name
if poke_name in self.smogon_data:
moveset = [m for m in self.smogon_data[poke_name].movesets if 'Overused' == m['tag'] or 'Underused' == m['tag'] or 'Rarelyused' == m['tag'] or 'Neverused' == m['tag'] or 'Unreleased' == m['tag'] or 'Ubers' == m['tag'] or 'PU' in m['tag']]
if len(moveset) > 1:
moveset = SmogonMoveset.from_dict(moveset[1])
elif len(moveset) == 1:
moveset = SmogonMoveset.from_dict(moveset[0])
else:
moveset = [m for m in self.smogon_bw_data[poke_name].movesets if 'Overused' == m['tag'] or 'Underused' == m['tag'] or 'Rarelyused' == m['tag'] or 'Neverused' == m['tag'] or 'Unreleased' == m['tag'] or 'Ubers' == m['tag'] or 'PU' in m['tag']]
moveset = SmogonMoveset.from_dict(moveset[0])
elif poke_name not in self.smogon_data and poke_name in self.smogon_bw_data:
moveset = [m for m in self.smogon_bw_data[poke_name].movesets if 'Overused' == m['tag'] or 'Underused' == m['tag'] or 'Rarelyused' == m['tag'] or 'Neverused' == m['tag'] or 'Unreleased' == m['tag'] or 'Ubers' == m['tag'] or 'PU' in m['tag']]
moveset = SmogonMoveset.from_dict(moveset[0])
else:
moveset = SmogonMoveset(None, None, None, {'hp': 88, 'patk': 84, 'pdef': 84, 'spatk': 84, 'spdef': 84, 'spe': 84}, {'hp': 1.0, 'patk': 1.0, 'pdef': 1.0, 'spatk': 1.0, 'spdef': 1.0, 'spe': 1.0}, None, 'ou')
moveset.moves = None
if poke_name in self.smogon_data:
typing = self.smogon_data[poke_name].typing
stats = self.smogon_data[poke_name].stats
elif poke_name not in self.smogon_data and poke_name in self.smogon_bw_data:
typing = self.smogon_bw_data[poke_name].typing
stats = self.smogon_bw_data[poke_name].stats
else:
typing = ['Normal']
stats = {'hp': 80, 'patk': 80, 'pdef': 80, 'spatk': 80, 'spdef': 80, 'spe': 80}
predictor = create_predictor(self.predictor_name, name, self.pokedata)
poke = Pokemon(name, typing, stats, moveset, predictor, calculate=True)
moves = [x[0] for x in poke.predict_moves([])]
poke.moveset.moves = moves[:4]
poke.health = poke.final_stats['hp']
poke.alive = True
opp_poke_list.append(poke)
my_primary = None
for event in log.events:
if event.type == "switch" and event.player == 0:
for poke in my_pokes.poke_list:
if poke.name == event.poke:
my_primary = my_pokes.poke_list.index(poke)
elif event.type == "switch" and event.player == 1:
for poke in opp_poke_list:
if poke.name == event.poke:
opp_primary = opp_poke_list.index(poke)
assert my_primary != None
self.opp_team = Team(opp_poke_list)
opp_pokes = self.opp_team.copy()
my_pokes.primary_poke = my_primary
opp_pokes.primary_poke = opp_primary
gamestate = GameState([my_pokes, opp_pokes])
return gamestate
def run(self, num_games=1, challenge=None):
self.state.update_state("status", "initializing_battle")
if challenge:
self.logger.info("Set to challenge: %s", challenge)
else:
self.logger.info("Set to play %u games", num_games)
self.init()
self.scores = {
'wins': 0,
'losses': 0,
'crashes': 0
}
def signal_handler(signal, frame):
self.update_monitor(done=True)
sys.exit(0)
#signal.signal(signal.SIGINT, signal_handler)
for i in range(num_games):
self.simulator.log.reset()
result, error = None, None
try:
self.play_game(challenge=challenge)
except GameOverException:
log = SimulatorLog.parse(self.selenium.get_log())
disconnected = log.disconnected()
if disconnected:
over,_ = log.is_over()
while not over:
time.sleep(5)
over,_ = log.is_over()
_, over_event = log.is_over()
result = over_event.details['username'] == self.username
except UserNotOnlineException:
self.logger.error("User not online: %s", challenge)
self.logger.info("Exiting...")
return
except:
error = traceback.format_exc()
print "Error", error
log = SimulatorLog.parse(self.selenium.get_log())
_, over_event = log.is_over()
if over_event is not None:
result = over_event.details['username'] == self.username
log = self.selenium.get_log()
id = self.selenium.get_battle_id()
battle_url = "http://replay.pokemonshowdown.com/battle-%s" % id
self.logger.info("Finished game! Replay can be found at: %s", battle_url)
user_folder = Path(".") / self.username
if not user_folder.exists():
user_folder.mkdir()
if not (user_folder / "wins").exists():
(user_folder / "wins").mkdir()
if not (user_folder / "losses").exists():
(user_folder / "losses").mkdir()
if not (user_folder / "crashes").exists():
(user_folder / "crashes").mkdir()
if result == True:
print "---------------"
print "Won the battle! - %s" % battle_url
print "---------------"
self.scores['wins'] += 1
with open(user_folder / "wins" / ("%s.log" % id), 'w') as fp:
fp.write(log)
with open(user_folder / "wins" / ("%s.score" % id), 'w') as fp:
print >>fp, self.simulator.score
print >>fp, self.simulator.total
elif result == False:
print "---------------"
print "Lost the battle! - %s" % battle_url
print "---------------"
self.scores['losses'] += 1
with open(user_folder / "losses" / ("%s.log" % id), 'w') as fp:
fp.write(log)
with open(user_folder / "losses" / ("%s.score" % id), 'w') as fp:
print >>fp, self.simulator.score
print >>fp, self.simulator.total
else:
print "---------------"
print "Crashed! - %s" % id
print "---------------"
self.scores['crashes'] += 1
with open(user_folder / "crashes" / ("%s.log" % id), 'w') as fp:
fp.write(log)
with open(user_folder / "crashes" / ("%s.err" % id), 'w') as fp:
fp.write(error)
with open(user_folder / "crashes" / ("%s.score" % id), 'w') as fp:
print >>fp, self.simulator.score
print >>fp, self.simulator.total
events = SimulatorLog.parse(self.selenium.get_log())
for event in events:
if event.type == "ladder":
if event.details['username'] == self.username:
with open(user_folder / "ladder_ratings.txt", "a") as fp:
fp.write(event.details['ladder'] + "\n")
self.reset()
self.update_monitor(done=True)
self.selenium.close()
self.logger.info("Done!")