Update address_ping_system.py

Naruno · Aug 1, 2023 · 6e63625 · 6e63625
1 parent f4aa312
commit 6e63625
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/address_ping_system/address_ping_system.py b/src/address_ping_system/address_ping_system.py
@@ -18,6 +18,15 @@
 
 
 ping_db = KOT("ping_db")
+settings_db = KOT("settings_db")
+
+
+def max_user():
+    record = settings_db.get("max_user")
+    if record is None:
+        return 1000
+    return record
+
 
 class aps:
     command_line = False
@@ -79,7 +88,7 @@ def run(self):
             data = self.integration.get()
             if data != []:
                 for each in data:
-                    if ping_db.get(each["fromUser"]) != True and ping_db.get_count() < 1:
+                    if ping_db.get(each["fromUser"]) != True and ping_db.get_count() < max_user():
                         self.integration.send("reply", "hello", each["fromUser"])
                         ping_db.set(each["fromUser"], True)
             time.sleep(5)