forget localisation data dict in generate_location.py

Universite-Gustave-Eiffel · Sep 23, 2024 · 782d12f · 782d12f
1 parent 059e528
commit 782d12f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/services/telit/generate_location.py b/services/telit/generate_location.py
@@ -42,6 +42,10 @@ def main():
     args = parser.parse_args()
     time_part = datetime.datetime.now().strftime(args.time_format)
     document = {"date": time_part,"TPV": {"lat": args.latitude,"lon": args.longitude,"alt": args.altitude},"hwa": args.hwa}
+    # create special entry specifically for elastic search
+    document["location"] = {"lat": document["TPV"]["lat"],
+                            "lon": document["TPV"]["lon"],
+                            "z": document["TPV"]["alt"]}
     document_path = os.path.join(args.output, "location_"+time_part+".json.gz")
     exists = os.path.exists(document_path)
     with open_file_for_write(document_path) as fp:

diff --git a/services/telit/zero_location.py b/services/telit/zero_location.py
@@ -226,7 +226,7 @@ def main():
                                     break
                                 document[result["class"]] = result
                             # Read stuff from telit
-                            if "TPV" in document.keys() and "lat" in document["TPV"].keys():
+                            if "TPV" in document.keys() and {"lat", "lon", "alt"}.issubset(document["TPV"].keys()):
                                 # create special entry specifically for elastic search
                                 document["location"] = {"lat": document["TPV"]["lat"],
                                                         "lon": document["TPV"]["lon"],