ia-storage-usage.py

1 files changed, 27 insertions, 0 deletions

diff --git a/ia-storage-usage.py b/ia-storage-usage.py
new file mode 100755
index 0000000..3df0e9e
--- /dev/null
+++ b/ia-storage-usage.py
@@ -0,0 +1,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)}")