def do_save(self, attribs):
attribs['data'] = cPickle.dumps(attribs['data'], 1)
if(self.is_clean()):
del attribs['data']
update_query = sql.build_update('session', attribs, {'id':self.id()})
self.debug(repr(update_query))
self._pool.runOperation(update_query)
elif(self.is_new()):
attribs['id'] = self.id()
insert_query = sql.build_insert('session', attribs)
self.debug(repr(insert_query))
self._pool.runOperation(insert_query)
else:
attribs['id'] = self.id()
#replace_query = sql.build_replace('session', attribs)
update_query = sql.build_update('session', attribs, {'id':self.id()})
self.debug(repr(update_query))
self._pool.runOperation(update_query)
def _save(self, storable, factory):
"""
Internal function that is responsible for doing the
actual saving.
"""
if not(storable.is_dirty()):
return False
table = storable.get_table()
data = storable.get_data()
locks_allowed = (not bool(self.req)) or self.req.app.config.get('use_db_locks', True)
use_locks = False
primary_key = factory.get_primary_key()
if(storable.is_new()):
if(factory.uses_guids()):
data[primary_key] = self.fetch_id(storable)
elif(locks_allowed):
use_locks = True
self.pool.runOperation('LOCK TABLES `%s` WRITE' % table)
else:
raise RuntimeError("Broken program logic is trying to lock an unlockable database.")
query = sql.build_insert(table, data)
else:
query = sql.build_update(table, data, {primary_key:storable.get_id()})
self.log(query)
try:
self.pool.runOperation(query)
if not(factory.uses_guids()):
if not(storable.get_id()):
rows = self.pool.runQuery('SELECT MAX(%s) AS `id` FROM `%s`' % (primary_key, table))
new_id = rows[0]['id']
storable.set_id(new_id)
finally:
if(use_locks):
self.pool.runOperation('UNLOCK TABLES')
storable.clean()
storable.set_new(False)
storable.set_factory(factory)
return True
