_logs: fix streaming text issue

Author: mplsgrant

src/warnet/control.py

@@ -369,15 +369,31 @@ def _logs(pod_name: str, follow: bool):
             return  # cancelled by user
 
     try:
-        stream = pod_log(pod_name, container_name=None, follow=follow)
-        for line in stream.stream():
-            print(line.decode("utf-8"), end=None)
+        stream = pod_log(pod_name, container_name=None, follow=follow, namespace=pod_namespace)
+        for line in iter_lines_from_stream(stream):
+            click.secho(line)
     except Exception as e:
         print(e)
     except KeyboardInterrupt:
         print("Interrupted streaming log!")
 
 
+def iter_lines_from_stream(log_stream, encoding="utf-8"):
+    decoder = codecs.getincrementaldecoder(encoding)()
+    buffer = ""
+    for chunk in log_stream.stream():
+        # Decode the chunk incrementally
+        text = decoder.decode(chunk)
+        buffer += text
+        # Split the buffer into lines
+        lines = buffer.split("\n")
+        buffer = lines.pop()  # Last item is incomplete line or empty
+        yield from lines
+    # Yield any remaining text in the buffer
+    if buffer:
+        yield buffer
+
+
 @click.command()
 @click.argument("tank_name", required=False)
 @click.option("--all", "-a", "snapshot_all", is_flag=True, help="Snapshot all running tanks")