Skip to content

Commit a23b356

Browse files
elicwhiteMFizz
authored andcommitted
Adding a WorkerResult and the MoveToFortWorker only takes a single step towards a fort (PokemonGoF#1146)
1 parent 426c8ca commit a23b356

File tree

2 files changed

+21
-15
lines changed

2 files changed

+21
-15
lines changed

pokemongo_bot/cell_workers/move_to_fort_worker.py

Lines changed: 18 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from pokemongo_bot.human_behaviour import sleep
33
from pokemongo_bot import logger
44
from pokemongo_bot.step_walker import StepWalker
5+
from pokemongo_bot.worker_result import WorkerResult
56

67
class MoveToFortWorker(object):
78
def __init__(self, fort, bot):
@@ -11,6 +12,7 @@ def __init__(self, fort, bot):
1112
self.config = bot.config
1213
self.navigator = bot.navigator
1314
self.position = bot.position
15+
self._step_walker = None
1416

1517
def work(self):
1618
lat = self.fort['latitude']
@@ -29,20 +31,21 @@ def work(self):
2931
position = (lat, lng, 0.0)
3032

3133
if self.config.walk > 0:
32-
step_walker = StepWalker(
33-
self.bot,
34-
self.config.walk,
35-
self.api._position_lat,
36-
self.api._position_lng,
37-
position[0],
38-
position[1]
39-
)
40-
41-
while distance(i2f(self.api._position_lat), i2f(self.api._position_lng), lat, lng) > 10:
42-
if step_walker.step():
43-
break
44-
else:
45-
self.bot.process_cells(work_on_forts=False)
34+
if self._step_walker == None:
35+
self._step_walker = StepWalker(
36+
self.bot,
37+
self.config.walk,
38+
self.api._position_lat,
39+
self.api._position_lng,
40+
position[0],
41+
position[1]
42+
)
43+
44+
45+
if self._step_walker.step():
46+
return WorkerResult.SUCCESS
47+
else:
48+
return WorkerResult.RUNNING
4649

4750
else:
4851
self.api.set_position(*position)
@@ -51,6 +54,6 @@ def work(self):
5154
response_dict = self.api.call()
5255
logger.log('Arrived at Pokestop')
5356
sleep(2)
54-
return response_dict
57+
return WorkerResult.SUCCESS
5558

5659
return None

pokemongo_bot/worker_result.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
class WorkerResult(object):
2+
RUNNING = 'RUNNING'
3+
SUCCESS = 'SUCCESS'

0 commit comments

Comments
 (0)