Keep extra pokemon in our bag so we can do smarter things later

thebigjc · Aug 9, 2016 · 183711f · 183711f
1 parent 0f2bddd
commit 183711f
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/configs/config.json.example b/configs/config.json.example
@@ -19,7 +19,10 @@
         }
       },
       {
-        "type": "TransferPokemon"
+        "type": "TransferPokemon",
+        "config": {
+          "min_empty_space": 15
+        }
       },
       {
         "type": "EvolvePokemon",

diff --git a/pokemongo_bot/__init__.py b/pokemongo_bot/__init__.py
@@ -362,6 +362,11 @@ def _register_events(self):
         self.event_manager.register_event('inventory_full')
 
         # release
+        self.event_manager.register_event(
+            'pokemon_transfer_skipped',
+            parameters=('space',)
+        )
+
         self.event_manager.register_event(
             'keep_best_release',
             parameters=(

diff --git a/pokemongo_bot/cell_workers/transfer_pokemon.py b/pokemongo_bot/cell_workers/transfer_pokemon.py
@@ -7,7 +7,24 @@
 class TransferPokemon(BaseTask):
     SUPPORTED_TASK_API_VERSION = 1
 
+    def initialize(self):
+        self.min_empty_space = self.config.get('min_empty_space', None)
+
+    def get_pokemon_slot_left(self):
+        return self.bot._player["max_pokemon_storage"] - self.bot.get_inventory_count('pokemon')
+
     def work(self):
+        if self.min_empty_space is not None:
+            free_poke_space = self.get_pokemon_slot_left()
+            if free_poke_space >= self.min_empty_space:
+                print free_poke_space
+                self.emit_event(
+                    'pokemon_transfer_skipped',
+                    formatted="Skipping Transfering of Pokemon.  {space} space left in poke bag.",
+                    data={ 'space': free_poke_space }
+                    )
+                return
+
         pokemon_groups = self._release_pokemon_get_groups()
         for pokemon_id in pokemon_groups:
             group = pokemon_groups[pokemon_id]