diff --git a/category/deviantart.py b/category/deviantart.py new file mode 100644 index 0000000..9aac679 --- /dev/null +++ b/category/deviantart.py @@ -0,0 +1,17 @@ +def makeKeyWordList(tags): + catPaths = { + 'digitalart/paintings/other': ['Digital Art', 'Paintings'] + } + output = [ + tags.da_category, + tags.license, + tags.author.username + ] + + if tags.is_mature: + output.append('is_mature') + + + for item in .get(tags.category_path,[]): + output.append(item) + \ No newline at end of file diff --git a/category/furaffinity.py b/category/furaffinity.py new file mode 100644 index 0000000..b01bf98 --- /dev/null +++ b/category/furaffinity.py @@ -0,0 +1,9 @@ +def makeKeyWordList(tags): + return [ + tags.theme, + tags.rating, + tags.species, + tags.gender, + tags.artist_url, + tags.fa_category + ] \ No newline at end of file diff --git a/main.py b/main.py deleted file mode 100644 index 1668ade..0000000 --- a/main.py +++ /dev/null @@ -1,22 +0,0 @@ -# module imports -import sys -import json -import logging - -# init logging -logging.basicConfig(stream=sys.stderr, level=logging.DEBUG) - -# variable init -infojson={} -source=sys.argv[1] - -logging.info(f"loading info.json for {source}...") -with open(f"{source}.json") as file: - infojson=json.load(file) - -# TODO: insert method to check url or dict -logging.debug("determining file type...") -import exif -if infojson.extension in exif.allowed_exts: - logging.debug(f"exif module supports {infojson.extesnion} files. using it... ") - \ No newline at end of file diff --git a/getpub.py b/misc/getpub.py similarity index 100% rename from getpub.py rename to misc/getpub.py diff --git a/exif.py b/standards/exif.py similarity index 100% rename from exif.py rename to standards/exif.py diff --git a/iptc.py b/standards/iptc.py similarity index 100% rename from iptc.py rename to standards/iptc.py diff --git a/matroska.py b/standards/matroska.py similarity index 100% rename from matroska.py rename to standards/matroska.py diff --git a/png.py b/standards/png.py similarity index 100% rename from png.py rename to standards/png.py diff --git a/xmp.py b/standards/xmp.py similarity index 84% rename from xmp.py rename to standards/xmp.py index ba9cb18..ca6f3aa 100644 --- a/xmp.py +++ b/standards/xmp.py @@ -102,22 +102,26 @@ def writeTags(file, tags): if date in tags: xmptags['Xmp.dc.date'] = tags.date - from getpub import getPub + from ..misc.getpub import getPub xmptags['Xmp.dc.publisher'] = getPub(tags) del getPub keywords=[] - if tags in tags: for item in tags.tags: if item not in keywords: keywords.append(item) + keywords.append(tags.category) - if theme in tags and tags.theme not in keywords: - keywords.append(tags.theme) + if tags in tags: + for item in tags.tags: + if item not in keywords: keywords.append(item) - if rating in tags and tags.rating not in keywords: - keywords.append(tags.rating) - - if species in tags and tags.species not in keywords: - keywords.append(tags.species) - - if gender in tags and tags.gender not in keywords: - keywords \ No newline at end of file + if tags.category == 'deviantart': + from ..category.deviantart import makeKeyWordList + for item in makeKeyWordList(tags): + if item not in keywords: keywords.append(item) + elif tags.category == 'furaffinity': + from ..category.furaffinity import makeKeyWordList + for item in makeKeyWordList(tags): + if item not in keywords: keywords.append(item) + + + \ No newline at end of file