From 3ab1ad32ced83f9051ccad73ab9e56245de93809 Mon Sep 17 00:00:00 2001
From: Bea Steers <bea.steers@gmail.com>
Date: Tue, 8 Oct 2024 23:24:26 -0400
Subject: [PATCH] add file command

---
 ptgctl/tools/display.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/ptgctl/tools/display.py b/ptgctl/tools/display.py
index 9355a6a..35f5230 100644
--- a/ptgctl/tools/display.py
+++ b/ptgctl/tools/display.py
@@ -130,9 +130,7 @@ async def file(api, stream_id, out_dir, include_timestamps=False, **kw):
                     if sid not in files:
                         files[sid] = stack.enter_context(open(os.path.join(out_dir, f'{sid}.txt'), 'w'))
                         pbars[sid] = tqdm.tqdm(desc=sid)
-                    if include_timestamps:
-                        files[sid].write(f'{ts}:')
-                    files[sid].write(f"{data.decode('utf-8')}\n")
+                    files[sid].write(f"{f'{ts}:' if include_timestamps else ''}{data.decode('utf-8')}\n")
                     pbars[sid].update()
                     pbars[sid].set_description(f'{sid}: {str(data)[:20]}')