From 56001e8a94efff01736cd8b6e135578566ffc31b Mon Sep 17 00:00:00 2001 From: DeXtroTip Date: Tue, 20 Sep 2016 19:51:30 +0100 Subject: [PATCH] Fixed bug when setting buddy for the first time Added "enabled" check --- pokemongo_bot/cell_workers/buddy_pokemon.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pokemongo_bot/cell_workers/buddy_pokemon.py b/pokemongo_bot/cell_workers/buddy_pokemon.py index 5262ad440a..1db8a0b5de 100644 --- a/pokemongo_bot/cell_workers/buddy_pokemon.py +++ b/pokemongo_bot/cell_workers/buddy_pokemon.py @@ -89,6 +89,9 @@ def _validate_config(self): self.buddy_distance_needed = pokemon.buddy_distance_needed def work(self): + if not self.enabled: + return WorkerResult.SUCCESS + if self.buddy_list: if self.force_first_change or not self.buddy or self.candy_limit != 0 and self.candy_awarded >= self.candy_limit: self.force_first_change = False @@ -105,7 +108,7 @@ def work(self): if pokemon is None: return WorkerResult.ERROR - if pokemon.name != self._get_pokemon_by_id(self.buddy['id']).name: + if not self.buddy or pokemon.name != self._get_pokemon_by_id(self.buddy['id']).name: self._set_buddy(pokemon) if not self.buddy: