From: Eugene Crosser <crosser@average.org>
Date: Fri, 15 Jul 2022 14:06:48 +0000 (+0200)
Subject: qry: initial support for saving data in a file
X-Git-Tag: 1.90~17
X-Git-Url: http://www.average.org/gitweb/?p=loctrkd.git;a=commitdiff_plain;h=a99136725dee12c541e9d866632ea9643005ba25

qry: initial support for saving data in a file
---

diff --git a/loctrkd/qry.py b/loctrkd/qry.py
index e33ba16..650830e 100644
--- a/loctrkd/qry.py
+++ b/loctrkd/qry.py
@@ -43,6 +43,13 @@ def main(
     else:
         proto = ""
         selector = ""
+    dopts = dict(opts)
+    if len(args) > 1 and "-o" in dopts:
+        attr = args[1]
+        fn = dopts["-o"]
+    else:
+        attr = ""
+        fn = ""
 
     c.execute(
         """select tstamp, imei, peeraddr, is_incoming, proto, packet
@@ -64,8 +71,11 @@ def main(
             peeraddr,
             msg,
         )
+        if fn and hasattr(msg, attr):
+            with open(fn, "wb") as fl:  # TODO support multiple files
+                fl.write(getattr(msg, attr))
 
 
 if __name__.endswith("__main__"):
-    opts, args = getopt(argv[1:], "c:d")
+    opts, args = getopt(argv[1:], "o:c:d")
     main(common.init(log, opts=opts), opts, args)