auto_chaos.base_system

 1"""
 2🪐
 3Base system
 4🪐
 5"""
 6import asyncio
 7
 8
 9class BaseSystem:
10    """
11    Defines the base system class, all systems must inherit from this
12    """
13
14    def __init__(self) -> None:
15        self.results = []
16        self.errors = []
17
18    def do_action(self, data: str) -> tuple[list[str], list[str]]:
19        """
20        Do the action defined in data.
21
22        Args:
23            data (str): ACTION_1 arg_1 arg_2, ACTION_2 arg_1
24        """
25        self.results = []
26        self.errors = []
27        all_tasks = []
28
29        # Remove quotes
30        if '"' in data:
31            data = data.replace('"', "")
32        if "'" in data:
33            data = data.replace("'", "")
34        # Parse the string and get the action
35        if "," in data:
36            datas = data.split(",")
37        elif "\n" in data:
38            datas = data.split("\n")
39        else:
40            datas = [data]
41
42        # Iterate over all actions and execute them in separate threads
43        for data in datas:
44            if not data:
45                continue
46            if data[0] == " ":
47                data = data[1:]
48            try:
49                all_tasks.append(
50                    asyncio.to_thread(
51                        # Parse the action string to get the action name (corresponding to a
52                        # method name in lower case in the given system) and arguments
53                        getattr(self, data.split(" ")[0].lower()),
54                        data.split(" ")[1:],
55                    )
56                )
57            except AttributeError:
58                pass
59            except Exception as error:
60                self.errors.append(error)
61        asyncio.get_event_loop().run_until_complete(
62            asyncio.gather(*all_tasks, return_exceptions=True)
63        )
64        return self.results, self.errors
65
66    def default_action(self, args: list[str] = None):
67        """
68        Actions must be defined like that default action
69
70        Args:
71            args (list[str], optional): List of arguments to use. Defaults to None.
72        """
73        pass