BotLi/pending_challenge.py at main · Liatomic/BotLi · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from threading import Event

from aliases import Challenge_ID, Has_Reached_Rate_Limit, Is_Misconfigured, Success
from botli_dataclasses import Challenge_Response


class Pending_Challenge:
    def __init__(self) -> None:
        self._challenge_id_event = Event()
        self._challenge_id: Challenge_ID | None = None
        self._finished_event = Event()
        self._success: Success | None = None
        self._has_reached_rate_limit: Has_Reached_Rate_Limit | None = None
        self._is_misconfigured: Is_Misconfigured | None = None

    def get_challenge_id(self) -> Challenge_ID | None:
        ''' This is blocking '''
        self._challenge_id_event.wait()
        return self._challenge_id

    def get_final_state(self) -> tuple[Success, Has_Reached_Rate_Limit, Is_Misconfigured]:
        ''' This is blocking '''
        self._finished_event.wait()
        return bool(self._success), bool(self._has_reached_rate_limit), bool(self._is_misconfigured)

    def set_challenge_id(self, challenge_id: Challenge_ID) -> None:
        self._challenge_id = challenge_id
        self._challenge_id_event.set()

    def set_final_state(self, challenge_response: Challenge_Response) -> None:
        self._success = challenge_response.success
        self._has_reached_rate_limit = challenge_response.has_reached_rate_limit
        self._is_misconfigured = challenge_response.is_misconfigured
        self._finished_event.set()
        self._challenge_id_event.set()

    def return_early(self) -> None:
        self._finished_event.set()
        self._challenge_id_event.set()