def test_get_new_name():
samples = {
'./file.jpg': './HASH.jpg',
'/path/file.jpg': '/path/HASH.jpg',
'~/file.with.many.dots': '~/HASH.dots',
'/foo/noextension': '/foo/HASH'
}
for filename in samples:
expected_new_name = samples[filename]
new_name = cli.get_new_name(filename, 'HASH')
assert new_name == expected_new_name
def test_cli_file_rename_dry_run(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
new_name = cli.get_new_name(filename, imghash)
with runner.isolated_filesystem():
if os.path.exists(new_name):
os.remove(new_name)
image.save(filename, format='JPEG')
result = runner.invoke(cli.main, [filename, '--rename', '--dry-run'])
assert result.exit_code == 0
assert not os.path.exists(new_name)
def test_cli_multiple_rename(runner, image):
num_images = 5
filenames = ['/tmp/test-%d.jpg' % (n) for n in range(num_images)]
imghash = cli.get_hash(image)
with runner.isolated_filesystem():
for path in filenames:
if os.path.exists(path):
os.remove(path)
image.save(path, format='JPEG')
result = runner.invoke(cli.main, filenames + ['--rename'])
assert result.exit_code == 0
for path in filenames:
new_name = cli.get_new_name(path, imghash)
assert os.path.exists(new_name)
assert not os.path.exists(path)
# clean up
for path in filenames:
new_name = cli.get_new_name(path, imghash)
# hashes might be the same
if os.path.exists(new_name):
os.remove(new_name)
def test_rename_with_template():
oldname = '/foo/bar/path/file.name.ext'
templates = {
'{path}/{name}{ext}': '/foo/bar/path/file.name.ext',
'{path}/{hash}{ext}': '/foo/bar/path/HASH.ext',
'/tmp/{hash}{ext}': '/tmp/HASH.ext',
'/tmp/{name}-{hash}.jpg': '/tmp/file.name-HASH.jpg',
None: '/foo/bar/path/HASH.ext'
}
for template in templates:
expected_new_name = templates[template]
new_name = cli.get_new_name(oldname, 'HASH', template)
assert new_name == expected_new_name
def test_cli_file_rename_bad_filename(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
template = '/tmp/test.jpg/not_a_directory'
new_name = cli.get_new_name(filename, imghash, template)
with runner.isolated_filesystem():
if os.path.exists(new_name):
os.remove(new_name)
image.save(filename, format='JPEG')
result = runner.invoke(cli.main,
[filename, '--rename', '--template', template])
assert result.exception
assert not os.path.exists(new_name)
def test_cli_file_rename_template(runner, image):
filename = '/tmp/test.jpg'
imghash = cli.get_hash(image, 'average')
template = '{path}/FOOBAR-{name}-{hash}{ext}'
new_name = cli.get_new_name(filename, imghash, template)
with runner.isolated_filesystem():
if os.path.exists(new_name):
os.remove(new_name)
image.save(filename, format='JPEG')
result = runner.invoke(cli.main,
[filename, '--rename', '--template', template])
assert result.exit_code == 0
assert os.path.exists(new_name)
# clean up the new file
os.remove(new_name)