Riot API: 画像のバッチダウンロード

この記事はRiot APIのAdvent Calendar 2014の８日目の記事です。

前回はプロフィールアイコンのurlをCDNから取得するところまでデモしましたが、最新の画像をバッチで一気にダウンロード、使いやすくなるようにファイルをローカルで保存するのをやってみます。

１）チャンピオンのリストを取得

バージョン取得などは前回とほぼ同じので、説明を飛ばします：

#!/usr/bin/env python

import os
import requests
import shutil

DD_URL = 'http://ddragon.leagueoflegends.com'

res = requests.get(DD_URL + '/realms/na.json')
res.raise_for_status()

version = res.json()['n']['champion']

champion_info_url= DD_URL + '/cdn/' + version + '/data/en_US/champion.json'
res = requests.get(champion_info_url)
res.raise_for_status()

data = res.json()['data']
 

ブラウザでチャンピオンjsonデーターを確認するとわかり安いと思いますが、全チャンピオンの名前や基本パラメーターが入っているhash tableを'data'に取得できました。

２）保存先のディレクトリーを用意する

今後ダウンロードをする画像を保存先ディレクトリがあるかどうか確認して、なければつくりましょう。

output_dir = os.path.dirname(os.path.realpath(__file__)) + '/champion'

try:
    os.stat(output_dir)
except:
    os.mkdir(output_dir)
 

３）キャラクターをループで回して、画像を保存しておく

DataDragonではチャンピオンの名前をファイル名として使われています：
http://ddragon.leagueoflegends.com/cdn/4.20.1/img/champion/Aatrox.png

チャンピオンの名前わかれば取得しやすいでしょうけど、たとえば、Riot APIで最近のゲーム履歴（/game/by-summoner/SUMMONER_ID/recent）取得した場合、利用したチャンピオンのkey（ID）しか返ってこないので、名前だけだとかなり不便ですね。
そのため、画像をダウンロードしたら、ファイル名をチャンピオンの名前じゃなくて、チャンピオンのkeyをつけましょう。

champion_img_base = DD_URL + "/cdn/" + version + "/img/champion/"

for name in data:
    img_url = champion_img_base + name + ".png"
    filename = data[name]['key'] + ".png"

    res = requests.get(img_url, stream=True)
    if res.status_code == 200:
        with open(output_dir + '/' + filename, 'wb') as f:
            res.raw.decode_content = True
            shutil.copyfileobj(res.raw, f)
 

以上！
ダウンロードできたかどうかを確認すると：

ls champion/
1.png 106.png 114.png 122.png 14.png 161.png 21.png 25.png 28.png 34.png 40.png 45.png 55.png 61.png 7.png 79.png 85.png 96.png
10.png 107.png 115.png 126.png 143.png 17.png 22.png 254.png 29.png 35.png 41.png 48.png 56.png 62.png 72.png 8.png 86.png 98.png
101.png 11.png 117.png 127.png 15.png 18.png 222.png 26.png 3.png 36.png 412.png 5.png 57.png 63.png 74.png 80.png 89.png 99.png
102.png 110.png 119.png 13.png 150.png 19.png 23.png 266.png 30.png 37.png 42.png 50.png 58.png 64.png 75.png 81.png 9.png
103.png 111.png 12.png 131.png 154.png 2.png 236.png 267.png 31.png 38.png 429.png 51.png 59.png 67.png 76.png 82.png 90.png
104.png 112.png 120.png 133.png 157.png 20.png 238.png 268.png 32.png 39.png 43.png 53.png 6.png 68.png 77.png 83.png 91.png
105.png 113.png 121.png 134.png 16.png 201.png 24.png 27.png 33.png 4.png 44.png 54.png 60.png 69.png 78.png 84.png 92.png
  

全部揃ってありますね！これで新しいチャンピオンが追加された場合でもスクリプトを動かすだけですぐに画像をダウンロードできそうですね。