From dc41d8d7dabbb688e5a258fce61e5abcd06c8c66 Mon Sep 17 00:00:00 2001
From: Bruce Flynn <brucef@ssec.wisc.edu>
Date: Fri, 30 Apr 2021 15:24:11 -0500
Subject: [PATCH] add --aqua option to truc
---
edosl0util/cli/trunc.py | 5 ++++-
edosl0util/trunc.py | 4 ++--
2 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/edosl0util/cli/trunc.py b/edosl0util/cli/trunc.py
index 79c5522..7b1ae97 100644
--- a/edosl0util/cli/trunc.py
+++ b/edosl0util/cli/trunc.py
@@ -1,6 +1,7 @@
import os
import io
from edosl0util.cli import util
+from edosl0util import stream
from edosl0util import trunc
@@ -10,12 +11,14 @@ def main():
parser.add_argument("filename")
parser.add_argument("start", type=util.timestamp, help="YYYY-MM-DD HH:MM:SS")
parser.add_argument("end", type=util.timestamp, help="YYYY-MM-DD HH:MM:SS")
+ parser.add_argument("-a", "--aqua", action="store_true", help="Stream is Aqua data")
args = parser.parse_args()
util.configure_logging(args)
+ stream_impl = stream.aqua_packet_stream if args.aqua else stream.jpss_packet_stream
output = args.output or os.path.basename(args.filename) + ".trunc"
with io.open(output, "wb") as fptr:
- for pkt in trunc.trunc_file(args.filename, args.start, args.end):
+ for pkt in trunc.trunc_file(args.filename, args.start, args.end, stream_impl=stream_impl):
fptr.write(pkt.bytes())
diff --git a/edosl0util/trunc.py b/edosl0util/trunc.py
index a0674e6..6e6f8af 100644
--- a/edosl0util/trunc.py
+++ b/edosl0util/trunc.py
@@ -19,6 +19,6 @@ def trunc_stream(stream, start, end):
pkt = stream.next()
-def trunc_file(filename, start, end):
- stream = jpss_packet_stream(io.open(filename, "rb"))
+def trunc_file(filename, start, end, stream_impl=jpss_packet_stream):
+ stream = stream_impl(io.open(filename, "rb"))
return trunc_stream(stream, start, end)
--
GitLab