From 898b80d34442c16399366a5744cc26a157a727cb Mon Sep 17 00:00:00 2001
From: zsloan
Date: Fri, 20 Aug 2021 17:46:20 +0000
Subject: Fixed some logic in trait.py that could cause a problem if a
 dataset_name was passed that wasn't Temp

---
 wqflask/base/trait.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'wqflask/base')

diff --git a/wqflask/base/trait.py b/wqflask/base/trait.py
index 10851e00..96a09302 100644
--- a/wqflask/base/trait.py
+++ b/wqflask/base/trait.py
@@ -27,11 +27,13 @@ def create_trait(**kw):
 
     assert bool(kw.get('name')), "Needs trait name"
 
-    if kw.get('dataset_name'):
+    if bool(kw.get('dataset')):
+        dataset = kw.get('dataset')
+    else:
         if kw.get('dataset_name') != "Temp":
             dataset = create_dataset(kw.get('dataset_name'))
-    else:
-        dataset = kw.get('dataset')
+        else:
+            dataset = create_dataset("Temp", group_name=kw.get('group_name'))
 
     if dataset.type == 'Publish':
         permissions = check_resource_availability(
-- 
cgit v1.2.3