mirror of
https://github.com/ZuneDev/PyZuneImageCatalogServer.git
synced 2026-07-27 13:12:06 -07:00
90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
import json
|
|
|
|
from typing import Dict, List
|
|
from flask import Flask, request, Response, abort, send_file, redirect
|
|
from io import BytesIO
|
|
import requests
|
|
from requests import RequestException
|
|
|
|
import musicbrainzngs
|
|
import api.discogs as discogs
|
|
from secrets import USER_AGENT
|
|
|
|
from locale import *
|
|
|
|
|
|
locale = getdefaultlocale()[0]
|
|
|
|
|
|
app = Flask(__name__)
|
|
caa_supported_sizes = [250, 500, 1200]
|
|
dc_artist_cache: Dict[str, dict] = {}
|
|
|
|
musicbrainzngs.set_useragent("Zune", "4.8", "https://github.com/yoshiask/PyZuneImageCatalogServer")
|
|
DEFAULT_HEADERS: Dict[str, str] = {
|
|
"User-Agent": USER_AGENT
|
|
}
|
|
|
|
|
|
import re
|
|
@app.after_request
|
|
def allow_zunestk_cors(response):
|
|
r = request.origin
|
|
if r is None:
|
|
return response
|
|
if re.match(r"https?://(127\.0\.0\.(?:\d*)|localhost(?:\:\d+)?|(?:\w*\.)*zunes\.tk)", r):
|
|
response.headers.add('Access-Control-Allow-Origin', r)
|
|
response.headers.add('Access-Control-Allow-Credentials', 'true')
|
|
response.headers.add('Access-Control-Allow-Headers', 'Content-Type')
|
|
response.headers.add('Access-Control-Allow-Headers', 'Cache-Control')
|
|
response.headers.add('Access-Control-Allow-Headers', 'X-Requested-With')
|
|
response.headers.add('Access-Control-Allow-Headers', 'Authorization')
|
|
response.headers.add('Access-Control-Allow-Methods', 'GET, POST, OPTIONS, PUT, DELETE')
|
|
return response
|
|
|
|
|
|
@app.route("/")
|
|
def default():
|
|
return Response("Welcome to the Social", mimetype="text/plain")
|
|
|
|
|
|
@app.route(f"/v3.2/<string:locale>/image/<string:mbid>")
|
|
def get_image(mbid: str, locale: str):
|
|
image_url: str = ""
|
|
if mbid.endswith('0' * 12):
|
|
dcid: int = int(mbid[:8], 16)
|
|
img_idx: int = int(mbid[9:13], 16)
|
|
|
|
# Get or update cached artist
|
|
dc_artist: dict = dc_artist_cache.get(dcid)
|
|
if dc_artist is None:
|
|
# Artist not in cache
|
|
dc_artist = discogs.get_artist_from_dcid(dcid)
|
|
dc_artist_cache[dcid] = dc_artist
|
|
|
|
# Get URL for requested image
|
|
image_url = dc_artist["images"][img_idx]["uri"]
|
|
else:
|
|
# The Cover Art Archive API supports sizes of 250, 500, and 1200
|
|
requested_width = request.args.get("width", default=500, type=int)
|
|
width = min(caa_supported_sizes, key=lambda x: abs(x - requested_width))
|
|
image_url = f"https://coverartarchive.org/release/{mbid}/front-{width}"
|
|
|
|
# Request the image from the API and forward it to the Zune software
|
|
return redirect(image_url, 302)
|
|
|
|
|
|
@app.route(f"/v3.2/<string:locale>/music/artist/<string:mbid>/<string:type>")
|
|
def get_artist_image(mbid: str, type: str, locale: str):
|
|
dc_artist, mb_artist = discogs.get_artist_from_mbid(mbid)
|
|
|
|
# Get URL for requested image
|
|
image_url: str = dc_artist["images"][0]["uri"]
|
|
|
|
# Request the image from the API and forward it to the Zune software
|
|
return redirect(image_url, 302)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="127.0.0.4", port=80)
|