Pythonでダウンロードなしでファイルサイズを調べる方法【requests】

最近Pythonに手を出し始めました。

今回気になったのは次のこと

URLからファイルサイズを調べたい
ただしダウンロード無しで

通信にはrequestsを使っていました。

それを使うこと前提の方法です

このページの目次

ヘッダー取得してrequestsからDLなしでサイズ取得
分かりやすく関数化しておく
MB・GBなどの単位付きでファイルサイズ取得してみる
URLからファイルサイズ取得は意外と簡単

ヘッダー取得してrequestsからDLなしでサイズ取得

まずURLからヘッダーだけ取得します。

▼ requests.head().headersについて

PUT、DELETE、HEAD、OPTIONSなどの他のHTTPリクエストについても本当にとてもシンプルです。:

>>> r = requests.put("http://httpbin.org/put")
>>> r = requests.delete("http://httpbin.org/delete")
>>> r = requests.head("http://httpbin.org/get")
>>> r = requests.options("http://httpbin.org/get")

Pythonの辞書形式で簡単にサーバーのレスポンスヘッダーを見ることができます。:

> r.headers
{
'status': '200 OK',
'content-encoding': 'gzip',
'transfer-encoding': 'chunked',
'connection': 'close',
'server': 'nginx/1.0.4',
'x-runtime': '148ms',
'etag': '"e1ca502697e5c9317743dc078f67693f"',
'content-type': 'application/json; charset=utf-8'
}
辞書とはいえ、特別です。辞書はHTTPヘッダーを作成するためだけに作られます。 RFC 2616 によると、HTTPヘッダーは大文字と小文字を区別しません。

引用元 : https://requests-docs-ja.readthedocs.io/en/latest/user/quickstart/

ヘッダーだけ取得するのに requests.head() が用意されており、そこから headers を参照すると content-length というキーも含まれてます。そこからファイルサイズ取得可能です

具体的には次のようなコードですね

▼ URLからファイルサイズ取得

url = 'https://hoge.com/video.mp4'
response = requests.head(url, allow_redirects=True)
size = response.headers.get('content-length', -1)

print('size : '+str(size)+' bytes')
    ## => size : 182645 bytes

url = 'https://hoge.com/video.mp4'

response = requests.head(url, allow_redirects=True)

size = response.headers.get('content-length', -1)

print('size : '+str(size)+' bytes')

## => size : 182645 bytes

ハイ、このように取得できました。

分かりやすく関数化しておく

こういう風に関数化しておきます

def len_from_url(url):
    url = 'https://hoge.com/video.mp4'
    r = requests.head(url, allow_redirects=True)
    if(r.status_code == 200):
        return r.headers.get('content-length', -1)
    else:
        return -1

def len_from_url(url):

url = 'https://hoge.com/video.mp4'

r = requests.head(url, allow_redirects=True)

if(r.status_code == 200):

return r.headers.get('content-length', -1)

else:

return -1

さっきは書かなかったけど r.status_code でステータスコードを調べ、正常ならファイルサイズ取得・それ以外では一応 -1 を返す仕様にしました

MB・GBなどの単位付きでファイルサイズ取得してみる

ここからは蛇足ですが…

ファイルサイズを単位付きで表示してみます

▼ こういう関数を用意する

def convert_size(size_bytes):
   size_bytes = float(size_bytes)
   if size_bytes == 0:
       return "0B"
   size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
   i = int(math.floor(math.log(size_bytes, 1024)))
   p = math.pow(1024, i)
   s = round(size_bytes / p, 2)
   return "%s %s" % (s, size_name[i])

def convert_size(size_bytes):

size_bytes = float(size_bytes)

if size_bytes == 0:

return "0B"

size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")

i = int(math.floor(math.log(size_bytes, 1024)))

p = math.pow(1024, i)

s = round(size_bytes / p, 2)

return "%s %s" % (s, size_name[i])

▼ 単位付きでファイルサイズ表示

url = 'https://hoge.com/video.mp4'
r = requests.head(url, allow_redirects=True)
if(r.status_code == 200):
    size = convert_size(
        r.headers.get('content-length', -1)
    )
else:
    size = ''
    
print('size : '+size)
    /// => size : 18.3MB