swingmusic/app/api/imgserver.py
2023-07-06 13:52:32 +03:00

89 lines
1.9 KiB
Python

from pathlib import Path
from flask import Blueprint, send_from_directory
from app.settings import Paths
api = Blueprint("imgserver", __name__, url_prefix="/img")
@api.route("/")
def hello():
return "<h1>Image Server</h1>"
def send_fallback_img(filename: str = "default.webp"):
path = Paths.get_assets_path()
img = Path(path) / filename
if not img.exists():
return "", 404
return send_from_directory(path, filename)
@api.route("/t/o/<imgpath>")
def send_original_thumbnail(imgpath: str):
path = Paths.get_original_thumb_path()
fpath = Path(path) / imgpath
if fpath.exists():
return send_from_directory(path, imgpath)
return send_fallback_img()
@api.route("/t/<imgpath>")
def send_lg_thumbnail(imgpath: str):
path = Paths.get_lg_thumb_path()
fpath = Path(path) / imgpath
if fpath.exists():
return send_from_directory(path, imgpath)
return send_fallback_img()
@api.route("/t/s/<imgpath>")
def send_sm_thumbnail(imgpath: str):
path = Paths.get_sm_thumb_path()
fpath = Path(path) / imgpath
if fpath.exists():
return send_from_directory(path, imgpath)
return send_fallback_img()
@api.route("/a/<imgpath>")
def send_lg_artist_image(imgpath: str):
path = Paths.get_artist_img_lg_path()
fpath = Path(path) / imgpath
if fpath.exists():
return send_from_directory(path, imgpath)
return send_fallback_img("artist.webp")
@api.route("/a/s/<imgpath>")
def send_sm_artist_image(imgpath: str):
path = Paths.get_artist_img_sm_path()
fpath = Path(path) / imgpath
if fpath.exists():
return send_from_directory(path, imgpath)
return send_fallback_img("artist.webp")
@api.route("/p/<imgpath>")
def send_playlist_image(imgpath: str):
path = Paths.get_playlist_img_path()
fpath = Path(path) / imgpath
if fpath.exists():
return send_from_directory(path, imgpath)
return send_fallback_img("playlist.svg")