From: Tony Mack <tmack@cs.princeton.edu>
Date: Mon, 14 Sep 2009 17:08:07 +0000 (+0000)
Subject: fix bug in insert()
X-Git-Tag: sfa-0.9-2~26
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=e55b669b21ef45d55f788f85674e9715b2ea350b;p=sfa.git

fix bug in insert()
---

diff --git a/sfa/util/genitable.py b/sfa/util/genitable.py
index a6c6c9f8..ffeab21c 100644
--- a/sfa/util/genitable.py
+++ b/sfa/util/genitable.py
@@ -82,8 +82,15 @@ class GeniTable(list):
                        "VALUES(" + ",".join(fieldvals) + ")"
         #print query_str
         self.cnx.query(query_str)
-        results = self.find(record)
-        return results[0]['record_id']
+        result = self.find({'hrn': record['hrn'], 'type': record['type']})
+        if not result:
+            record_id = None
+        elif isinstance(result, list):
+            record_id = result[0]['record_id']
+        else:
+            record_id = result['record_id']
+
+        return record_id
 
     def update(self, record):
         dont_update = ['date_created', 'last_updated', 'record_id']