def test_init_no_id(self):
lookup = Lookup(key='key1', value='value1')
self.assertIsNotNone(lookup)
self.assertIsNone(lookup.id)
self.assertEqual('key1', lookup.key)
self.assertEqual('value1', lookup.value)
self.assertEqual({
'key': 'key1',
'value': 'value1'
}, lookup.to_document())
def add_to_lookup(self, key, value, overwrite_existing=False):
collection = self.collection()
lookup = collection.find_one({'key': key})
if lookup is not None:
if overwrite_existing is True:
YLogger.debug(self, "Updating lookup in Mongo [%s] [%s]", key,
value)
lookup['value'] = value
collection.replace_one({'key': key}, lookup)
else:
YLogger.error(self,
"Existing value in Mongo lookup [%s] = [%s]",
key, value)
return False
else:
YLogger.debug(self, "Adding lookup to Mongo [%s] = [%s]", key,
value)
lookup = Lookup(key, value)
self.add_document(lookup)
return True