status codeで分岐処理(urllib)

Question

下記の画像取得を行う処理で、
status code = 400のときは何もせず終了
それ以外はretry処理をする
といった実装を組み込みたいのですが、書き方がわからずご教授いただけないでしょうか。

from retry import retry

@retry(tries=30, delay=10)
def download_file(url, dst_path):
    with urllib.request.urlopen(url) as web_file, open(dst_path, 'wb') as local_file:
        local_file.write(web_file.read())

上記処理で、status code = 400のときは何もせず終了を組み込みたいです。

Answer 1

公式ドキュメントにサンプルコードがありました。
下記のようにエラーをハンドリングし、ステータスコードで分岐してリトライの関数をよべば良いかと思います。

from urllib.request import Request, urlopen
from urllib.error import URLError
req = Request(someurl)
try:
    response = urlopen(req)
except URLError as e:
    if hasattr(e, 'reason'):
        print('We failed to reach a server.')
        print('Reason: ', e.reason)
    elif hasattr(e, 'code'):
        print('The server couldn\'t fulfill the request.')
        print('Error code: ', e.code)
else:
    # everything is fine

参考: https://docs.python.org/ja/3/howto/urllib2.html