def test_exists(): """Test the exists() function. """ hooks.reset() assert hooks.exists("alien_invasion") == True assert hooks.exists("worldpeace") == False hooks.register("worldpeace") assert hooks.exists("worldpeace") == True hooks.reset() assert hooks.exists("worldpeace") == False
def _load_data(self): """Download the image while yielding chunks as they are coming in. Called internally when access to the image data is needed. The fact that the data is yielded live means the caller may already start using it before the download is complete. """ # TODO: store bigger files on disk? self._data = StringIO.StringIO() while True: chunk = self.request.read(self.chunk_size) if not chunk: break self._data.write(chunk) # HOOK: FEED_IMAGE_DOWNLOAD_CHUNK if hooks.exists('feed_image_download_chunk'): hooks.trigger('feed_image_download_chunk', args=[self, self.data.tell()]) yield chunk # reset once we initially loaded the data self.data.seek(0)