Using shlex.split instead of pythons own split. (#63)

olijeffers0n · Mar 15, 2024 · 20b4a1d · 20b4a1d
1 parent b4925d8
commit 20b4a1d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -6,4 +6,5 @@ http-ece
 requests
 numpy
 scipy
-betterproto==2.0.0b6
+betterproto==2.0.0b6
+shlex
diff --git a/rustplus/commands/command_handler.py b/rustplus/commands/command_handler.py
@@ -1,4 +1,5 @@
 import asyncio
+import shlex
 from datetime import datetime
 
 from . import Command, CommandTime
@@ -22,7 +23,7 @@ def register_command(self, data: CommandData) -> None:
 
     async def run_command(self, message: RustChatMessage, prefix) -> None:
         if prefix == self.command_options.prefix:
-            command = message.message.split(" ")[0][len(prefix) :]
+            command = shlex.split(message.message)[0][len(prefix) :]
         else:
             command = prefix
 
@@ -35,7 +36,7 @@ async def run_command(self, message: RustChatMessage, prefix) -> None:
                     message.steam_id,
                     CommandTime(datetime.utcfromtimestamp(message.time), message.time),
                     command,
-                    message.message.split(" ")[1:],
+                    shlex.split(message.message)[1:],
                 )
             )
         else:
@@ -52,7 +53,7 @@ async def run_command(self, message: RustChatMessage, prefix) -> None:
                                 datetime.utcfromtimestamp(message.time), message.time
                             ),
                             command,
-                            message.message.split(" ")[1:],
+                            shlex.split(message.message)[1:],
                         ),
                     )
                     break