first commit

control/request.py

@@ -0,0 +1,42 @@
+import logging
+import toml
+
+class request:
+    def __init__(self,msg):
+        self.time = msg.time
+        self.self_id = msg.self_id
+        self.request_type = msg.request_type
+        self.sub_type = msg.sub_type
+        self.group_id = msg.group_id
+        self.user_id = msg.user_id
+        self.comment = msg.comment
+        self.flag = msg.flag
+
+    def main(self):
+        if self.request_type == "friend":
+            friend_auto = self.get_info()
+            if friend_auto:
+                return True
+            else:
+                return False
+
+
+    def get_info(self):
+        try:
+            with open("./config.toml", "r", encoding="utf-8") as f:
+                config = toml.load(f)
+                friend_auto = config.get("friend_auto")
+            return friend_auto
+        except Exception as e:
+            logging.error(f"读取配置文件错误：{e}")
+            return False
+
+    def get_allow_group(self):
+        try:
+            with open("./config.toml", "r", encoding="utf-8") as f:
+                config = toml.load(f)
+                allow_group = config.get("allowed_groups")
+            return allow_group
+        except Exception as e:
+            logging.error(f"读取配置文件错误：{e}")
+            return []