def get_insert_settings(self):
defaults = {
'durability': 'hard',
'return_changes': False,
'conflict': 'error'
}
return util.extend(defaults, self.optargs)
def get_update_settings(self):
defaults = {
'durability': 'hard',
'return_changes': False,
'non_atomic': False
}
return util.extend(defaults, self.optargs)
def test_extend(self):
dict_1 = {'x': 'x1-val', 'y': 'y1-val'}
dict_2 = {'x': 'x2-val', 'z': 'z2-val'}
extended = util.extend(dict_1, dict_2)
assertEqual({
'x': 'x2-val',
'y': 'y1-val',
'z': 'z2-val'
}, extended)
assertEqual({
'x': 'x1-val',
'y': 'y1-val'
}, dict_1)
assertEqual({
'x': 'x2-val',
'z': 'z2-val'
}, dict_2)
def do_run(self, table, lower_key, upper_key, arg, scope):
defaults = {
'left_bound': 'closed',
'right_bound': 'open',
'index': 'id'
}
options = util.extend(defaults, self.optargs)
if options['index'] == 'id':
map_fn = util.getter('id')
else:
map_fn, _ = self.find_index_func_for_scope(options['index'], arg)
left_test, right_test = operators_for_bounds(options['left_bound'],
options['right_bound'])
for document in table:
doc_val = map_fn(document)
if left_test(doc_val, lower_key) and right_test(
doc_val, upper_key):
yield document
def do_run(self, sequence, arg, scope):
out = []
for elem in sequence:
out.append(util.extend(elem['left'], elem['right']))
return out
def ensure_id(elem):
if ('id' not in elem) or (elem['id'] is None):
uid = text_type(uuid.uuid4())
elem = util.extend(elem, {'id': uid})
generated_keys.append(uid)
return elem
def get_delete_settings(self):
defaults = {'durability': 'hard', 'return_changes': False}
return util.extend(defaults, self.optargs)
def do_run(self, to_test, left, right, arg, scope):
defaults = {'left_bound': 'closed', 'right_bound': 'open'}
options = util.extend(defaults, self.optargs)
left_test, right_test = operators_for_bounds(options['left_bound'],
options['right_bound'])
return (left_test(to_test, left) and right_test(to_test, right))