78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
# save_cover.py
|
|
#
|
|
# Copyright 2022-2023 kramo
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
|
|
from pathlib import Path
|
|
from shutil import copyfile
|
|
|
|
from gi.repository import GdkPixbuf, Gio
|
|
from PIL import Image, ImageSequence
|
|
|
|
|
|
def resize_animation(cover_path):
|
|
image = Image.open(cover_path)
|
|
frames = tuple(
|
|
frame.copy().resize((200, 300)) for frame in ImageSequence.Iterator(image)
|
|
)
|
|
|
|
tmp_path = Path(Gio.File.new_tmp("XXXXXX.gif")[0].get_path())
|
|
frames[0].save(
|
|
tmp_path,
|
|
format="gif",
|
|
save_all=True,
|
|
append_images=frames[1:],
|
|
)
|
|
|
|
return tmp_path
|
|
|
|
|
|
def save_cover(
|
|
win,
|
|
game_id,
|
|
cover_path=None,
|
|
pixbuf=None,
|
|
animation_path=None,
|
|
):
|
|
win.covers_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
# Remove previous covers
|
|
(win.covers_dir / f"{game_id}.tiff").unlink(missing_ok=True)
|
|
(win.covers_dir / f"{game_id}.gif").unlink(missing_ok=True)
|
|
|
|
if animation_path:
|
|
copyfile(animation_path, win.covers_dir / f"{game_id}.gif")
|
|
win.game_covers[game_id].new_pixbuf(animation_path)
|
|
return
|
|
|
|
if not pixbuf:
|
|
if not cover_path:
|
|
return
|
|
pixbuf = GdkPixbuf.Pixbuf.new_from_file_at_scale(
|
|
str(cover_path), *win.image_size, False
|
|
)
|
|
|
|
pixbuf.savev(
|
|
str(win.covers_dir / f"{game_id}.tiff"),
|
|
"tiff",
|
|
["compression"],
|
|
["8"] if win.schema.get_boolean("high-quality-images") else ["7"],
|
|
)
|
|
|
|
win.game_covers[game_id].new_pixbuf(win.covers_dir / f"{game_id}.tiff")
|