def test_step(): config = create_core_test_config({ "movement": { "walk_speed": 5, "path_finder": "direct", "distance_unit": "m" } }) api_wrapper = create_mock_api_wrapper(config) path_finder = DirectPathFinder(config) logger = Mock() logger.log = Mock(return_value=None) stepper = Stepper(config, api_wrapper, path_finder, logger) stepper.start(51.50451, -0.07607, 10) destination = Destination(51.506000, -0.075049, 11, name="Test Destination", exact_location=False) steps = [(51.504778, -0.075838, 10), (51.505092, -0.075650, 11), (51.505436, -0.075446, 11)] destination.set_steps(steps) pgo = api_wrapper.get_api() # This route is being walked: http://www.darrinward.com/lat-long/?id=2163411 # pre-calculated distance is 17.8 meters pointer = 0 for _ in stepper.step(destination): target_lat, target_lng, target_alt = steps[pointer] assert stepper.current_lat == target_lat assert stepper.current_lng == target_lng assert stepper.current_alt == target_alt bot_lat, bot_lng, bot_alt = pgo.get_position() assert bot_lat == target_lat assert bot_lng == target_lng assert bot_alt == target_alt pointer += 1 assert pointer == 3 bot_lat, bot_lng, bot_alt = pgo.get_position() assert bot_lat == 51.505436 assert bot_lng == -0.075446 assert bot_alt == 11 assert stepper.current_lat == 51.505436 assert stepper.current_lng == -0.075446 assert stepper.current_alt == 11
def test_step(): config = create_core_test_config({ "movement": { "walk_speed": 5, "path_finder": "direct", "distance_unit": "m" } }) api_wrapper = create_mock_api_wrapper(config) path_finder = DirectPathFinder(config) logger = Mock() logger.log = Mock(return_value=None) stepper = Stepper(config, api_wrapper, path_finder, logger) stepper.start(51.50451, -0.07607, 10) destination = Destination(51.506000, -0.075049, 11, name="Test Destination", exact_location=False) steps = [ (51.504778, -0.075838, 10), (51.505092, -0.075650, 11), (51.505436, -0.075446, 11) ] destination.set_steps(steps) pgo = api_wrapper.get_api() # This route is being walked: http://www.darrinward.com/lat-long/?id=2163411 # pre-calculated distance is 17.8 meters pointer = 0 for _ in stepper.step(destination): target_lat, target_lng, target_alt = steps[pointer] assert stepper.current_lat == target_lat assert stepper.current_lng == target_lng assert stepper.current_alt == target_alt bot_lat, bot_lng, bot_alt = pgo.get_position() assert bot_lat == target_lat assert bot_lng == target_lng assert bot_alt == target_alt pointer += 1 assert pointer == 3 bot_lat, bot_lng, bot_alt = pgo.get_position() assert bot_lat == 51.505436 assert bot_lng == -0.075446 assert bot_alt == 11 assert stepper.current_lat == 51.505436 assert stepper.current_lng == -0.075446 assert stepper.current_alt == 11
def test_step_already_near_fort(): calls = [] config = create_core_test_config({ "movement": { "walk_speed": 5, "path_finder": "direct", "distance_unit": "m" } }) api_wrapper = create_mock_api_wrapper(config) path_finder = DirectPathFinder(config) destination = Destination(51.50436, -0.07616, 11, name="Test Destination", exact_location=False) steps = [(51.50442, -0.07612, 10), (51.50448, -0.07609, 11), (51.50436, -0.07616, 11)] destination.set_steps(steps) logger = Mock() logger.log = Mock(return_value=None) stepper = Stepper(config, api_wrapper, path_finder, logger) stepper.start(51.50451, -0.07607, 10) pgo = api_wrapper.get_api() # This route is being walked: http://www.darrinward.com/lat-long/?id=2163408 calls.append(call("walking_started", coords=(51.50436, -0.07616, 11))) pointer = 0 for _ in stepper.step(destination): pointer += 1 assert pointer == 0 bot_lat, bot_lng, bot_alt = pgo.get_position() assert bot_lat == 51.50451 assert bot_lng == -0.07607 assert bot_alt == 10 assert stepper.current_lat == 51.50451 assert stepper.current_lng == -0.07607 assert stepper.current_alt == 10
def test_step_already_near_fort(): calls = [] config = create_core_test_config({ "movement": { "walk_speed": 5, "path_finder": "direct", "distance_unit": "m" } }) api_wrapper = create_mock_api_wrapper(config) path_finder = DirectPathFinder(config) destination = Destination(51.50436, -0.07616, 11, name="Test Destination", exact_location=False) steps = [ (51.50442, -0.07612, 10), (51.50448, -0.07609, 11), (51.50436, -0.07616, 11) ] destination.set_steps(steps) logger = Mock() logger.log = Mock(return_value=None) stepper = Stepper(config, api_wrapper, path_finder, logger) stepper.start(51.50451, -0.07607, 10) pgo = api_wrapper.get_api() # This route is being walked: http://www.darrinward.com/lat-long/?id=2163408 calls.append(call("walking_started", coords=(51.50436, -0.07616, 11))) pointer = 0 for _ in stepper.step(destination): pointer += 1 assert pointer == 0 bot_lat, bot_lng, bot_alt = pgo.get_position() assert bot_lat == 51.50451 assert bot_lng == -0.07607 assert bot_alt == 10 assert stepper.current_lat == 51.50451 assert stepper.current_lng == -0.07607 assert stepper.current_alt == 10
class PokemonGoBot(object): process_ignored_pokemon = False def __init__(self, config): self.config = config self.pokemon_list = json.load(open('data/pokemon.json')) self.item_list = {} for item_id, item_name in json.load(open('data/items.json')).items(): self.item_list[int(item_id)] = item_name self.log = None self.api_wrapper = None self.inventory = [] self.candies = {} self.ignores = [] self.position = (0, 0, 0) self.plugin_manager = None self.last_session_check = time.gmtime() self.stepper = None self.navigator = None self.mapper = None def _init_plugins(self): # create a plugin manager self.plugin_manager = PluginManager('./plugins') # load all plugin modules for plugin in self.plugin_manager.get_available_plugins(): if plugin not in self.config.exclude_plugins: self.plugin_manager.load_plugin(plugin) else: logger.log("Not loading plugin \"{}\"".format(plugin), color="red", prefix="Plugins") loaded_plugins = sorted(self.plugin_manager.get_loaded_plugins().keys()) sleep(2) logger.log("Plugins loaded: {}".format(loaded_plugins), color="green", prefix="Plugins") if self.config.print_events: logger.log("Events available: {}".format(manager.get_registered_events()), color="green", prefix="Events") manager.print_all_event_pipelines() def start(self): self._setup_logging() self._init_plugins() self._setup_api() random.seed() self.stepper = Stepper(self) self.mapper = Mapper(self) if self.config.navigator == 'fort': self.navigator = FortNavigator(self) # pylint: disable=redefined-variable-type elif self.config.navigator == 'waypoint': self.navigator = WaypointNavigator(self) # pylint: disable=redefined-variable-type elif self.config.navigator == 'camper': self.navigator = CamperNavigator(self) # pylint: disable=redefined-variable-type self.fire('bot_initialized') if self.config.initial_transfer: self.fire("pokemon_bag_full") if self.config.recycle_items: self.fire("item_bag_full") logger.log('[#]') self.update_player_and_inventory() def fire(self, event, *args, **kwargs): # type: (str, *Any, **Any) -> None manager.fire_with_context(event, self, *args, **kwargs) def run(self): map_cells = self.mapper.get_cells_at_current_position() # Work on all the initial cells self.work_on_cells(map_cells) for destination in self.navigator.navigate(map_cells): position_lat = self.stepper.current_lat position_lng = self.stepper.current_lng destination.set_steps( self.stepper.get_route_between( position_lat, position_lng, destination.target_lat, destination.target_lng, destination.target_alt ) ) for _ in self.stepper.step(destination): self.work_on_cells( self.mapper.get_cells_at_current_position() ) def work_on_cells(self, map_cells): # type: (Cell, bool) -> None encounters = [] pokestops = [] for cell in map_cells: encounters += cell.catchable_pokemon + cell.wild_pokemon pokestops += cell.pokestops if len(encounters): self.fire("pokemon_found", encounters=encounters) if len(pokestops): self.fire("pokestops_found", pokestops=pokestops) def _setup_logging(self): self.log = logging.getLogger(__name__) # log settings # log format logging.basicConfig( level=logging.DEBUG, format='%(asctime)s [%(module)10s] [%(levelname)5s] %(message)s') if self.config.debug: logging.getLogger("requests").setLevel(logging.DEBUG) logging.getLogger("pgoapi").setLevel(logging.DEBUG) logging.getLogger("rpc_api").setLevel(logging.DEBUG) else: logging.getLogger("requests").setLevel(logging.ERROR) logging.getLogger("pgoapi").setLevel(logging.ERROR) logging.getLogger("rpc_api").setLevel(logging.ERROR) def _setup_api(self): # instantiate api self.api_wrapper = PoGoApi(provider=self.config.auth_service, username=self.config.username, password=self.config.password, shared_lib=self.config.load_library) # provide player position on the earth self._set_starting_position() while not self.api_wrapper.login(): logger.log('Login Error, server busy', 'red') logger.log('Waiting 15 seconds before trying again...') time.sleep(15) logger.log('[+] Login to Pokemon Go successful.', 'green') # chain subrequests (methods) into one RPC call # get player profile call # ---------------------- response_dict = self.update_player_and_inventory() if response_dict is not None: player = response_dict['player'] inventory = response_dict['inventory'] self.candies = response_dict['candy'] pokemon = response_dict['pokemon'] eggs = response_dict['eggs'] creation_date = player.get_creation_date() balls_stock = self.pokeball_inventory() pokecoins = player.pokecoin stardust = player.stardust logger.log('[#]') logger.log('[#] Username: {}'.format(player.username)) logger.log('[#] Acccount creation: {}'.format(creation_date)) logger.log('[#] Bag storage: {}/{}'.format(inventory["count"], player.max_item_storage)) logger.log('[#] Pokemon storage: {}/{}'.format(len(pokemon) + len(eggs), player.max_pokemon_storage)) logger.log('[#] Stardust: {:,}'.format(stardust)) logger.log('[#] Pokecoins: {}'.format(pokecoins)) logger.log('[#] Poke Balls: {}'.format(balls_stock[1])) logger.log('[#] Great Balls: {}'.format(balls_stock[2])) logger.log('[#] Ultra Balls: {}'.format(balls_stock[3])) logger.log('[#] -- Level: {}'.format(player.level)) logger.log('[#] -- Experience: {:,}'.format(player.experience)) logger.log('[#] -- Experience until next level: {:,}'.format(player.next_level_xp - player.experience)) logger.log('[#] -- Pokemon captured: {:,}'.format(player.pokemons_captured)) logger.log('[#] -- Pokestops visited: {:,}'.format(player.poke_stop_visits)) # Testing # self.drop_item(Item.ITEM_POTION.value,1) # exit(0) def update_player_and_inventory(self): # type: () -> Dict[str, object] response_dict = self.api_wrapper.get_player().get_inventory().call() self.candies = response_dict['candy'] return response_dict def add_candies(self, name=None, pokemon_candies=None): for pokemon in self.pokemon_list: if pokemon['Name'] is not name: continue else: previous_evolutions = pokemon.get("Previous evolution(s)", []) if previous_evolutions: candy_name = previous_evolutions[0]['Number'] else: candy_name = pokemon.get("Number") if self.candies.get(candy_name, None) is not None: self.candies[candy_name] += pokemon_candies else: self.candies[candy_name] = pokemon_candies logger.log("[#] Added {} candies for {}".format(pokemon_candies, self.pokemon_list[int(candy_name) - 1]['Name']), 'green') break def pokeball_inventory(self): balls_stock = {Item.ITEM_POKE_BALL.value: 0, Item.ITEM_GREAT_BALL.value: 0, Item.ITEM_ULTRA_BALL.value: 0, Item.ITEM_MASTER_BALL.value: 0} result = self.api_wrapper.get_inventory().call() if result is None: return balls_stock inventory_list = result["inventory"] for item_id in inventory_list: if item_id in balls_stock: balls_stock[item_id] = inventory_list[item_id] return balls_stock def _set_starting_position(self): if self.config.test: return if self.config.location_cache: try: # # save location flag used to pull the last known location from # the location.json with open('data/last-location-%s.json' % self.config.username) as last_location_file: location_json = json.load(last_location_file) self.position = (location_json['lat'], location_json['lng'], 0.0) self.api_wrapper.set_position(*self.position) logger.log('') logger.log('[x] Last location flag used. Overriding passed in location') logger.log('[x] Last in-game location was set as: {}'.format(self.position)) logger.log('') return except IOError: if not self.config.location: sys.exit("No cached Location. Please specify initial location.") else: self._read_config_location() else: self._read_config_location() logger.log('[x] Position in-game set as: {}'.format(self.position)) logger.log('') def _read_config_location(self): self.position = self._get_pos_by_name(self.config.location) self.api_wrapper.set_position(*self.position) logger.log('') logger.log(u'[x] Address found: {}'.format(self.config.location)) def _get_pos_by_name(self, location_name): # type: (str) -> Tuple[float, float, float] if location_name.count(',') == 1: try: logger.log("[x] Fetching altitude from google") parts = location_name.split(',') pos_lat = float(parts[0]) pos_lng = float(parts[1]) # we need to ask google for the altitude gmaps = googlemaps.Client(key=self.config.gmapkey) response = gmaps.elevation((pos_lat, pos_lng)) if len(response) and "elevation" in response[0]: return pos_lat, pos_lng, response[0]["elevation"] else: raise ValueError except ApiError: logger.log("[x] Could not fetch altitude from google. Trying geolocator.") except ValueError: logger.log("[x] Location was not Lat/Lng.") # Fallback to geolocation if no Lat/Lng can be found geolocator = GoogleV3(api_key=self.config.gmapkey) loc = geolocator.geocode(location_name, timeout=10) # self.log.info('Your given location: %s', loc.address.encode('utf-8')) # self.log.info('lat/long/alt: %s %s %s', loc.latitude, loc.longitude, loc.altitude) return loc.latitude, loc.longitude, loc.altitude def heartbeat(self): self.api_wrapper.get_player() self.api_wrapper.get_hatched_eggs() self.api_wrapper.get_inventory() self.api_wrapper.check_awarded_badges() self.api_wrapper.call() def get_pokemon_count(self): # type: () -> int response_dict = self.update_player_and_inventory() if response_dict is None: return 0 return len(response_dict["pokemon"]) def get_item_count(self): # type: () -> int response_dict = self.update_player_and_inventory() if response_dict is None: return 0 return response_dict["inventory"]["count"] def get_username(self): # type: () -> str response_dict = self.update_player_and_inventory() if response_dict is None: return "Unknown" return response_dict["player"].username def get_position(self): return self.position
class PokemonGoBot(object): process_ignored_pokemon = False def __init__(self, config): self.config = config self.pokemon_list = json.load(open('data/pokemon.json')) self.item_list = {} for item_id, item_name in json.load(open('data/items.json')).items(): self.item_list[int(item_id)] = item_name self.log = None self.api_wrapper = None self.ignores = [] self.position = (0, 0, 0) self.plugin_manager = None self.last_session_check = time.gmtime() self.stepper = None self.navigator = None self.mapper = None def _init_plugins(self): # create a plugin manager self.plugin_manager = PluginManager('./plugins') # load all plugin modules for plugin in self.plugin_manager.get_available_plugins(): if plugin not in self.config.exclude_plugins: self.plugin_manager.load_plugin(plugin) else: logger.log("Not loading plugin \"{}\"".format(plugin), color="red", prefix="Plugins") loaded_plugins = sorted(self.plugin_manager.get_loaded_plugins().keys()) sleep(2) logger.log("Plugins loaded: {}".format(loaded_plugins), color="green", prefix="Plugins") logger.log("Events available: {}".format(manager.get_registered_events()), color="green", prefix="Events") def start(self): self._setup_logging() self._init_plugins() self._setup_api() random.seed() self.stepper = Stepper(self) self.mapper = Mapper(self) if self.config.navigator == 'fort': self.navigator = FortNavigator(self) # pylint: disable=redefined-variable-type elif self.config.navigator == 'waypoint': self.navigator = WaypointNavigator(self) # pylint: disable=redefined-variable-type elif self.config.navigator == 'camper': self.navigator = CamperNavigator(self) # pylint: disable=redefined-variable-type def fire(self, event, *args, **kwargs): # type: (str, *Any, **Any) -> None manager.fire_with_context(event, self, *args, **kwargs) def run(self): map_cells = self.mapper.get_cells_at_current_position() # Work on all the initial cells self.work_on_cells(map_cells) position_lat = self.stepper.current_lat position_lng = self.stepper.current_lng for destination in self.navigator.navigate(map_cells): destination.set_steps( self.stepper.get_route_between(position_lat, position_lng, destination.target_lat, destination.target_lng, destination.target_alt) ) for _ in self.stepper.step(destination): self.work_on_cells( self.mapper.get_cells_at_current_position() ) position_lat = destination.target_lat position_lng = destination.target_lng def work_on_cells(self, map_cells): # type: (Cell, bool) -> None encounters = [] pokestops = [] for cell in map_cells: encounters += cell.catchable_pokemon + cell.wild_pokemon pokestops += cell.pokestops if len(encounters): self.fire("pokemon_found", encounters=encounters) if len(pokestops): self.fire("pokestops_found", pokestops=pokestops) def _setup_logging(self): self.log = logging.getLogger(__name__) # log settings # log format logging.basicConfig( level=logging.DEBUG, format='%(asctime)s [%(module)10s] [%(levelname)5s] %(message)s') if self.config.debug: logging.getLogger("requests").setLevel(logging.DEBUG) logging.getLogger("pgoapi").setLevel(logging.DEBUG) logging.getLogger("rpc_api").setLevel(logging.DEBUG) else: logging.getLogger("requests").setLevel(logging.ERROR) logging.getLogger("pgoapi").setLevel(logging.ERROR) logging.getLogger("rpc_api").setLevel(logging.ERROR) def _setup_api(self): # instantiate api self.api_wrapper = PoGoApi(provider=self.config.auth_service, username=self.config.username, password=self.config.password) # provide player position on the earth self._set_starting_position() while not self.api_wrapper.login(): logger.log('Login Error, server busy', 'red') logger.log('Waiting 15 seconds before trying again...') time.sleep(15) logger.log('[+] Login to Pokemon Go successful.', 'green') # chain subrequests (methods) into one RPC call # get player profile call # ---------------------- response_dict = self.update_player_and_inventory() if response_dict is not None: player = response_dict['player'] inventory = response_dict['inventory'] pokemon = response_dict['pokemon'] creation_date = player.get_creation_date() balls_stock = self.pokeball_inventory() pokecoins = player.pokecoin stardust = player.stardust logger.log('[#]') logger.log('[#] Username: {}'.format(player.username)) logger.log('[#] Acccount Creation: {}'.format(creation_date)) logger.log('[#] Bag Storage: {}/{}'.format(inventory["count"], player.max_item_storage)) logger.log('[#] Pokemon Storage: {}/{}'.format(len(pokemon), player.max_pokemon_storage)) logger.log('[#] Stardust: {}'.format(stardust)) logger.log('[#] Pokecoins: {}'.format(pokecoins)) logger.log('[#] PokeBalls: {}'.format(balls_stock[1])) logger.log('[#] GreatBalls: {}'.format(balls_stock[2])) logger.log('[#] UltraBalls: {}'.format(balls_stock[3])) logger.log('[#] -- Level: {}'.format(player.level)) logger.log('[#] -- Experience: {}'.format(player.experience)) logger.log('[#] -- Experience until next level: {}'.format(player.next_level_xp - player.experience)) logger.log('[#] -- Pokemon Captured: {}'.format(player.pokemons_captured)) logger.log('[#] -- Pokestops Visited: {}'.format(player.poke_stop_visits)) # Testing # self.drop_item(Item.ITEM_POTION.value,1) # exit(0) if self.config.initial_transfer: self.fire("pokemon_bag_full") if self.config.recycle_items: self.fire("item_bag_full") logger.log('[#]') self.update_player_and_inventory() def update_player_and_inventory(self): # type: () -> Dict[str, object] self.api_wrapper.get_player().get_inventory() return self.api_wrapper.call() def pokeball_inventory(self): balls_stock = {Item.ITEM_POKE_BALL.value: 0, Item.ITEM_GREAT_BALL.value: 0, Item.ITEM_ULTRA_BALL.value: 0, Item.ITEM_MASTER_BALL.value: 0} result = self.api_wrapper.get_inventory().call() if result is None: return balls_stock inventory_list = result["inventory"] for item_id in inventory_list: if item_id in balls_stock: balls_stock[item_id] = inventory_list[item_id] return balls_stock def _set_starting_position(self): if self.config.test: return if self.config.location_cache: try: # # save location flag used to pull the last known location from # the location.json with open('data/last-location-%s.json' % self.config.username) as last_location_file: location_json = json.load(last_location_file) self.position = (location_json['lat'], location_json['lng'], 0.0) self.api_wrapper.set_position(*self.position) logger.log('') logger.log('[x] Last location flag used. Overriding passed in location') logger.log('[x] Last in-game location was set as: {}'.format(self.position)) logger.log('') return except IOError: if not self.config.location: sys.exit("No cached Location. Please specify initial location.") else: self._read_config_location() else: self._read_config_location() logger.log('[x] Position in-game set as: {}'.format(self.position)) logger.log('') def _read_config_location(self): self.position = self._get_pos_by_name(self.config.location) self.api_wrapper.set_position(*self.position) logger.log('') logger.log(u'[x] Address found: {}'.format(self.config.location)) def _get_pos_by_name(self, location_name): # type: (str) -> Tuple[float, float, float] if location_name.count(',') == 1: try: logger.log("[x] Fetching altitude from google") parts = location_name.split(',') pos_lat = float(parts[0]) pos_lng = float(parts[1]) # we need to ask google for the altitude gmaps = googlemaps.Client(key=self.config.gmapkey) response = gmaps.elevation((pos_lat, pos_lng)) if len(response) and "elevation" in response[0]: return pos_lat, pos_lng, response[0]["elevation"] else: raise ValueError except ApiError: logger.log("[x] Could not fetch altitude from google. Trying geolocator.") except ValueError: logger.log("[x] Location was not Lat/Lng.") # Fallback to geolocation if no Lat/Lng can be found geolocator = GoogleV3(api_key=self.config.gmapkey) loc = geolocator.geocode(location_name, timeout=10) # self.log.info('Your given location: %s', loc.address.encode('utf-8')) # self.log.info('lat/long/alt: %s %s %s', loc.latitude, loc.longitude, loc.altitude) return loc.latitude, loc.longitude, loc.altitude def heartbeat(self): self.api_wrapper.get_player() self.api_wrapper.get_hatched_eggs() self.api_wrapper.get_inventory() self.api_wrapper.check_awarded_badges() self.api_wrapper.call() def get_pokemon_count(self): # type: () -> int response_dict = self.update_player_and_inventory() if response_dict is None: return 0 return len(response_dict["pokemon"]) def get_item_count(self): # type: () -> int response_dict = self.update_player_and_inventory() if response_dict is None: return 0 return response_dict["inventory"]["count"]