aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/tests/test.py
blob: 8c3807b7817cdd61557f4c62014291b6dc0a505f (plain) (blame)
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import threading
import pickle
import os


class ConfigManager:
    def __init__(self, filename):
        self.lock = threading.RLock()
        self.filename = filename
        try:
            with open(filename, 'rb') as f:
                self.data = pickle.load(f)
        except FileNotFoundError:
            self.data = {}

        self.properties = {}
        self.apis = {}

    def load(self):
        with self.lock:
            try:
                with open(self.filename, 'rb') as f:
                    self.data = pickle.load(f)
                    self.properties.update(self.data)
            except FileNotFoundError:
                self.data = {}

    def save(self):
        with self.lock:
            try:
                with open(self.filename, 'wb') as f:
                    # 将self.properties字典中的内容合并至self.data字典中
                    for k, v in self.properties.items():
                        self.data[k] = v
                    # 将self.apis字典中的函数状态保存至self.data字典中
                    for k, v in self.apis.items():
                        if v is not None:
                            self.data[k] = v
                    pickle.dump(self.data, f)
            except Exception as e:
                print('Error saving config data:', repr(e))
                raise

    def register_property(self, name, default_value):
        self.properties[name] = default_value

        def getter(self, default_value=default_value):
            with self.lock:
                return self.data.get(name, default_value)

        def setter(self, value):
            with self.lock:
                self.data[name] = value

        setattr(ConfigManager, name, property(getter, setter))

    def register_api(self, name, func, default_value=None):
        self.apis[name] = default_value

        def wrapper(*args, **kwargs):
            with self.lock:
                return func(*args, **kwargs)

        setattr(ConfigManager, name, wrapper)


c = ConfigManager("test.dat")

# c.register_property("bot_off_list", [999, 777])
# def send_message(self, message):
#     print("Sending message:", message)

# c.register_api("send_message", send_message, default_value="Hello, world!")

# c.save()
c.load()
print(c.properties,c.properties['bot_off_list'],c.apis)