def is_object_none(self, msg): obj_id = msg.object_id if obj_id not in self.object_store._objects: # If the object is not present on the worker, raise an error raise ObjectNotFoundError(obj_id, self) obj = self.get_obj(msg.object_id) return obj is None
def _get_obj(self, obj_id: Union[str, int]) -> object: try: obj = self._objects[obj_id] except KeyError as e: if obj_id not in self._objects: # Try to recover object on database obj = redis_db.hget(self.id, obj_id) if obj: self._objects[obj_id] = deserialize(obj) else: raise ObjectNotFoundError(obj_id, self) else: raise e return obj
def get_obj(self, obj_id: Union[str, int]) -> object: """Returns the object from registry. Look up an object from the registry using its ID. Args: obj_id: A string or integer id of an object to look up. Returns: Object with id equals to `obj_id`. """ try: obj = self._objects[obj_id] except KeyError as e: if obj_id not in self._objects: raise ObjectNotFoundError(obj_id, self) else: raise e return obj