Lib.utils.StateManager

 1"""
 2状态管理器
 3注意，数据存储于内存中，重启丢失，需要持久化保存的数据请勿放在里面
 4"""
 5
 6from Lib.core import PluginManager
 7
 8states = {}
 9
10
11def get_state(state_id: str, plugin_data: dict = None):
12    """
13    获取状态数据
14    Args:
15        state_id: 状态ID
16        plugin_data: 插件数据
17
18    Returns:
19        状态数据，结构类似
20        {
21            "state_id": state_id,
22            "data": {
23                k1: v1,
24                k2: v2
25            },
26            "other_plugin_data": {
27                plugin1_path: {
28                    "data": {
29                        k1: v1,
30                        k2: v2
31                    },
32                    "meta": {
33                        "plugin_data": plugin_data
34                    }
35                },
36                plugin2_path: {
37                    "data": {
38                        k1: v1,
39                        k2: v2
40                    },
41                    "meta": {
42                        "plugin_data": plugin_data
43                    }
44                }
45            }
46        }
47    """
48    if plugin_data is None:
49        plugin_data = PluginManager.get_caller_plugin_data()
50
51    if state_id not in states:
52        states[state_id] = {}
53
54    if plugin_data["path"] not in states[state_id]:
55        states[state_id][plugin_data["path"]] = {
56            "data": {},
57            "meta": {
58                "plugin_data": plugin_data
59            }
60        }
61
62    return {
63        "state_id": state_id,
64        "data": states[state_id][plugin_data["path"]]["data"],
65        "other_plugin_data": {
66            k: v for k, v in states[state_id].items() if k != plugin_data["path"]
67        }
68    }