hunt more dynamic

Client/hunt.py

@@ -16,6 +16,8 @@ img_base = 'https://i.pximg.net/img-original/img/'
 os.chdir(os.path.dirname(os.path.abspath(__file__)))
 os.makedirs("images/Stash", exist_ok=True)
 
+images_cache = os.listdir("images/Stash")
+
 db = lmdb.open("db", subdir=True, map_size=524288)
 valid = [f for f in os.listdir() if f.endswith(".txt")]
 
@@ -60,6 +62,7 @@ for index in indexs:
                 txn.put(post_id.encode(), b'\0')
 
     to_download = {post_id: url for post_id, url in post_ids.items() if url not in blacklist and f"{post_id}.png" not in images_cache}
+    post_ids.update(to_download)
     print(f"Downloading {len(to_download)} images...")
     if to_download:
         images_zstd = requests.post(f'{endpoint}/download', json={'posts': to_download}, stream=True)
@@ -79,7 +82,7 @@ for index in indexs:
     for i, post_id in enumerate(post_ids.keys()):
         if f"{post_id}.png" in images_cache:
             if not os.path.exists(f"images/{group_name}/{i}_{post_id}.png"):
-                os.link(f"images/Stash/{post_id}.png", f"images/{group_name}/{i}_{post_id}.png")
+                os.link(f"images/Stash/{post_id}.png", f"images/{group_name}/{i}_{post_id}_{post_ids[post_id].split('p')[1].split('.')[0]}.png")
 
     if len(os.listdir(f'images/{group_name}')) == 0:
         os.rmdir(f"images/{group_name}")