From e55b669b21ef45d55f788f85674e9715b2ea350b Mon Sep 17 00:00:00 2001 From: Tony Mack Date: Mon, 14 Sep 2009 17:08:07 +0000 Subject: [PATCH] fix bug in insert() --- sfa/util/genitable.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) 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'] -- 2.47.0