fix: catch empty proto_map

Sieboldianus · Sep 19, 2019 · 23559d5 · 23559d5
1 parent 0bd58aa
commit 23559d5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lbsntransform/classes/helper_functions.py b/lbsntransform/classes/helper_functions.py
@@ -462,8 +462,10 @@ def is_composite_field_container(in_obj):
     def map_to_dict(proto_map):
         """Converts protobuf field map (ScalarMapContainer)
         to Dictionary"""
-        mapped_dict = dict(zip(proto_map.keys(), proto_map.values()))
-        return mapped_dict
+        if proto_map:
+            mapped_dict = dict(zip(proto_map.keys(), proto_map.values()))
+            return mapped_dict
+        return {}
 
     @staticmethod
     def merge_existing_records(oldrecord, newrecord):
@@ -484,7 +486,7 @@ def load_importer_mapping_module(origin: int):
         """
         if origin == 2:
             from .field_mapping_flickr import FieldMappingFlickr as importer
-        if origin == 21:
+        elif origin == 21:
             # Flickr YFCC100M dataset
             from .field_mapping_yfcc100m import FieldMappingYFCC100M as importer
         elif origin == 3: