Skip to content

Commit b750aec

Browse files
committed
tagImagesWithPerfTags bulk processing loggs progress now
1 parent 83a0f0a commit b750aec

File tree

1 file changed

+18
-15
lines changed

1 file changed

+18
-15
lines changed

plugins/tagImagesWithPerfTags/tagImagesWithPerfTags.py

Lines changed: 18 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,16 @@ def processAll():
2222
performersTotal = stash.find_performers(f=query, filter={"page": 0, "per_page": 0}, get_count=True)[0]
2323
i = 0
2424
while i < performersTotal:
25+
log.progress((i / performersTotal))
26+
2527
perf = stash.find_performers(f=query, filter={"page": i, "per_page": 1})
2628

2729
performer_tags_ids = []
2830
performer_tags_names = []
2931
for performer_tag in perf[0]["tags"]:
3032
performer_tags_ids.append(performer_tag["id"])
3133
performer_tags_names.append(performer_tag["name"])
32-
34+
3335
image_query = {
3436
"performers": {
3537
"value": [perf[0]["id"]],
@@ -46,22 +48,23 @@ def processAll():
4648

4749
performer_image_count = stash.find_images(f=image_query, filter={"page": 0, "per_page": 0}, get_count=True)[0]
4850

49-
log.info(f"updating {performer_image_count} images of performer \"{ perf[0]['name']}\" with tags {performer_tags_names}")
51+
if performer_image_count > 0:
52+
log.info(f"updating {performer_image_count} images of performer \"{ perf[0]['name']}\" with tags {performer_tags_names}")
5053

51-
performer_image_page_size = 100
52-
performer_image_page = 0
53-
while performer_image_page * performer_image_page_size < performer_image_count:
54-
performer_images = stash.find_images(f=image_query, filter={"page": performer_image_page, "per_page": performer_image_page_size}, fragment='id')
55-
performer_image_ids = [performer_image['id'] for performer_image in performer_images]
54+
performer_image_page_size = 100
55+
performer_image_page = 0
56+
while performer_image_page * performer_image_page_size < performer_image_count:
57+
performer_images = stash.find_images(f=image_query, filter={"page": performer_image_page, "per_page": performer_image_page_size}, fragment='id')
58+
performer_image_ids = [performer_image['id'] for performer_image in performer_images]
59+
60+
stash.update_images(
61+
{
62+
"ids": performer_image_ids,
63+
"tag_ids": {"mode": "ADD", "ids": performer_tags_ids},
64+
}
65+
)
66+
performer_image_page += 1
5667

57-
stash.update_images(
58-
{
59-
"ids": performer_image_ids,
60-
"tag_ids": {"mode": "ADD", "ids": performer_tags_ids},
61-
}
62-
)
63-
performer_image_page += 1
64-
6568
i = i + 1
6669

6770

0 commit comments

Comments
 (0)