def test_delete(self): bucket = Bucket(self.path, self.ids.pop) self.assertEqual(bucket.put(b'some data'), 'id_0000') self.assertEqual(bucket.put(b'some more data'), 'id_0001') self.assertEqual(sorted(bucket.keys()), ['id_0000', 'id_0001']) for item_id in bucket.keys(): bucket.delete(item_id) self.assertEqual(sorted(bucket.keys()), [])
def add_image(self, *_, **kwargs): """accept uploaded images and attach them to the record""" dummy = zoom.Record( filename='dummy.png', file=io.StringIO('test'), ) # put the uploaded image data in a bucket path = os.path.join(zoom.system.site.data_path, 'buckets') bucket = Bucket(path) f = kwargs.get('file', dummy) name = f.filename data = f.file.read() item_id = bucket.put(data) # create an image record image = Image( image_id=item_id, image_size=len(data), image_name=name, draft=True, ) images = zoom.store.store_of(Image) images.put(image) return item_id
def add_image(self, *_, **kwargs): """accept uploaded images and attach them to the record""" dummy = Record( filename='dummy.png', file=io.StringIO('test'), ) # put the uploaded image data in a bucket path = os.path.join(zoom.system.site.data_path, 'buckets') bucket = Bucket(path) f = kwargs.get('file', dummy) name = f.filename data = f.file.read() item_id = bucket.put(data) # create an attachment record for this bucket c = self.collection field_name = kwargs.get('field_name', 'unknown') field_value = kwargs.get('field_value', 'unknown') attachment = Attachment( record_kind=c.store.kind, field_name=field_name, field_value=field_value, attachment_id=item_id, attachment_size=len(data), attachment_name=name, ) attachments = zoom.store.store_of(Attachment) attachments.put(attachment) return item_id
def add_image(self, *a, **k): """add a ImagesField image""" path = os.path.join(system.site.data_path, 'buckets') bucket = Bucket(path) f = k.get('file') name = f.filename data = f.file.read() item_id = bucket.put(data) return item_id
def test_keys(self): bucket = Bucket(self.path, self.ids.pop) self.assertEqual(bucket.put(b'some data'), 'id_0000') self.assertEqual(bucket.put(b'some more data'), 'id_0001') self.assertEqual(sorted(bucket.keys()), ['id_0000', 'id_0001'])
def test_item_exists(self): bucket = Bucket(self.path, self.ids.pop) item_id = bucket.put(b'some data') self.assertEqual(item_id, 'id_0000') self.assertEqual(bucket.exists(item_id), True)
def test_put_get(self): bucket = Bucket(self.path, self.ids.pop) item_id = bucket.put(b'some data') self.assertEqual(item_id, 'id_0000') self.assertEqual(bucket.get(item_id), b'some data')