fix: add origin to lbsn mapping input scheme

Sieboldianus · Jan 19, 2020 · be7ea94 · be7ea94
1 parent dcd096e
commit be7ea94
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lbsntransform/output/hll/shared_structure_proto_hlldb.py b/lbsntransform/output/hll/shared_structure_proto_hlldb.py
@@ -207,6 +207,7 @@ def make_base(cls, facet: str, base: str, record):
     def get_hll_metrics(cls, record) -> hll.HllMetrics:
         """Extracts hll metrics based on record type"""
         dict_switcher = {
+            lbsn.Origin().DESCRIPTOR.name: cls.get_origin_metrics,
             lbsn.Country().DESCRIPTOR.name: cls.get_country_metrics,
             lbsn.City().DESCRIPTOR.name: cls.get_city_metrics,
             lbsn.Place().DESCRIPTOR.name: cls.get_place_metrics,
@@ -220,6 +221,11 @@ def get_hll_metrics(cls, record) -> hll.HllMetrics:
         record_hll_metrics = extract_function(record)
         return record_hll_metrics
 
+    @staticmethod
+    def get_origin_metrics(record) -> hll.HllMetrics:
+        """Get hll metrics from lbsn.Origin record"""
+        return
+
     @staticmethod
     def get_country_metrics(record) -> hll.HllMetrics:
         """Get hll metrics from lbsn.Country record"""