mirror of
https://github.com/cainan-c/TaikoPythonTools.git
synced 2024-11-24 07:00:12 +01:00
49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
|
import json
|
||
|
|
|
||
|
|
def merge_wordlists(file1_path, file2_path, output_path):
|
||
|
|
# Load the contents of the first wordlist file
|
||
|
|
with open(file1_path, 'r', encoding='utf-8') as file1:
|
||
|
|
data1 = json.load(file1)
|
||
|
|
|
||
|
|
# Load the contents of the second wordlist file
|
||
|
|
with open(file2_path, 'r', encoding='utf-8') as file2:
|
||
|
|
data2 = json.load(file2)
|
||
|
|
|
||
|
|
# Define keys to remove from data1, for space saving reasons. (sorry south americans)
|
||
|
|
keys_to_remove_data1 = ["neutralSpanishText","neutralSpanishFontType","brazilPortugueseText","brazilPortugueseFontType"]
|
||
|
|
|
||
|
|
# Filter out entries from file 1 where key starts with "song_" and remove specific keys
|
||
|
|
filtered_items_data1 = []
|
||
|
|
for item in data1['items']:
|
||
|
|
if not item['key'].startswith('song_'):
|
||
|
|
# Remove specific keys from item
|
||
|
|
filtered_item = {k: v for k, v in item.items() if k not in keys_to_remove_data1}
|
||
|
|
#filtered_items = [item for item in data1['items'] if not item['key'].startswith('song_')]
|
||
|
|
filtered_items_data1.append(filtered_item)
|
||
|
|
|
||
|
|
# Define keys to remove from data2
|
||
|
|
keys_to_remove_data2 = ["japaneseText", "japaneseFontType", "chineseTText","chineseTFontType","koreanText","koreanFontType"]
|
||
|
|
|
||
|
|
# Filter out specific keys from entries in file 2
|
||
|
|
filtered_items_data2 = []
|
||
|
|
for item in data2['items']:
|
||
|
|
# Remove specific keys from item
|
||
|
|
filtered_item = {k: v for k, v in item.items() if k not in keys_to_remove_data2}
|
||
|
|
filtered_items_data2.append(filtered_item)
|
||
|
|
|
||
|
|
# Extend filtered data1 with filtered data2
|
||
|
|
filtered_items_data1.extend(filtered_items_data2)
|
||
|
|
|
||
|
|
# Update data1 with the merged and filtered items
|
||
|
|
data1['items'] = filtered_items_data1
|
||
|
|
|
||
|
|
# Save the updated JSON back to file
|
||
|
|
with open(output_path, 'w', encoding='utf-8') as output_file:
|
||
|
|
json.dump(data1, output_file, indent=4, ensure_ascii=False)
|
||
|
|
|
||
|
|
print(f"Merged wordlists saved to '{output_path}'.")
|
||
|
|
|
||
|
|
# Example usage:
|
||
|
|
merge_wordlists('data\\_console\\ORBIS\\datatableint\\wordlist.json', 'out\\Data\\ORBIS\\datatable\\wordlist.json', 'out\\Data\\ORBIS\\datatable\\wordlist.json')
|
||
|
|
|