mirror of
https://github.com/Kvan7/Exiled-Exchange-2.git
synced 2026-09-20 19:55:52 +00:00
Kvan7
51faeb4dc9
* image data stuff * ignore lookup file * update testing * sort items * change sort ot be by refname * more code * working pulling * Fixes #456 Create script to request item's images from trade site #456 * Fixes #457 Update ImageFix to use new saved images before poe1 ones #457 * add images to items
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
import json
|
|
import os
|
|
|
|
SUPPORTED_LANG = ("en", "ru", "ko", "cmn-Hant", "ja", "de", "es")
|
|
|
|
|
|
def get_script_dir(lang="en"):
|
|
"""Returns the directory where the script is located."""
|
|
return os.path.dirname(os.path.realpath(__file__)) + "/" + lang
|
|
|
|
|
|
def read_ndjson(file_path, encoding="utf-8"):
|
|
"""Reads a newline-delimited JSON file and returns a list of parsed JSON objects."""
|
|
with open(file_path, "r", encoding=encoding) as file:
|
|
return [json.loads(line) for line in file]
|
|
|
|
|
|
def read_cache(file_path):
|
|
"""Reads a cache file and returns a dictionary of parsed JSON objects."""
|
|
with open(file_path, "r", encoding="utf-8") as file:
|
|
return json.loads(file.read())
|
|
|
|
|
|
def write_ndjson(file_path, data):
|
|
"""Writes a list of JSON objects to a newline-delimited JSON file."""
|
|
with open(file_path, "w", encoding="utf-8") as file:
|
|
for item in data:
|
|
file.write(json.dumps(item, ensure_ascii=False) + "\n")
|
|
|
|
|
|
def get_save_name(item: dict[str, str]) -> str:
|
|
return item["namespace"] + "=" + item["name"]
|
|
|
|
|
|
def update_icons(new_items, old_items, api_cache={}):
|
|
"""Update icons in new_items if they are missing and present in old_items."""
|
|
old_icons = {
|
|
get_save_name(item): item["icon"] for item in old_items if "icon" in item
|
|
}
|
|
|
|
for new_item in new_items:
|
|
lookup_name = get_save_name(new_item)
|
|
if new_item.get("icon") == "%NOT_FOUND%":
|
|
if lookup_name in api_cache and api_cache[lookup_name] != "%NOT_FOUND%":
|
|
new_item["icon"] = api_cache[lookup_name]
|
|
elif lookup_name in old_icons:
|
|
new_item["icon"] = old_icons[lookup_name]
|
|
|
|
return new_items
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for lang in SUPPORTED_LANG:
|
|
# Determine the directory this script is in
|
|
script_dir = get_script_dir(lang)
|
|
|
|
print(script_dir)
|
|
|
|
# Load the old and new items from the respective files in the script's directory
|
|
old_items = read_ndjson(os.path.join(script_dir, "items.ndjson.old"))
|
|
new_items = read_ndjson(os.path.join(script_dir, "items.ndjson"))
|
|
|
|
# Load the cache from the cache file in the script's directory
|
|
cache_items = read_cache(
|
|
os.path.join(get_script_dir(lang=""), "itemImageCache.json")
|
|
)
|
|
|
|
# Update new_items with icons from old_items where applicable
|
|
updated_new_items = update_icons(new_items, old_items, api_cache=cache_items)
|
|
|
|
# Write the updated new items back to the new file or a new file as needed
|
|
write_ndjson(os.path.join(script_dir, "items.ndjson"), updated_new_items)
|