Release 0.9.0

bot/modules/players/actions/kick_player.py

@@ -0,0 +1,107 @@
+from bot import loaded_modules_dict
+from bot import telnet_prefixes
+from bot.logger import get_logger
+from os import path, pardir
+from time import time, sleep
+import re
+
+module_name = path.basename(path.normpath(path.join(path.abspath(__file__), pardir, pardir)))
+action_name = path.basename(path.abspath(__file__))[:-3]
+logger = get_logger("players.kick_player")
+
+
+def main_function(module, event_data, dispatchers_steamid):
+    action = event_data[1].get("action", None)
+    event_data[1]["action_identifier"] = action_name
+    action_is_confirmed = event_data[1].get("confirmed", "False")
+    player_to_be_kicked = event_data[1].get("steamid", None)
+
+    if action == "kick_player":
+        if action_is_confirmed == "True":
+            timeout = 5  # [seconds]
+            timeout_start = time()
+
+            reason = event_data[1].get("reason")
+
+            # Get player entity ID - game requires entity ID instead of steamid
+            dataset = module.dom.data.get("module_game_environment", {}).get("active_dataset", None)
+            player_dict = (
+                module.dom.data
+                .get("module_players", {})
+                .get("elements", {})
+                .get(dataset, {})
+                .get(player_to_be_kicked, {})
+            )
+            player_entity_id = player_dict.get("id")
+
+            if not player_entity_id:
+                module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
+                return
+
+            command = "kick {} \"{}\"".format(player_entity_id, reason)
+            """
+            i was trying re.escape, string replacements... the only thing that seems to work is all of them together
+            Had some big trouble filtering out stuff like ^ and " and whatnot
+            """
+            regex = (
+                telnet_prefixes["telnet_log"]["timestamp"] +
+                r"Executing\scommand\s\'" + re.escape(command) + r"\'\s"
+                r"by\sTelnet\s"
+                r"from\s(?P<called_by>.*)"
+            ).replace('"', '\\"')
+
+            logger.debug("kick_command_prepared",
+                        command=command,
+                        user=dispatchers_steamid,
+                        target=player_to_be_kicked,
+                        reason=reason)
+
+            if not module.telnet.add_telnet_command_to_queue(command):
+                module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
+                return
+
+            poll_is_finished = False
+            while not poll_is_finished and (time() < timeout_start + timeout):
+                sleep(0.25)
+                match = False
+                for match in re.finditer(regex, module.telnet.telnet_buffer, re.DOTALL):
+                    poll_is_finished = True
+
+                if match:
+                    module.callback_success(callback_success, module, event_data, dispatchers_steamid, match)
+                    return
+
+        else:
+            module.set_current_view(dispatchers_steamid, {
+                "current_view": "kick-modal",
+                "current_view_steamid": player_to_be_kicked
+            })
+            return
+
+    elif action == "cancel_kick_player":
+        module.callback_success(callback_success, module, event_data, dispatchers_steamid)
+        return
+
+    module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
+
+
+def callback_success(module, event_data, dispatchers_steamid, match=None):
+    module.set_current_view(dispatchers_steamid, {
+        "current_view": "frontend",
+    })
+
+
+def callback_fail(module, event_data, dispatchers_steamid):
+    pass
+
+
+action_meta = {
+    "description": "kicks a player",
+    "main_function": main_function,
+    "callback_success": callback_success,
+    "callback_fail": callback_fail,
+    "requires_telnet_connection": True,
+    "enabled": True
+}
+
+loaded_modules_dict["module_" + module_name].register_action(action_name, action_meta)