Add last_insert_id module to database

This commit is contained in:
JonnyWong16 2016-10-01 11:47:00 -07:00 committed by JonnyWong16
parent 8925a0dc78
commit 7d3711bf5a
2 changed files with 7 additions and 2 deletions

View file

@ -265,8 +265,7 @@ class ActivityProcessor(object):
'reference_id': result[1]['reference_id']} 'reference_id': result[1]['reference_id']}
else: else:
# Get the last insert row id # Get the last insert row id
result = self.db.select(query='SELECT last_insert_rowid() AS last_id') last_id = self.db.last_insert_id()
last_id = result[0]['last_id'] if result else None
query = 'UPDATE session_history SET reference_id = ? WHERE id = ? ' query = 'UPDATE session_history SET reference_id = ? WHERE id = ? '
# If rating_key is the same in the previous session, then set the reference_id to the previous row, else set the reference_id to the new id # If rating_key is the same in the previous session, then set the reference_id to the previous row, else set the reference_id to the new id

View file

@ -203,3 +203,9 @@ class MonitorDatabase(object):
# We want to know if it was an update or insert # We want to know if it was an update or insert
return trans_type return trans_type
def last_insert_id(self):
# Get the last insert row id
result = self.select_single(query='SELECT last_insert_rowid() AS last_id')
if result:
return result.get('last_id', None)