def _delete_entities(self, entity_ids, user=None): for entity_id in entity_ids: entity_id = str(entity_id) try: entity = self._db[entity_id] if isinstance(entity, Resource): for l in entity.links: self._db.pop(l.id, None) elif isinstance(entity, Link): try: entity.source.links.remove(entity) except ValueError: pass del self._db[entity_id] except KeyError: raise Entity.DoesNotExist(entity_id)
def get_entity(self, entity_id, user=None): entity_id = str(entity_id) try: return self._db[entity_id] except KeyError: raise Entity.DoesNotExist(entity_id)