def execute(self, sql, args=None): self.logger.info(self.mogrify(sql, args)) try: cursor.execute(self, sql, args) except Exception as exc: self.logger.error("%s: %s" % (exc.__class__.__name__, exc)) raise
def execute(self, sql, args=None): self.logger.info(self.mogrify(sql, args)) try: cursor.execute(self, sql, args) except Exception as exc: self.logger.error("%s: %s" % (exc.__class__.__name__, exc)) raise
def test_get_and_set(self): queue = self.make_one("test") queue.put({'foo': 'bar'}) job = queue.get() self.assertEqual(job.data, {'foo': 'bar'}) job.data = {'foo': 'boo'} with queue._transaction() as cursor: cursor.execute("SELECT data FROM %s WHERE id = %s", (queue.table, job.id)) data = cursor.fetchone()[0] self.assertEqual(job.data, data) self.assertIn('size=%d' % len(dumps(data)), repr(job))
def test_get_and_set(self): queue = self.make_one("test") queue.put({'foo': 'bar'}) task = queue.get() self.assertEqual(task.data, {'foo': 'bar'}) task.data = {'foo': 'boo'} with queue._transaction() as cursor: cursor.execute( "SELECT data FROM %s WHERE id = %s", (queue.table, task.id) ) data = cursor.fetchone()[0] self.assertEqual(task.data, data) self.assertIn('size=%d' % len(dumps(data)), repr(task))
def test_context_manager_get_and_set(self): queue = self.make_one("test") queue.put({'foo': 'bar'}) with queue: task = queue.get() self.assertEqual(task.data, {'foo': 'bar'}) task.data = {'foo': 'boo'} with queue as cursor: cursor.execute("SELECT data FROM %s WHERE id = %s", (queue.table, task.id)) data = cursor.fetchone()[0] self.assertEqual(task.data, data) self.assertIn('size=%d' % len(dumps(data)), repr(task))
def test_context_manager_get_and_set(self): queue = self.make_one("test") queue.put({'foo': 'bar'}) with queue: job = queue.get() self.assertEqual(job.data, {'foo': 'bar'}) job.data = {'foo': 'boo'} with queue as cursor: cursor.execute( "SELECT data FROM %s WHERE id = %s", (queue.table, job.id) ) data = cursor.fetchone()[0] self.assertEqual(job.data, data) self.assertIn('size=%d' % len(dumps(data)), repr(job))