Update move_to_map_pokemon.py (#4725)

Hardcoded get_pokemon_from_social
PokemonGoF · Aug 25, 2016 · ba0893f · ba0893f
1 parent 9605728
commit ba0893f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pokemongo_bot/cell_workers/move_to_map_pokemon.py b/pokemongo_bot/cell_workers/move_to_map_pokemon.py
@@ -291,9 +291,11 @@ def work(self):
 
         self.update_map_location()
         self.dump_caught_pokemon()
-        pokemon_list = self.get_pokemon_from_social()
-        #Temp works as it, need add more configuration
-        #pokemon_list = self.get_pokemon_from_map()
+        if self.bot.config.enable_social:
+            pokemon_list = self.get_pokemon_from_social()
+        else:
+            pokemon_list = self.get_pokemon_from_map()
+
         pokemon_list.sort(key=lambda x: x['dist'])
         if self.config['mode'] == 'priority':
             pokemon_list.sort(key=lambda x: x['priority'], reverse=True)