def test_wrong_extension(square_png_image, square_jpg_image, tmp_path, fmt):
src, dst = get_src_dst(tmp_path,
fmt,
png_image=square_png_image,
jpg_image=square_jpg_image)
with pytest.raises(ValueError):
create_favicon(src, dst)
def test_create_favicon(png_image, jpg_image, tmp_path, fmt, exp_size):
src, dst = get_src_dst(tmp_path,
fmt,
png_image=png_image,
jpg_image=jpg_image)
dst = dst.parent.joinpath("favicon.ico")
create_favicon(src, dst)
im = Image.open(dst)
assert im.format == "ICO"
assert im.size == (exp_size, exp_size)
def test_create_favicon_square(square_png_image, square_jpg_image, tmp_path,
fmt):
src, dst = get_src_dst(tmp_path,
fmt,
png_image=square_png_image,
jpg_image=square_jpg_image)
dst = dst.parent.joinpath("favicon.ico")
create_favicon(src, dst)
im = Image.open(dst)
assert im.format == "ICO"
assert im.size == (256, 256)
def add_favicon(self):
favicon_orig = self.build_dir / "favicon"
# if user provided a custom favicon, retrieve that
if self.favicon:
handle_user_provided_file(source=self.favicon, dest=favicon_orig)
# otherwise, get thumbnail from database
else:
# add channel thumbnail as favicon
try:
favicon_prefix, favicon_data = self.db.get_channel_metadata(
self.channel_id)["thumbnail"].split(";base64,", 1)
favicon_data = base64.standard_b64decode(favicon_data)
# favicon_mime = favicon_prefix.replace("data:", "")
with open(favicon_orig, "wb") as fh:
fh.write(favicon_data)
del favicon_data
except Exception as exc:
logger.warning(
"Unable to extract favicon from DB; using default")
logger.exception(exc)
# use a default favicon
handle_user_provided_file(source=self.templates_dir /
"kolibri-logo.png",
dest=favicon_orig)
# convert to PNG (might already be PNG but it's OK)
favicon_fpath = favicon_orig.with_suffix(".png")
convert_image(favicon_orig, favicon_fpath)
# resize to appropriate size (ZIM uses 48x48 so we double for retina)
for size in (96, 48):
resize_image(favicon_fpath,
width=size,
height=size,
method="thumbnail")
with open(favicon_fpath, "rb") as fh:
self.creator.add_illustration(size, fh.read())
# resize to appropriate size (ZIM uses 48x48)
resize_image(favicon_fpath, width=96, height=96, method="thumbnail")
# generate favicon
favicon_ico_path = favicon_fpath.with_suffix(".ico")
create_favicon(src=favicon_fpath, dst=favicon_ico_path)
self.creator.add_item_for("favicon.png", fpath=favicon_fpath)
self.creator.add_item_for("favicon.ico", fpath=favicon_ico_path)