def _custom_upload(f, profile, label, conf):
t = adjust_image(f,
max_size=conf['MAX_SIZE'],
new_format=conf['FORMAT'],
jpeg_quality=conf['JPEG_QUALITY'],
fill=conf['FILL'],
stretch=conf['STRETCH'],
return_new_image=True)
img_id = generate_img_id(profile,
ext=image_get_format(f),
label=label,
tmp=True)
relative_path = get_relative_path_from_img_id(img_id)
full_path = media_path(relative_path)
save_file(t, full_path)
for v_conf in conf['VARIANTS']:
v_label = v_conf['LABEL']
if not v_label:
v_label = get_variant_label(v_conf)
v_t = adjust_image(t,
max_size=v_conf['MAX_SIZE'],
new_format=v_conf['FORMAT'],
jpeg_quality=v_conf['JPEG_QUALITY'],
fill=v_conf['FILL'],
stretch=v_conf['STRETCH'],
return_new_image=True)
v_relative_path = get_relative_path_from_img_id(
img_id, variant_label=v_label, ext=image_get_format(v_t))
v_full_path = media_path(v_relative_path)
save_file(v_t, v_full_path)
return img_id
def upload_from_fs(fn, profile=None, label=None):
"""
Saves image from fn with TMP prefix and returns img_id.
"""
if not os.path.isfile(fn):
raise ValueError('File is not exists: {}'.format(fn))
if profile is None:
profile = 'default'
conf = get_profile_configs(profile)
with open(fn, 'rb') as f:
if not is_image(f, types=conf['TYPES']):
msg = (('Format of uploaded file "%(name)s" is not allowed. '
'Allowed formats is: %(formats)s.') %
{'name': fn, 'formats': ', '.join(map(lambda t: t.upper(), conf['TYPES']))})
raise RuntimeError(msg)
t = adjust_image(f, max_size=conf['MAX_SIZE'], new_format=conf['FORMAT'],
jpeg_quality=conf['JPEG_QUALITY'], fill=conf['FILL'],
stretch=conf['STRETCH'], return_new_image=True)
img_id = generate_img_id(profile, ext=image_get_format(f), label=label, tmp=True)
relative_path = get_relative_path_from_img_id(img_id)
full_path = media_path(relative_path)
save_file(t, full_path)
for v_conf in conf['VARIANTS']:
v_label = v_conf['LABEL']
if not v_label:
v_label = get_variant_label(v_conf)
v_t = adjust_image(t, max_size=v_conf['MAX_SIZE'], new_format=v_conf['FORMAT'],
jpeg_quality=v_conf['JPEG_QUALITY'], fill=v_conf['FILL'],
stretch=v_conf['STRETCH'], return_new_image=True)
v_relative_path = get_relative_path_from_img_id(img_id, variant_label=v_label,
ext=image_get_format(v_t))
v_full_path = media_path(v_relative_path)
save_file(v_t, v_full_path)
return img_id
def test_save_file(self):
f = get_img_file(create_test_image(1000, 1000))
img_id = generate_img_id('simple1', ext=image_get_format(f), label='test-save-file')
relative_path = get_relative_path_from_img_id(img_id)
full_path = media_path(relative_path)
save_file(f, full_path)
file_exists = os.path.exists(full_path)
self.assertTrue(file_exists)
if file_exists:
f.seek(0)
h1 = hashlib.md5(f.read()).hexdigest()
h2 = hashlib.md5(open(full_path, 'rb').read()).hexdigest()
self.assertEqual(h1, h2)
def test_save_file(self):
f = get_img_file(create_test_image(1000, 1000))
img_id = generate_img_id('simple1',
ext=image_get_format(f),
label='test-save-file')
relative_path = get_relative_path_from_img_id(img_id)
full_path = media_path(relative_path)
save_file(f, full_path)
file_exists = os.path.exists(full_path)
self.assertTrue(file_exists)
if file_exists:
f.seek(0)
h1 = hashlib.md5(f.read()).hexdigest()
h2 = hashlib.md5(open(full_path, 'rb').read()).hexdigest()
self.assertEqual(h1, h2)
profile = request.POST.get('profile', 'default')
conf = get_profile_configs(profile)
except ValueError, e:
result['errors'].append(unicode(e))
return send_json(result)
for i in xrange(min(len(files), dju_settings.DJU_IMG_UPLOAD_MAX_FILES)):
f = files[i]
if not is_image(f, types=conf['TYPES']):
result['errors'].append(
unicode(ERROR_MESSAGES['wrong_file_format']) %
{'name': f.name, 'formats': ', '.join(map(lambda t: t.upper(), conf['TYPES']))}
)
continue
adjust_image(f, max_size=conf['MAX_SIZE'], new_format=conf['FORMAT'],
jpeg_quality=conf['JPEG_QUALITY'], fill=conf['FILL'], stretch=conf['STRETCH'])
img_id = generate_img_id(profile, ext=image_get_format(f),
label=request.POST.get('label'), tmp=True)
relative_path = get_relative_path_from_img_id(img_id)
full_path = media_path(relative_path)
save_file(f, full_path)
data = {
'url': settings.MEDIA_URL + relative_path,
'rel_url': relative_path,
'img_id': img_id,
'variants': {},
}
for v_conf in conf['VARIANTS']:
label = v_conf['LABEL']
if not label:
label = get_variant_label(v_conf)
v_f = adjust_image(f, max_size=v_conf['MAX_SIZE'], new_format=v_conf['FORMAT'],
f = files[i]
if not is_image(f, types=conf["TYPES"]):
result["errors"].append(
unicode(ERROR_MESSAGES["wrong_file_format"])
% {"name": f.name, "formats": ", ".join(map(lambda t: t.upper(), conf["TYPES"]))}
)
continue
adjust_image(
f,
max_size=conf["MAX_SIZE"],
new_format=conf["FORMAT"],
jpeg_quality=conf["JPEG_QUALITY"],
fill=conf["FILL"],
stretch=conf["STRETCH"],
)
img_id = generate_img_id(profile, ext=image_get_format(f), label=request.POST.get("label"), tmp=True)
relative_path = get_relative_path_from_img_id(img_id)
full_path = media_path(relative_path)
save_file(f, full_path)
data = {"url": settings.MEDIA_URL + relative_path, "rel_url": relative_path, "img_id": img_id, "variants": {}}
for v_conf in conf["VARIANTS"]:
label = v_conf["LABEL"]
if not label:
label = get_variant_label(v_conf)
v_f = adjust_image(
f,
max_size=v_conf["MAX_SIZE"],
new_format=v_conf["FORMAT"],
jpeg_quality=v_conf["JPEG_QUALITY"],
fill=v_conf["FILL"],
stretch=v_conf["STRETCH"],
def test_bad_format(self):
self.assertIsNone(image_get_format(cStringIO.StringIO('x' * 1000)))
def test_format(self):
self.assertEqual(image_get_format(self.img_jpeg), 'jpeg')
self.assertEqual(image_get_format(self.img_png), 'png')
self.assertEqual(image_get_format(self.img_gif), 'gif')
def test_bad_format(self):
self.assertIsNone(image_get_format(cStringIO.StringIO("x" * 1000)))
def test_format(self):
self.assertEqual(image_get_format(self.img_jpeg), "jpeg")
self.assertEqual(image_get_format(self.img_png), "png")
self.assertEqual(image_get_format(self.img_gif), "gif")