随机动漫壁纸 API

早就看首页单调的背景不爽了，这次趁着迁移网站的机会顺便造了一个随机获取和显示动漫壁纸的轮子。

实现思路：从 /r/Animewallpaper 上爬墙带 Desktop flair 的帖子。来源

实现细节：

从 https://www.reddit.com/r/Animewallpaper/search.rss?q=flair_name%3A%22Desktop%22&restrict_sr=1 获取 RSS feed，解析获取前 25 条结果。该结果缓存 10 分钟。

def entry_to_dictionary(entry):
    return {
        "source": entry.find("{http://www.w3.org/2005/Atom}link").attrib["href"],
        "title":entry.find("{http://www.w3.org/2005/Atom}title").text,
        "img": RE_EXTRACT_IMG.search(entry.find("{http://www.w3.org/2005/Atom}content").text).groups(0)[0] 
    }

def fetch_feed():
    global last_fetch_time
    global cached_results
    
    if time.time() - last_fetch_time <= CACHE_VALID_TIME and cached_results:
        return cached_results
    
    results = []
    e = xml.etree.ElementTree.fromstring(requests.get(FEED_URL, headers={"User-Agent":USER_AGENT}).text.strip())
    for entry in e.findall("{http://www.w3.org/2005/Atom}entry"):
        try:
            results.append(entry_to_dictionary(entry))
        except:
            pass
            
    cached_results = results
    last_fetch_time = time.time()
    
    return results

前端通过ajax从/random_anime_wallpaper获取一份随机壁纸。

为了优化国内用户的加载速度，考虑在服务端获取壁纸后直接发给用户（相当于一个反代）。

一开始打算用 base64 编码壁纸然后丢在 json 里返回。然而这么实现的话客户端要下载完整张图片才能渲染显示。除此以外，由于使用了 gunicorn，每个 worker 的缓存都是独立的，也非常浪费内存。
故打算在服务端下载到硬盘，并直接返回静态路径，这样只有在第一次调用 API 的时候会稍有延迟。实现如下：

def download_img(src):
 ext = src.split(".")[-1]
 name = "static/" + hashlib.sha1(src.encode()).hexdigest() + "." + ext
 if os.path.isfile(name):
     return name
 print("Downloading ", src, " to ", name)
 with open(name, 'wb') as f:
     f.write(requests.get(src).content)
 return name

@app.route("/random_anime_wallpaper")
 def random_anime_wallpaper():
     res = fetch_feed()
     selected = copy.deepcopy(res[random.randint(0, len(res))])
     if request.args.get('download', False):
         selected["img_src"] = selected["img"]
         selected["img"] = URL_BASE_PATH + download_img(selected["img"])
     return jsonify(selected)

最后直接使用 nginx 反代请求到 python 上就可以了：

 location /api/wallpapers/ {
     proxy_pass http://127.0.0.1:8001/;
 }

从思路到实现都非常简单~

Demo | API | Source