tqdm

Client/hunt.py

@@ -2,6 +2,7 @@ import os
 import lmdb
 import requests
 from concurrent.futures import ThreadPoolExecutor
+from tqdm import tqdm
 
 local = 0
 if local:
@@ -54,7 +55,7 @@ for index in indexs:
         post_ids_dict = {post_id: txn.get(post_id.encode()) for post_id in post_ids}
         post_ids_dict = {post_id: url.decode() if url != None else None for post_id, url in post_ids_dict.items()}
 
-        filtered = [post_id for post_id, url in post_ids_dict.items() if post_ids_dict[post_id] != '' and f"{post_id}.png" not in images_cache]
+        filtered = [post_id for post_id, url in post_ids_dict.items() if url == None and f"{post_id}.png" not in images_cache]
         print(f"Group: {group_name}\nFiltered: {len(filtered)}/{len(post_ids_dict)}")
         if filtered:
             data = requests.post(f'{endpoint}/pixif', json={"post_ids": filtered}).json()
@@ -78,7 +79,8 @@ for index in indexs:
     if to_download_items:
         print("Starting download...")
         with ThreadPoolExecutor(max_workers=max_workers) as executor:
-            results = list(executor.map(download_image, to_download_items))
+            for result in tqdm(executor.map(download_image, to_download_items), total=len(to_download_items), desc="Downloading"):
+                pass
 
     images_cache = os.listdir("images/Stash")