redstrate.com/scripts/gen_art.py

410 lines
12 KiB
Python

# SPDX-FileCopyrightText: 2023 Joshua Goins <josh@redstrate.com>
# SPDX-License-Identifier: MIT
import os
import shutil
import json
def write_field(f, key, value):
f.write(key + ": " + value + "\n")
def parse_art_json(output_directory, filename, json_file, threed = False, animation = False, guest = False, comic = False):
print(filename)
json_data = json.load(json_file)
year = None
if "date" in json_data:
if "-" in json_data["date"]:
year = int(json_data["date"].split('-')[0])
else:
year = int(json_data["date"])
directory = str(year)
if guest:
directory = "guest"
base_directory = os.path.join(art_output_directory, directory)
if not os.path.exists(base_directory):
os.mkdir(base_directory)
with open(os.path.join(base_directory, filename) + '.md', 'w') as f:
f.write('---\n')
if "title" in json_data:
write_field(f, 'title', "\"{}\"".format(json_data["title"]))
if "artist" in json_data:
write_field(f, 'artist', "\"{}\"".format(json_data["artist"]))
if "date" in json_data:
if "-" in json_data["date"]:
write_field(f, 'date', json_data["date"])
else:
write_field(f, 'date', str(json_data["date"]) + '-01-01')
write_field(f, 'excludefeed', "true")
write_field(f, 'layout', 'art-detail')
if threed:
write_field(f, 'filename', '/3d/' + filename + '.glb')
if "camera-orbit" in json_data:
write_field(f, 'orbit', json_data["camera-orbit"])
if "camera-target" in json_data:
write_field(f, 'target', json_data["camera-target"])
if "camera-fov" in json_data:
write_field(f, 'fov', json_data["camera-fov"])
else:
if animation:
write_field(f, 'filename', '/animation/' + filename + '.webm')
else:
write_field(f, 'filename', '/art/' + filename + '.webp')
if comic:
write_field(f, 'comic', 'true')
f.write('pages:\n')
for page in json_data["pages"]:
f.write('- filename: ' + page["filename"] + "\n")
f.write(' alt_text: ' + page["alt_text"] + "\n")
if "alt_text" in json_data:
write_field(f, 'alt_text',
"\"" + json_data["alt_text"].replace('\n', '').replace('"', '\\"') + "\"")
write_field(f, 'slug', filename)
if threed:
write_field(f, 'threed', 'true')
if animation:
write_field(f, 'animation', 'true')
characters = []
if "characters" in json_data:
f.write("characters:\n")
for character in json_data["characters"]:
f.write("- " + character + "\n")
characters.append(character)
tags = []
if "tags" in json_data:
f.write("arttags:\n")
for tag in json_data["tags"]:
f.write("- " + tag.lower() + "\n")
tags.append(tag.lower())
if "nsfw" in json_data:
write_field(f, 'nsfw', str(json_data["nsfw"]).lower())
if "mastodon_url" in json_data:
write_field(f, 'mastodon_url', json_data["mastodon_url"])
if "pixiv_url" in json_data:
write_field(f, 'pixiv_url', json_data["pixiv_url"])
if "newgrounds_url" in json_data:
write_field(f, 'newgrounds_url', json_data["newgrounds_url"])
if "program" in json_data:
write_field(f, 'program', json_data["program"])
f.write('---\n')
if "description" in json_data:
f.write(json_data["description"])
f.write('\n')
return (year, characters, tags)
def parse_character_json(output_directory, filename, json_file):
print(f"parsed character: {filename}")
json_data = json.load(json_file)
base_directory = os.path.join(characters_output_directory, filename)
if not os.path.exists(base_directory):
os.mkdir(base_directory)
with open(os.path.join(base_directory, '_index.md'), 'w') as f:
f.write('---\n')
write_field(f, 'layout', 'art-character')
write_field(f, 'title', json_data["name"])
if "age" in json_data:
write_field(f, 'age', json_data["age"])
if "pronouns" in json_data:
write_field(f, 'pronouns', json_data["pronouns"])
if "origin" in json_data:
write_field(f, 'origin', json_data["origin"])
write_field(f, 'slug', filename)
f.write('---\n')
if "description" in json_data:
f.write(json_data["description"])
f.write('\n')
art_data_directory = '../art'
threed_data_directory = '../3d'
animation_data_directory = '../animation'
characters_data_directory = '../characters'
art_output_directory = '../content/art'
guest_art_data_directory = '../guestart'
characters_output_directory = '../content/characters'
comic_data_directory = '../comic'
shutil.rmtree(art_output_directory)
os.mkdir(art_output_directory)
shutil.rmtree(characters_output_directory)
os.mkdir(characters_output_directory)
collected_years = set()
year_stats = {}
total_art = 0
character_stats = {}
tag_stats = {}
new_banner = ""
comissions_enabled = False
for filename in os.listdir(art_data_directory):
f = os.path.join(art_data_directory, filename)
if os.path.isfile(f):
filename_without_ext = os.path.splitext(filename)[0]
if filename_without_ext == ".DS_Store" or filename_without_ext == ".directory":
continue
with open(f, "r") as file:
year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file)
if year in year_stats:
year_stats[year] += 1
else:
year_stats[year] = 1
for character in characters:
if character in character_stats:
character_stats[character] += 1
else:
character_stats[character] = 1
for tag in tags:
if tag in tag_stats:
tag_stats[tag] += 1
else:
tag_stats[tag] = 1
collected_years.add(year)
total_art += 1
for filename in os.listdir(characters_data_directory):
f = os.path.join(characters_data_directory, filename)
if os.path.isfile(f):
filename_without_ext = os.path.splitext(filename)[0]
if filename_without_ext == ".DS_Store" or filename_without_ext == ".directory":
continue
with open(f, "r") as file:
parse_character_json(characters_data_directory, filename_without_ext, file)
for filename in os.listdir(threed_data_directory):
f = os.path.join(threed_data_directory, filename)
if os.path.isfile(f):
filename_without_ext = os.path.splitext(filename)[0]
with open(f, "r") as file:
year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file, True)
if year in year_stats:
year_stats[year] += 1
else:
year_stats[year] = 1
for character in characters:
if character in character_stats:
character_stats[character] += 1
else:
character_stats[character] = 1
for tag in tags:
if tag in tag_stats:
tag_stats[tag] += 1
else:
tag_stats[tag] = 1
collected_years.add(year)
total_art += 1
for filename in os.listdir(animation_data_directory):
f = os.path.join(animation_data_directory, filename)
if os.path.isfile(f):
filename_without_ext = os.path.splitext(filename)[0]
with open(f, "r") as file:
year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file, False, True)
if year in year_stats:
year_stats[year] += 1
else:
year_stats[year] = 1
for character in characters:
if character in character_stats:
character_stats[character] += 1
else:
character_stats[character] = 1
for tag in tags:
if tag in tag_stats:
tag_stats[tag] += 1
else:
tag_stats[tag] = 1
collected_years.add(year)
total_art += 1
for filename in os.listdir(comic_data_directory):
f = os.path.join(comic_data_directory, filename)
if os.path.isfile(f):
filename_without_ext = os.path.splitext(filename)[0]
with open(f, "r") as file:
year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file, False, False, False, True)
if year in year_stats:
year_stats[year] += 1
else:
year_stats[year] = 1
for character in characters:
if character in character_stats:
character_stats[character] += 1
else:
character_stats[character] = 1
for tag in tags:
if tag in tag_stats:
tag_stats[tag] += 1
else:
tag_stats[tag] = 1
collected_years.add(year)
total_art += 1
for year in collected_years:
with open(os.path.join(art_output_directory, str(year), '_index.md'), 'w') as f:
f.write('---\n')
write_field(f, 'title', str(year))
write_field(f, 'layout', 'gallery')
write_field(f, 'json', 'art')
write_field(f, 'thumbnails', 'true')
write_field(f, 'selectedyear', str(year))
f.write('---\n')
with open('../data/art-config.json', 'r') as f:
json_data = json.load(f)
featured_pieces = json_data["featured"]
if "new-banner" in json_data:
new_banner = json_data["new-banner"]
comissions_enabled = json_data["commissions"]
with open(art_output_directory + '/_index.md', 'w') as f:
f.write('---\n')
write_field(f, 'title', 'Art')
write_field(f, 'layout', 'art')
write_field(f, 'summary', 'My personal art gallery.')
write_field(f, 'new_banner', new_banner)
write_field(f, 'commissions', str(comissions_enabled).lower())
f.write('aliases:\n')
f.write('- /gallery\n')
f.write('featured:\n')
for piece_name in featured_pieces:
piece_path = os.path.join(art_data_directory, piece_name + ".json")
with open(piece_path, 'r') as piecef:
json_data = json.load(piecef)
f.write("- filename: " + piece_name + '.webp\n')
f.write(' date: ' + json_data["date"] + '\n')
f.write(' title: ' + json_data["title"] + '\n')
f.write('years:\n')
for year in reversed(list(collected_years)):
f.write('- ' + str(year) + '\n')
f.write('---\n')
os.mkdir(art_output_directory + "/stats")
with open(art_output_directory + '/stats/_index.md', 'w') as f:
f.write('---\n')
write_field(f, 'title', 'Stats')
write_field(f, 'layout', 'art-stats')
write_field(f, 'total', str(total_art))
f.write('years:\n')
for year, num in reversed(sorted(year_stats.items(), key=lambda t: t[1])):
f.write('- year: ' + str(year) + '\n')
f.write(' num: ' + str(num) + '\n')
f.write('characters:\n')
for name, num in sorted(character_stats.items(), key=lambda t: t[1], reverse=True)[:10]:
f.write('- name: ' + str(name) + '\n')
f.write(' num: ' + str(num) + '\n')
f.write('tags:\n')
for name, num in sorted(tag_stats.items(), key=lambda t: t[1]):
f.write('- name: ' + str(name) + '\n')
f.write(' num: ' + str(num) + '\n')
f.write('---\n')
# guest art
os.mkdir(art_output_directory + "/guest")
with open(art_output_directory + '/guest/_index.md', 'w') as f:
f.write('---\n')
write_field(f, 'title', 'Guest Art')
write_field(f, 'layout', 'guestart')
f.write('---\n')
for filename in os.listdir(guest_art_data_directory):
f = os.path.join(guest_art_data_directory, filename)
if os.path.isfile(f):
filename_without_ext = os.path.splitext(filename)[0]
if filename_without_ext == ".DS_Store" or filename_without_ext == ".directory":
continue
with open(f, "r") as file:
year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file, False, False, True)