mirror of
https://github.com/Dvorinka/swingmusic-extended.git
synced 2026-06-04 04:23:01 +00:00
mungai-njoroge
123 lines
2.9 KiB
Python
123 lines
2.9 KiB
Python
from pathlib import Path
|
|
from flask_openapi3 import Tag
|
|
from flask_openapi3 import APIBlueprint
|
|
from pydantic import BaseModel, Field
|
|
from flask import send_from_directory
|
|
|
|
from app.settings import Defaults, Paths
|
|
|
|
bp_tag = Tag(
|
|
name="Images", description="Image filenames are constructured as '{itemhash}.webp'"
|
|
)
|
|
api = APIBlueprint("imgserver", __name__, url_prefix="/img", abp_tags=[bp_tag])
|
|
|
|
|
|
def send_fallback_img(filename: str = "default.webp"):
|
|
folder = Paths.get_assets_path()
|
|
img = Path(folder) / filename
|
|
|
|
if not img.exists():
|
|
return "", 404
|
|
|
|
return send_from_directory(folder, filename)
|
|
|
|
|
|
class ImagePath(BaseModel):
|
|
imgpath: str = Field(
|
|
description="The image filename",
|
|
example=Defaults.API_ALBUMHASH + ".webp",
|
|
)
|
|
|
|
|
|
# @api.get("/t/o/<imgpath>")
|
|
# def send_original_thumbnail(path: ImagePath):
|
|
# """
|
|
# Get original thumbnail
|
|
# """
|
|
# folder = Paths.get_original_thumb_path()
|
|
# fpath = Path(folder) / path.imgpath
|
|
|
|
# if fpath.exists():
|
|
# return send_from_directory(folder, path.imgpath)
|
|
|
|
# return send_fallback_img()
|
|
|
|
|
|
@api.get("/t/<imgpath>")
|
|
def send_lg_thumbnail(path: ImagePath):
|
|
"""
|
|
Get large thumbnail (500 x 500)
|
|
"""
|
|
folder = Paths.get_lg_thumb_path()
|
|
fpath = Path(folder) / path.imgpath
|
|
|
|
if fpath.exists():
|
|
return send_from_directory(folder, path.imgpath)
|
|
|
|
return send_fallback_img()
|
|
|
|
|
|
@api.get("/t/s/<imgpath>")
|
|
def send_sm_thumbnail(path: ImagePath):
|
|
"""
|
|
Get small thumbnail (64 x 64)
|
|
"""
|
|
folder = Paths.get_sm_thumb_path()
|
|
fpath = Path(folder) / path.imgpath
|
|
|
|
if fpath.exists():
|
|
return send_from_directory(folder, path.imgpath)
|
|
|
|
return send_fallback_img()
|
|
|
|
|
|
@api.get("/a/<imgpath>")
|
|
def send_lg_artist_image(path: ImagePath):
|
|
"""
|
|
Get large artist image (500 x 500)
|
|
"""
|
|
folder = Paths.get_artist_img_lg_path()
|
|
fpath = Path(folder) / path.imgpath
|
|
|
|
if fpath.exists():
|
|
return send_from_directory(folder, path.imgpath)
|
|
|
|
return send_fallback_img("artist.webp")
|
|
|
|
|
|
@api.get("/a/s/<imgpath>")
|
|
def send_sm_artist_image(path: ImagePath):
|
|
"""
|
|
Get small artist image (64 x 64)
|
|
"""
|
|
folder = Paths.get_artist_img_sm_path()
|
|
fpath = Path(folder) / path.imgpath
|
|
|
|
if fpath.exists():
|
|
return send_from_directory(folder, path.imgpath)
|
|
|
|
return send_fallback_img("artist.webp")
|
|
|
|
|
|
class PlaylistImagePath(BaseModel):
|
|
imgpath: str = Field(
|
|
description="The image path",
|
|
example="1.webp",
|
|
)
|
|
|
|
|
|
@api.get("/p/<imgpath>")
|
|
def send_playlist_image(path: PlaylistImagePath):
|
|
"""
|
|
Get playlist image
|
|
|
|
Images are constructed as '{playlist_id}.webp'
|
|
"""
|
|
folder = Paths.get_playlist_img_path()
|
|
fpath = Path(folder) / path.imgpath
|
|
|
|
if fpath.exists():
|
|
return send_from_directory(folder, path.imgpath)
|
|
|
|
return send_fallback_img("playlist.svg")
|