Code is all tore up

Plugins/UEWingman/ue-wingman.py

@@ -20,19 +20,38 @@ def main():
         print("Usage: ue-wingman.py ShowCommands [key=value ...]")
         sys.exit(1)
 
-    msg = {"command": args[0]}
-    for arg in args[1:]:
-        key, _, value = arg.partition("=")
-        if value.lower() == "true":
-            value = True
-        elif value.lower() == "false":
-            value = False
-        else:
+    no_args_commands = {"ShowCommands", "UserManual"}
+    if len(args) == 1 and args[0] not in no_args_commands:
+        # No extra arguments: read JSON object from stdin.
+        # Accumulate lines and try to parse after each one.
+        decoder = json.JSONDecoder()
+        raw = ""
+        msg = None
+        for line in sys.stdin:
+            raw += line
             try:
-                value = int(value)
-            except ValueError:
-                pass
-        msg[key] = value
+                msg, _ = decoder.raw_decode(raw.lstrip())
+                break
+            except json.JSONDecodeError:
+                continue
+        if msg is None:
+            print("Could not parse a complete JSON object from stdin")
+            sys.exit(1)
+        msg["command"] = args[0]
+    else:
+        msg = {"command": args[0]}
+        for arg in args[1:]:
+            key, _, value = arg.partition("=")
+            if value.lower() == "true":
+                value = True
+            elif value.lower() == "false":
+                value = False
+            else:
+                try:
+                    value = int(value)
+                except ValueError:
+                    pass
+            msg[key] = value
 
     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
     sock.settimeout(TIMEOUT)