From 68ee0a995fceaf6aefdd3c8f780e46a83b51a0e8 Mon Sep 17 00:00:00 2001
From: zsloan
Date: Mon, 19 Apr 2021 17:11:41 +0000
Subject: Specify only getting the first two items after splitting the
 trait/dataset input string

---
 wqflask/utility/helper_functions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'wqflask')

diff --git a/wqflask/utility/helper_functions.py b/wqflask/utility/helper_functions.py
index 7eb7f013..46eeb35d 100644
--- a/wqflask/utility/helper_functions.py
+++ b/wqflask/utility/helper_functions.py
@@ -40,7 +40,7 @@ def get_trait_db_obs(self, trait_db_list):
         data, _separator, hmac_string = trait.rpartition(':')
         data = data.strip()
         assert hmac_string==hmac.hmac_creation(data), "Data tampering?"
-        trait_name, dataset_name = data.split(":")
+        trait_name, dataset_name = data.split(":")[:2]
         if dataset_name == "Temp":
             dataset_ob = data_set.create_dataset(dataset_name=dataset_name, dataset_type="Temp", group_name=trait_name.split("_")[2])
         else:
-- 
cgit v1.2.3