mirror of
https://github.com/dragoonDorise/EmuDeck.git
synced 2025-05-05 16:02:48 +00:00
89 lines
3.8 KiB
Python
89 lines
3.8 KiB
Python
import os
|
|
import json
|
|
import sys
|
|
import re
|
|
|
|
def generate_game_lists(roms_path):
|
|
def collect_game_data(system_dir, extensions):
|
|
game_data = []
|
|
for root, _, files in os.walk(system_dir):
|
|
for file in files:
|
|
filename = os.path.basename(file)
|
|
extension = filename.split('.')[-1]
|
|
name = '.'.join(filename.split('.')[:-1])
|
|
if extension in extensions:
|
|
|
|
name_cleaned = re.sub(r'\(.*?\)', '', name)
|
|
name_cleaned = re.sub(r'\[.*?\]', '', name_cleaned)
|
|
name_cleaned = name_cleaned.strip()
|
|
|
|
name_cleaned = name_cleaned.replace(' ', '_')
|
|
name_cleaned = name_cleaned.replace('-', '_')
|
|
name_cleaned = re.sub(r'_+', '_', name_cleaned)
|
|
name_cleaned = name_cleaned.replace('+', '')
|
|
name_cleaned = name_cleaned.replace('&', '')
|
|
name_cleaned = name_cleaned.replace('!', '')
|
|
name_cleaned = name_cleaned.replace("'", '')
|
|
name_cleaned = name_cleaned.replace('.', '')
|
|
name_cleaned_pegasus = name.replace(',_', ',')
|
|
|
|
clean_name = name_cleaned
|
|
game_img = f"/customimages/emudeck/{clean_name}.jpg"
|
|
game_info = {
|
|
"name": clean_name,
|
|
"filename": os.path.join(root, file),
|
|
"file": name_cleaned_pegasus,
|
|
"img": game_img
|
|
}
|
|
game_data.append(game_info)
|
|
game_data_sorted = sorted(game_data, key=lambda x: x['name'])
|
|
return game_data_sorted
|
|
|
|
roms_dir = roms_path
|
|
valid_system_dirs = []
|
|
|
|
for system_dir in os.listdir(roms_dir):
|
|
if system_dir == "wiiu":
|
|
system_dir = "wiiu/roms"
|
|
if system_dir == "xbox360":
|
|
system_dir = "xbox360/roms"
|
|
full_path = os.path.join(roms_dir, system_dir)
|
|
if os.path.isdir(full_path) and not os.path.islink(full_path) and os.path.isfile(os.path.join(full_path, 'metadata.txt')):
|
|
file_count = sum([len(files) for r, d, files in os.walk(full_path)])
|
|
if file_count > 2:
|
|
valid_system_dirs.append(full_path)
|
|
|
|
game_list = []
|
|
|
|
for system_dir in valid_system_dirs:
|
|
if any(x in system_dir for x in ["/ps3", "/xbox360", "/model2", "/genesiswide", "/mame", "/emulators", "/desktop"]):
|
|
continue
|
|
|
|
with open(os.path.join(system_dir, 'metadata.txt')) as f:
|
|
metadata = f.read()
|
|
collection = next((line.split(':')[1].strip() for line in metadata.splitlines() if line.startswith('collection:')), '')
|
|
shortname = next((line.split(':')[1].strip() for line in metadata.splitlines() if line.startswith('shortname:')), '')
|
|
launcher = next((line.split(':')[1].strip() for line in metadata.splitlines() if line.startswith('launch:')), '').replace('"', '\\"')
|
|
extensions = next((line.split(':')[1].strip().replace(',', ' ') for line in metadata.splitlines() if line.startswith('extensions:')), '').split()
|
|
|
|
games = collect_game_data(system_dir, extensions)
|
|
system_info = {
|
|
"title": collection,
|
|
"id": shortname,
|
|
"launcher": launcher,
|
|
"games": games
|
|
}
|
|
game_list.append(system_info)
|
|
game_list_sorted = sorted(game_list, key=lambda x: x['title'])
|
|
|
|
|
|
json_output = json.dumps(game_list_sorted, indent=4)
|
|
home_directory = os.path.expanduser("~")
|
|
output_file = os.path.join(home_directory, 'emudeck', 'cache', 'roms_games.json')
|
|
with open(output_file, 'w') as f:
|
|
f.write(json_output)
|
|
|
|
roms_path = sys.argv[1]
|
|
|
|
generate_game_lists(f"{roms_path}")
|