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