github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[sankaku] improve tag categorization code

Browse Source 
translate tag type ID to name for each category
instead of for each tag
This commit is contained in:
Mike Fährmann
2024-11-03 09:21:39 +01:00
parent 5ba1280220
commit d5fa1d6aba
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
gallery_dl/extractor/sankaku.py
View File

@@ -76,14 +76,15 @@ class SankakuExtractor(BooruExtractor):
def _tags(self, post, page): def _tags(self, post, page):
tags = collections.defaultdict(list) tags = collections.defaultdict(list)
types = self.TAG_TYPES
for tag in post["tags"]: for tag in post["tags"]:
name = tag["name"] name = tag["name"]
if name: if name:
tags[types[tag["type"]]].append(name.lower().replace(" ", "_")) tags[tag["type"]].append(name.lower().replace(" ", "_"))
for key, value in tags.items(): types = self.TAG_TYPES
post["tags_" + key] = value for type, values in tags.items():
post["tag_string_" + key] = " ".join(value) name = types[type]
post["tags_" + name] = values
post["tag_string_" + name] = " ".join(values)
def _notes(self, post, page): def _notes(self, post, page):
if post.get("has_notes"): if post.get("has_notes"):