Return set_queue_length result as a bool and not str and change param…

…eter name
MarechJ · May 22, 2024 · 9de1b87 · 9de1b87
1 parent 1c32221
commit 9de1b87
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/rcon/commands.py b/rcon/commands.py
@@ -552,8 +552,10 @@ def set_autobalance_threshold(self, max_diff: int):
     def set_team_switch_cooldown(self, minutes):
         return self._str_request(f"setteamswitchcooldown {minutes}", log_info=True)
 
-    def set_queue_length(self, num):
-        return self._str_request(f"setmaxqueuedplayers {num}", log_info=True)
+    def set_queue_length(self, value: int) -> bool:
+        return (
+            self._str_request(f"setmaxqueuedplayers {value}", log_info=True) == SUCCESS
+        )
 
     def set_vip_slots_num(self, num):
         return self._str_request(f"setnumvipslots {num}", log_info=True)

diff --git a/rcon/rcon.py b/rcon/rcon.py
@@ -922,9 +922,9 @@ def set_max_ping_autokick(self, max_ms) -> bool:
     def get_queue_length(self) -> int:
         return int(super().get_queue_length())
 
-    def set_queue_length(self, num) -> str:
+    def set_queue_length(self, value: int) -> bool:
         with invalidates(Rcon.get_queue_length):
-            return super().set_queue_length(num)
+            return super().set_queue_length(value)
 
     @ttl_cache(ttl=60 * 10)
     def get_vip_slots_num(self) -> int: