add ns to debug_log and grep_logs

Author: mplsgrant

src/warnet/bitcoin.py

@@ -50,11 +50,14 @@ def _rpc(tank: str, method: str, params: str, namespace: Optional[str] = None):
 
 @bitcoin.command()
 @click.argument("tank", type=str, required=True)
-def debug_log(tank: str):
+@click.option("--namespace", default=None, show_default=True)
+def debug_log(tank: str, namespace: Optional[str]):
     """
     Fetch the Bitcoin Core debug log from <tank pod name>
     """
-    cmd = f"kubectl logs {tank}"
+    if not namespace:
+        namespace = get_default_namespace()
+    cmd = f"kubectl logs {tank} --namespace {namespace}"
     try:
         print(run_command(cmd))
     except Exception as e:
@@ -77,8 +80,12 @@ def grep_logs(pattern: str, show_k8s_timestamps: bool, no_sort: bool):
         sys.exit(1)
 
     matching_logs = []
+    longest_namespace_len = 0
 
     for tank in tanks:
+        if len(tank.metadata.namespace) > longest_namespace_len:
+            longest_namespace_len = len(tank.metadata.namespace)
+
         pod_name = tank.metadata.name
         # Get container names for this pod
         containers = tank.spec.containers
@@ -91,31 +98,35 @@ def grep_logs(pattern: str, show_k8s_timestamps: bool, no_sort: bool):
             continue
 
         # Get logs from the specific container
-        command = f"kubectl logs {pod_name} -c {container_name} --timestamps"
+        command = f"kubectl logs {pod_name} -c {container_name} --timestamps --namespace {tank.metadata.namespace}"
         logs = run_command(command)
 
         if logs is not False:
             # Process logs
             for log_entry in logs.splitlines():
                 if re.search(pattern, log_entry):
-                    matching_logs.append((log_entry, pod_name))
+                    matching_logs.append((log_entry, tank.metadata.namespace, pod_name))
 
     # Sort logs if needed
     if not no_sort:
         matching_logs.sort(key=lambda x: x[0])
 
     # Print matching logs
-    for log_entry, pod_name in matching_logs:
+    for log_entry, namespace, pod_name in matching_logs:
         try:
             # Split the log entry into Kubernetes timestamp, Bitcoin timestamp, and the rest of the log
             k8s_timestamp, rest = log_entry.split(" ", 1)
             bitcoin_timestamp, log_message = rest.split(" ", 1)
 
             # Format the output based on the show_k8s_timestamps option
             if show_k8s_timestamps:
-                print(f"{pod_name}: {k8s_timestamp} {bitcoin_timestamp} {log_message}")
+                print(
+                    f"{pod_name} {namespace:<{longest_namespace_len}} {k8s_timestamp} {bitcoin_timestamp} {log_message}"
+                )
             else:
-                print(f"{pod_name}: {bitcoin_timestamp} {log_message}")
+                print(
+                    f"{pod_name} {namespace:<{longest_namespace_len}} {bitcoin_timestamp} {log_message}"
+                )
         except ValueError:
             # If we can't parse the timestamps, just print the original log entry
             print(f"{pod_name}: {log_entry}")