fishyboteso/fishy/engine/semifisher/fishing_mode.py

from enum import Enum
from time import time, sleep

subscribers = []
checkpoint = 0

class State(Enum):
    IDLE     =  0
    LOOKAWAY =  1
    LOOKING  =  2
    DEPLETED =  3
    NOBAIT   =  5
    FISHING  =  6
    REELIN   =  7
    LOOT     =  8
    INVFULL  =  9
    FIGHT    = 14
    DEAD     = 15


def _notify(event):
    for subscriber in subscribers:
        subscriber(event)


class FishingMode:
    CurrentMode = State.IDLE
    PrevMode = State.IDLE


def loop(state_num: int):
    """
    Executed in the start of the main loop in fishy.py
    Changes modes, calls mode events (callbacks) when mode is changed
    """
    global checkpoint
    FishingMode.CurrentMode = State(state_num)

    if FishingMode.CurrentMode != FishingMode.PrevMode:
        checkpoint = time()
        _notify(FishingMode.CurrentMode)
    elif FishingMode.CurrentMode == State.LOOKING:
        if time() - checkpoint > 5:
            _notify(FishingMode.CurrentMode)
            checkpoint = time()
        else:
            sleep(0.5)

    FishingMode.PrevMode = FishingMode.CurrentMode