from bot.States import States

MIDJOURNEY_SET_PROMPT = 41

def set_states():
    states_arr = {
        "set_prompt": MIDJOURNEY_SET_PROMPT
    }

    return states_arr


class MidjourneyStates(States):

    def __init__(self):
        pass

    @staticmethod
    def get_state_by_key(key: str):
        states = set_states()
        if key in states:
            return states[key]

        return None