ef07f1e7a7
The IPTC spec defines title and description fields so use those to populate these fields on the Image unless they've been set by some other means.
295 lines
10 KiB
Python
295 lines
10 KiB
Python
# -*- coding:utf-8 -*-
|
|
|
|
import locale
|
|
import logging
|
|
import os
|
|
import pytest
|
|
import datetime
|
|
|
|
from os.path import join
|
|
from sigal.gallery import Album, Media, Image, Video, Gallery
|
|
from sigal.video import SubprocessException
|
|
|
|
CURRENT_DIR = os.path.dirname(__file__)
|
|
|
|
REF = {
|
|
'dir1': {
|
|
'title': 'An example gallery',
|
|
'name': 'dir1',
|
|
'thumbnail': 'dir1/test1/thumbnails/11.tn.jpg',
|
|
'subdirs': ['test1', 'test2', 'test3'],
|
|
'medias': [],
|
|
},
|
|
'dir1/test1': {
|
|
'title': 'An example sub-category',
|
|
'name': 'test1',
|
|
'thumbnail': 'test1/thumbnails/11.tn.jpg',
|
|
'subdirs': [],
|
|
'medias': ['11.jpg', 'archlinux-kiss-1024x640.png',
|
|
'flickr_jerquiaga_2394751088_cc-by-nc.jpg',
|
|
'50a1d0bc-763d-457e-b634-c87f16a64270.gif'],
|
|
},
|
|
'dir1/test2': {
|
|
'title': 'test2',
|
|
'name': 'test2',
|
|
'thumbnail': 'test2/thumbnails/21.tn.jpg',
|
|
'subdirs': [],
|
|
'medias': ['21.jpg', '22.jpg', 'archlinux-kiss-1024x640.png'],
|
|
},
|
|
'dir1/test3': {
|
|
'title': '01 First title alphabetically',
|
|
'name': 'test3',
|
|
'thumbnail': 'test3/thumbnails/3.tn.jpg',
|
|
'subdirs': [],
|
|
'medias': ['3.jpg'],
|
|
},
|
|
'dir2': {
|
|
'title': 'Another example gallery with a very long name',
|
|
'name': 'dir2',
|
|
'thumbnail': 'dir2/thumbnails/m57_the_ring_nebula-587px.tn.jpg',
|
|
'subdirs': [],
|
|
'medias': ['exo20101028-b-full.jpg',
|
|
'Hubble Interacting Galaxy NGC 5257.jpg',
|
|
'Hubble ultra deep field.jpg',
|
|
'm57_the_ring_nebula-587px.jpg'],
|
|
},
|
|
u'accentué': {
|
|
'title': u'accentué',
|
|
'name': u'accentué',
|
|
'thumbnail': u'accentué/thumbnails/hélicoïde.tn.jpg',
|
|
'subdirs': [],
|
|
'medias': [u'hélicoïde.jpg', 'superdupont_source_wikipedia_en.jpg'],
|
|
},
|
|
'video': {
|
|
'title': 'video',
|
|
'name': 'video',
|
|
'thumbnail': ('video/thumbnails/'
|
|
'stallman software-freedom-day-low.tn.jpg'),
|
|
'subdirs': [],
|
|
'medias': ['stallman software-freedom-day-low.ogv']
|
|
}
|
|
}
|
|
|
|
|
|
def test_media(settings):
|
|
m = Media('11.jpg', 'dir1/test1', settings)
|
|
path = join('dir1', 'test1')
|
|
file_path = join(path, '11.jpg')
|
|
thumb = join('thumbnails', '11.tn.jpg')
|
|
|
|
assert m.filename == '11.jpg'
|
|
assert m.src_path == join(settings['source'], file_path)
|
|
assert m.dst_path == join(settings['destination'], file_path)
|
|
assert m.thumb_name == thumb
|
|
assert m.thumb_path == join(settings['destination'], path, thumb)
|
|
assert m.title == "Foo Bar"
|
|
assert m.description == "<p>This is a funny description of this image</p>"
|
|
|
|
assert repr(m) == "<Media>('{}')".format(file_path)
|
|
assert str(m) == file_path
|
|
|
|
|
|
def test_media_orig(settings, tmpdir):
|
|
settings['keep_orig'] = False
|
|
m = Media('11.jpg', 'dir1/test1', settings)
|
|
assert m.big is None
|
|
|
|
settings['keep_orig'] = True
|
|
settings['destination'] = str(tmpdir)
|
|
|
|
m = Image('11.jpg', 'dir1/test1', settings)
|
|
assert m.big == 'original/11.jpg'
|
|
|
|
m = Video('stallman software-freedom-day-low.ogv', 'video', settings)
|
|
assert m.filename == 'stallman software-freedom-day-low.webm'
|
|
assert m.big == 'original/stallman software-freedom-day-low.ogv'
|
|
assert os.path.isfile(join(settings['destination'], m.path, m.big))
|
|
|
|
settings['use_orig'] = True
|
|
|
|
m = Image('21.jpg', 'dir1/test2', settings)
|
|
assert m.big == '21.jpg'
|
|
|
|
|
|
def test_media_iptc_override(settings):
|
|
img_with_md = Image('2.jpg', 'iptcTest', settings)
|
|
assert img_with_md.title == "Markdown title beats iptc"
|
|
# Markdown parsing adds formatting. Let's just focus on content
|
|
assert "Markdown description beats iptc" in img_with_md.description
|
|
img_no_md = Image('1.jpg', 'iptcTest', settings)
|
|
assert img_no_md.title == 'Haemostratulus clouds over Canberra - ' + \
|
|
'2005-12-28 at 03-25-07'
|
|
assert img_no_md.description == \
|
|
'"Haemo" because they look like haemoglobin ' + \
|
|
'cells and "stratulus" because I can\'t work out whether ' + \
|
|
'they\'re Stratus or Cumulus clouds.\nWe\'re driving down ' + \
|
|
'the main drag in Canberra so it\'s Parliament House that ' + \
|
|
'you can see at the end of the road.'
|
|
|
|
|
|
def test_image(settings, tmpdir):
|
|
settings['destination'] = str(tmpdir)
|
|
settings['datetime_format'] = '%d/%m/%Y'
|
|
m = Image('11.jpg', 'dir1/test1', settings)
|
|
assert m.date == datetime.datetime(2006, 1, 22, 10, 32, 42)
|
|
assert m.exif['datetime'] == u'22/01/2006'
|
|
|
|
os.makedirs(join(settings['destination'], 'dir1', 'test1', 'thumbnails'))
|
|
assert m.thumbnail == join('thumbnails', '11.tn.jpg')
|
|
assert os.path.isfile(m.thumb_path)
|
|
|
|
|
|
def test_video(settings, tmpdir):
|
|
settings['destination'] = str(tmpdir)
|
|
m = Video('stallman software-freedom-day-low.ogv', 'video', settings)
|
|
file_path = join('video', 'stallman software-freedom-day-low.webm')
|
|
assert str(m) == file_path
|
|
assert m.dst_path == join(settings['destination'], file_path)
|
|
|
|
os.makedirs(join(settings['destination'], 'video', 'thumbnails'))
|
|
assert m.thumbnail == join('thumbnails',
|
|
'stallman software-freedom-day-low.tn.jpg')
|
|
assert os.path.isfile(m.thumb_path)
|
|
|
|
|
|
@pytest.mark.parametrize("path,album", REF.items())
|
|
def test_album(path, album, settings, tmpdir):
|
|
# store current locale
|
|
old_locale = locale.setlocale(locale.LC_ALL)
|
|
|
|
# locale.setlocale(locale.LC_ALL, 'fr_FR')
|
|
locale.setlocale(locale.LC_ALL, 'fr_FR.UTF-8')
|
|
|
|
try:
|
|
gal = Gallery(settings, ncpu=1)
|
|
a = Album(path, settings, album['subdirs'], album['medias'], gal)
|
|
|
|
assert a.title == album['title']
|
|
assert a.name == album['name']
|
|
assert a.subdirs == album['subdirs']
|
|
assert a.thumbnail == album['thumbnail']
|
|
if path == 'video':
|
|
assert list(a.images) == []
|
|
assert [m.filename for m in a.medias] == \
|
|
[album['medias'][0].replace('.ogv', '.webm')]
|
|
else:
|
|
assert list(a.videos) == []
|
|
assert [m.filename for m in a.medias] == album['medias']
|
|
assert len(a) == len(album['medias'])
|
|
finally:
|
|
# restore locale back
|
|
locale.setlocale(locale.LC_ALL, old_locale)
|
|
|
|
|
|
def test_albums_sort(settings):
|
|
gal = Gallery(settings, ncpu=1)
|
|
album = REF['dir1']
|
|
subdirs = list(album['subdirs'])
|
|
|
|
settings['albums_sort_reverse'] = False
|
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_subdirs('')
|
|
assert [alb.name for alb in a.albums] == subdirs
|
|
|
|
settings['albums_sort_reverse'] = True
|
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_subdirs('')
|
|
assert [alb.name for alb in a.albums] == list(reversed(subdirs))
|
|
|
|
titles = [im.title for im in a.albums]
|
|
titles.sort()
|
|
settings['albums_sort_reverse'] = False
|
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_subdirs('title')
|
|
assert [im.title for im in a.albums] == titles
|
|
|
|
settings['albums_sort_reverse'] = True
|
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_subdirs('title')
|
|
assert [im.title for im in a.albums] == list(reversed(titles))
|
|
|
|
orders = ['01', '02', '03']
|
|
orders.sort()
|
|
settings['albums_sort_reverse'] = False
|
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_subdirs('meta.order')
|
|
assert [d.meta['order'][0] for d in a.albums] == orders
|
|
|
|
settings['albums_sort_reverse'] = True
|
|
a = Album('dir1', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_subdirs('meta.order')
|
|
assert [d.meta['order'][0] for d in a.albums] == list(reversed(orders))
|
|
|
|
|
|
def test_medias_sort(settings):
|
|
gal = Gallery(settings, ncpu=1)
|
|
album = REF['dir1/test2']
|
|
|
|
settings['medias_sort_reverse'] = True
|
|
a = Album('dir1/test2', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_medias(settings['medias_sort_attr'])
|
|
assert [im.filename for im in a.images] == list(reversed(album['medias']))
|
|
|
|
settings['medias_sort_attr'] = 'date'
|
|
settings['medias_sort_reverse'] = False
|
|
a = Album('dir1/test2', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_medias(settings['medias_sort_attr'])
|
|
assert [im.filename for im in a.images] == ['22.jpg', '21.jpg',
|
|
'archlinux-kiss-1024x640.png']
|
|
|
|
settings['medias_sort_attr'] = 'meta.order'
|
|
settings['medias_sort_reverse'] = False
|
|
a = Album('dir1/test2', settings, album['subdirs'], album['medias'], gal)
|
|
a.sort_medias(settings['medias_sort_attr'])
|
|
assert [im.filename for im in a.images] == [
|
|
'archlinux-kiss-1024x640.png', '21.jpg', '22.jpg']
|
|
|
|
|
|
def test_gallery(settings, tmpdir):
|
|
"Test the Gallery class."
|
|
|
|
settings['destination'] = str(tmpdir)
|
|
settings['webm_options'] = ['-missing-option', 'foobar']
|
|
|
|
gal = Gallery(settings, ncpu=1)
|
|
gal.build()
|
|
|
|
out_html = os.path.join(settings['destination'], 'index.html')
|
|
assert os.path.isfile(out_html)
|
|
|
|
with open(out_html, 'r') as f:
|
|
html = f.read()
|
|
|
|
assert '<title>Sigal test gallery</title>' in html
|
|
|
|
logger = logging.getLogger('sigal')
|
|
logger.setLevel(logging.DEBUG)
|
|
try:
|
|
gal = Gallery(settings, ncpu=1)
|
|
with pytest.raises(SubprocessException):
|
|
gal.build()
|
|
finally:
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
def test_empty_dirs(settings):
|
|
gal = Gallery(settings, ncpu=1)
|
|
assert 'empty' not in gal.albums
|
|
assert 'dir1/empty' not in gal.albums
|
|
|
|
|
|
def test_ignores(settings, tmpdir):
|
|
tmp = str(tmpdir)
|
|
settings['destination'] = tmp
|
|
settings['ignore_directories'] = ['*test2', u'accentué']
|
|
settings['ignore_files'] = ['dir2/Hubble*', '*.png']
|
|
gal = Gallery(settings, ncpu=1)
|
|
gal.build()
|
|
|
|
assert 'test2' not in os.listdir(join(tmp, 'dir1'))
|
|
assert 'accentué' not in os.listdir(tmp)
|
|
assert 'archlinux-kiss-1024x640.png' not in os.listdir(
|
|
join(tmp, 'dir1', 'test1'))
|
|
assert 'Hubble Interacting Galaxy NGC 5257.jpg' not in os.listdir(
|
|
join(tmp, 'dir2'))
|