atlas/tools/merge_out.py

import json
import os
from aformatter import format_all_entries, per_line_entries
import scale_back

from scale_back import ScaleConfig

merge_source_file = 'temp-atlas.json'

with open(merge_source_file, 'r', encoding='UTF-8') as f1:
	out_json = json.loads(f1.read())

format_all_entries(out_json)

base_image_path = os.path.join('..', 'web', '_img', 'canvas', 'place30')
ScaleConfig.image1 = os.path.join(base_image_path, '159.png')
scale_back.swap_source_dest('164', '165', os.path.join(base_image_path, '163_159.png'))
scale_back.scale_back_entries(out_json)
scale_back.swap_source_dest('165', '166', os.path.join(base_image_path, '164_159.png'))
scale_back.scale_back_entries(out_json)
scale_back.swap_source_dest('166', '167', os.path.join(base_image_path, '165_159.png'))
scale_back.scale_back_entries(out_json)

out_ids = set()
out_dupe_ids = set()
atlas_ids = {}

with open('../web/atlas.json', 'r', encoding='utf-8') as atlas_file:
	atlas_json = json.loads(atlas_file.read())

for i, entry in enumerate(atlas_json):
	atlas_ids[entry['id']] = i

for entry in out_json:
	if entry['id'] in out_ids:
		print(f"Entry {entry['id']} has duplicates! Please resolve this conflict. This will be excluded from the merge.")
		out_dupe_ids.add(entry['id'])
	out_ids.add(entry['id'])

for entry in out_json:
	if entry['id'] in out_dupe_ids:
		continue

	if 'edit' in entry and entry['edit']:
		assert entry['id'] in atlas_ids, "Edit failed! ID not found on Atlas."
		index = atlas_ids[entry['id']]

		assert index != None, "Edit failed! ID not found on Atlas."

		print(f"Edited {atlas_json[index]['id']} with {entry['edit']}")

		del entry['edit']
		atlas_json[index] = entry
	elif entry['id'] in atlas_ids:
		print(f"Edited {entry['id']} manually.")
		atlas_json[atlas_ids[entry['id']]] = entry
	else:
		print(f"Added {entry['id']}.")
		atlas_json.append(entry)

print('Writing...')
with open('../web/atlas.json', 'w', encoding='utf-8') as atlas_file:
	atlas_file.write(per_line_entries(atlas_json))

with open('../data/read-ids.txt', 'a', encoding='utf-8') as read_ids_file:
	with open('temp-read-ids.txt', 'r+', encoding='utf-8') as read_ids_temp_file:
		read_ids_file.writelines(read_ids_temp_file.readlines())
		read_ids_temp_file.truncate(0)

print('All done.')