use dict.get() in kml

asfadmin · May 7, 2024 · 151b1f2 · 151b1f2
1 parent 967ffc1
commit 151b1f2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/asf_search/export/kml.py b/asf_search/export/kml.py
@@ -139,12 +139,12 @@ def getItem(self, p):
     # Helper method for getting additional fields in <ul> tag
     def metadata_fields(self, item: Dict):
         required = {
-            'Processing type: ': item['processingTypeDisplay'],
-            'Frame: ': item['frameNumber'],
-            'Path: ': item['pathNumber'],
-            'Orbit: ': item['orbit'],
-            'Start time: ': item['startTime'],
-            'End time: ': item['stopTime'],
+            'Processing type: ': item.get('processingTypeDisplay'),
+            'Frame: ': item.get('frameNumber'),
+            'Path: ': item.get('pathNumber'),
+            'Orbit: ': item.get('orbit'),
+            'Start time: ': item.get('startTime'),
+            'End time: ': item.get('stopTime'),
         }
 
         optional = {}

diff --git a/asf_search/search/search_generator.py b/asf_search/search/search_generator.py
@@ -289,7 +289,7 @@ def as_ASFProduct(item: Dict, session: ASFSession) -> ASFProduct:
 
     # if there's a direct entry in our dataset to product type dict
     # perf = time.time()
-    subclass = dataset_collections.get(product_type_key)
+    subclass = dataset_to_product_types.get(product_type_key)
     if subclass is not None:
         # ASF_LOGGER.warning(f'subclass selection time {time.time() - perf}')
         return subclass(item, session=session)