mirror of
https://github.com/fishyboteso/fishyboteso.git
synced 2024-08-30 18:32:13 +00:00
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
|
import logging
|
||
|
|
import pickle
|
||
|
|
from pprint import pprint
|
||
|
|
|
||
|
|
import typing
|
||
|
|
from threading import Thread
|
||
|
|
|
||
|
|
from fishy.engine.fullautofisher.mode.imode import IMode
|
||
|
|
from fishy.engine.semifisher import fishing_event, fishing_mode
|
||
|
|
from fishy.helper.helper import log_raise, wait_until, kill_thread
|
||
|
|
|
||
|
|
if typing.TYPE_CHECKING:
|
||
|
|
from fishy.engine.fullautofisher.engine import FullAuto
|
||
|
|
|
||
|
|
from fishy.helper import helper
|
||
|
|
from fishy.helper.config import config
|
||
|
|
|
||
|
|
|
||
|
|
def _get_rec_file():
|
||
|
|
file = config.get("full_auto_rec_file")
|
||
|
|
|
||
|
|
if not file:
|
||
|
|
logging.error("Please select a fishy file first from config")
|
||
|
|
return None
|
||
|
|
|
||
|
|
file = open(file, 'rb')
|
||
|
|
data = pickle.load(file)
|
||
|
|
file.close()
|
||
|
|
pprint(data)
|
||
|
|
if "full_auto_path" not in data:
|
||
|
|
logging.error("invalid file")
|
||
|
|
return None
|
||
|
|
return data["full_auto_path"]
|
||
|
|
|
||
|
|
|
||
|
|
class Player(IMode):
|
||
|
|
def __init__(self, engine: 'FullAuto'):
|
||
|
|
self.recording = False
|
||
|
|
self.engine = engine
|
||
|
|
self.hole_complete_flag = False
|
||
|
|
self.start_moving_flag = False
|
||
|
|
self.i = 0
|
||
|
|
self.forward = True
|
||
|
|
self.timeline = None
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
self._init()
|
||
|
|
while self.engine.start:
|
||
|
|
self._loop()
|
||
|
|
logging.info("player stopped")
|
||
|
|
|
||
|
|
def _init(self):
|
||
|
|
self.timeline = _get_rec_file()
|
||
|
|
if not self.timeline:
|
||
|
|
log_raise("data not found, can't start")
|
||
|
|
logging.info("starting player")
|
||
|
|
|
||
|
|
def _loop(self):
|
||
|
|
action = self.timeline[self.i]
|
||
|
|
|
||
|
|
if action[0] == "move_to":
|
||
|
|
self.engine.move_to(action[1])
|
||
|
|
elif action[0] == "check_fish":
|
||
|
|
self.engine.move_to(action[1])
|
||
|
|
self.engine.rotate_to(action[1][2])
|
||
|
|
fishing_event.subscribe()
|
||
|
|
fishing_mode.subscribers.append(self._hole_complete_callback)
|
||
|
|
# scan for fish hole
|
||
|
|
logging.info("scanning")
|
||
|
|
# if found start fishing and wait for hole to complete
|
||
|
|
if self.engine.look_for_hole():
|
||
|
|
logging.info("starting fishing")
|
||
|
|
self.hole_complete_flag = False
|
||
|
|
helper.wait_until(lambda: self.hole_complete_flag or not self.engine.start)
|
||
|
|
self.engine.rotate_back()
|
||
|
|
else:
|
||
|
|
logging.info("no hole found")
|
||
|
|
# continue when hole completes
|
||
|
|
fishing_event.unsubscribe()
|
||
|
|
fishing_mode.subscribers.remove(self._hole_complete_callback)
|
||
|
|
|
||
|
|
self.i += 1 if self.forward else -1
|
||
|
|
if self.i >= len(self.timeline):
|
||
|
|
self.forward = False
|
||
|
|
self.i = len(self.timeline) - 1
|
||
|
|
elif self.i < 0:
|
||
|
|
self.forward = True
|
||
|
|
self.i = 0
|
||
|
|
|
||
|
|
def _hole_complete_callback(self, e):
|
||
|
|
if e == fishing_event.State.IDLE:
|
||
|
|
self.hole_complete_flag = True
|