catch_lured_pokemon_改.py

# updated to https://github.com/PokemonGoF/PokemonGo-Bot/commit/0c3c4c004d5081b91ffe278850d1b1821e4f63b4

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from pokemongo_bot.base_task import BaseTask
from pokemongo_bot.worker_result import WorkerResult
from pokemongo_bot.constants import Constants
from pokemongo_bot.cell_workers.utils import fort_details, distance
from pokemongo_bot.cell_workers.pokemon_catch_worker import PokemonCatchWorker

ignored_poke_arr = []

class CatchLuredPokemon(BaseTask):
    SUPPORTED_TASK_API_VERSION = 1

    def work(self):
        lured_pokemon = self.get_lured_pokemon()
        if len(lured_pokemon) > 0:
            self.catch_pokemon(lured_pokemon[0])

            if len(lured_pokemon) > 1:
                return WorkerResult.RUNNING

        return WorkerResult.SUCCESS

    def get_lured_pokemon(self):
        forts_in_range = []
        pokemon_to_catch = []
        forts = self.bot.get_forts(order_by_distance=True)

        if len(forts) == 0:
            return []

        for fort in forts:
            distance_to_fort = distance(
                self.bot.position[0],
                self.bot.position[1],
                fort['latitude'],
                fort['longitude']
            )

            encounter_id = fort.get('lure_info', {}).get('encounter_id', None)
            if distance_to_fort < Constants.MAX_DISTANCE_FORT_IS_REACHABLE and encounter_id:
                forts_in_range.append(fort)


        for fort in forts_in_range:
            details = fort_details(self.bot, fort_id=fort['id'],
                                  latitude=fort['latitude'],
                                  longitude=fort['longitude'])
            fort_name = details.get('name', 'Unknown')
            encounter_id = fort['lure_info']['encounter_id']

            result = {
                'encounter_id': encounter_id,
                'fort_id': fort['id'],
                'fort_name': u"{}".format(fort_name),
                'latitude': fort['latitude'],
                'longitude': fort['longitude']
            }
            
            if encounter_id in ignored_poke_arr:
                print('fixed infinite loop, CatchLuredPokemon')
                continue
            else:
                ignored_poke_arr.append(encounter_id)
                if len(ignored_poke_arr) > 20:
                    ignored_poke_arr.pop(0)
            
            pokemon_to_catch.append(result)

            self.emit_event(
                'lured_pokemon_found',
                formatted='Lured pokemon at fort {fort_name} ({fort_id})',
                data=result
            )
        return pokemon_to_catch

    def catch_pokemon(self, pokemon):
        worker = PokemonCatchWorker(pokemon, self.bot)
        return_value = worker.work()

        return return_value