mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-06 19:25:34 +00:00
1e4484b1c8
+ use getters instead of constants in settings classes + refactor previous references + move get_xdg_config_dir() from settings.py to app.utils.xdg_utils.py
111 lines
2.9 KiB
Python
111 lines
2.9 KiB
Python
"""
|
|
This library contains all the functions related to playlists.
|
|
"""
|
|
import os
|
|
import random
|
|
import string
|
|
from typing import Any
|
|
|
|
from PIL import Image, ImageSequence
|
|
|
|
from app import settings
|
|
|
|
|
|
def create_thumbnail(image: Any, img_path: str) -> str:
|
|
"""
|
|
Creates a 250 x 250 thumbnail from a playlist image
|
|
"""
|
|
thumb_path = "thumb_" + img_path
|
|
full_thumb_path = os.path.join(settings.Paths.get_app_dir(), "images", "playlists", thumb_path)
|
|
|
|
aspect_ratio = image.width / image.height
|
|
|
|
new_w = round(250 * aspect_ratio)
|
|
|
|
thumb = image.resize((new_w, 250), Image.ANTIALIAS)
|
|
thumb.save(full_thumb_path, "webp")
|
|
|
|
return thumb_path
|
|
|
|
|
|
def create_gif_thumbnail(image: Any, img_path: str):
|
|
"""
|
|
Creates a 250 x 250 thumbnail from a playlist image
|
|
"""
|
|
thumb_path = "thumb_" + img_path
|
|
full_thumb_path = os.path.join(settings.Paths.get_app_dir(), "images", "playlists", thumb_path)
|
|
|
|
frames = []
|
|
|
|
for frame in ImageSequence.Iterator(image):
|
|
aspect_ratio = frame.width / frame.height
|
|
|
|
new_w = round(250 * aspect_ratio)
|
|
|
|
thumb = frame.resize((new_w, 250), Image.ANTIALIAS)
|
|
frames.append(thumb)
|
|
|
|
frames[0].save(full_thumb_path, save_all=True, append_images=frames[1:])
|
|
|
|
return thumb_path
|
|
|
|
|
|
def save_p_image(file, pid: str):
|
|
"""
|
|
Saves a playlist banner image and returns the filepath.
|
|
"""
|
|
img = Image.open(file)
|
|
|
|
random_str = "".join(random.choices(string.ascii_letters + string.digits, k=5))
|
|
|
|
filename = pid + str(random_str) + ".webp"
|
|
|
|
full_img_path = os.path.join(settings.Paths.get_playlist_img_path(), filename)
|
|
|
|
if file.content_type == "image/gif":
|
|
frames = []
|
|
|
|
for frame in ImageSequence.Iterator(img):
|
|
frames.append(frame.copy())
|
|
|
|
frames[0].save(full_img_path, save_all=True, append_images=frames[1:])
|
|
create_gif_thumbnail(img, img_path=filename)
|
|
|
|
return filename
|
|
|
|
img.save(full_img_path, "webp")
|
|
create_thumbnail(img, img_path=filename)
|
|
|
|
return filename
|
|
|
|
#
|
|
# class ValidatePlaylistThumbs:
|
|
# """
|
|
# Removes all unused images in the images/playlists folder.
|
|
# """
|
|
#
|
|
# def __init__(self) -> None:
|
|
# images = []
|
|
# playlists = Get.get_all_playlists()
|
|
#
|
|
# log.info("Validating playlist thumbnails")
|
|
# for playlist in playlists:
|
|
# if playlist.image:
|
|
# img_path = playlist.image.split("/")[-1]
|
|
# thumb_path = playlist.thumb.split("/")[-1]
|
|
#
|
|
# images.append(img_path)
|
|
# images.append(thumb_path)
|
|
#
|
|
# p_path = os.path.join(settings.APP_DIR, "images", "playlists")
|
|
#
|
|
# for image in os.listdir(p_path):
|
|
# if image not in images:
|
|
# os.remove(os.path.join(p_path, image))
|
|
#
|
|
# log.info("Validating playlist thumbnails ... ✅")
|
|
#
|
|
|
|
|
|
# TODO: Fix ValidatePlaylistThumbs
|