summaryrefslogtreecommitdiff
path: root/ia-storage-usage.py
blob: 3df0e9e68c1da9cb3a72eaedfb14a621d8df7ca4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!/usr/bin/env python3

from tqdm import tqdm
from internetarchive import get_item
from internetarchive.session import ArchiveSession
from internetarchive.search import Search

def convert_bytes(size):
    for x in ['bytes', 'KB', 'MB', 'GB', 'TB']:
        if size < 1024.0:
            return "%3.1f%s" % (size, x)
        size /= 1024.0

    return size

s = ArchiveSession()
search = Search(s, '(uploader:example@domain.com)')

total = 0
for result in tqdm(search):
    ident = result['identifier']
    size = int(get_item(ident).item_size)
    print(f"[{convert_bytes(size)}] {ident}")

    total += size

print(f"\nTotal size: {convert_bytes(total)}")