81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
import os
|
|
import argparse
|
|
import json
|
|
import sys
|
|
|
|
#
|
|
# [ TD2 SCENERIES JSON LIST GENERATOR ]
|
|
# by masuo
|
|
# v1.0
|
|
#
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description="List and save .lite.sc files from a directory")
|
|
parser.add_argument(
|
|
"directory",
|
|
help="Path to the directory to scan for .sc files"
|
|
)
|
|
parser.add_argument(
|
|
"--output-file", "-o",
|
|
required=True,
|
|
help="Path to save the resulting JSON file"
|
|
)
|
|
parser.add_argument(
|
|
"--save-version", "-v",
|
|
help="Version string to include in the JSON info"
|
|
)
|
|
parser.add_argument(
|
|
"--save-version-date", "-d",
|
|
help="Version date to include in the JSON info"
|
|
)
|
|
return parser.parse_args()
|
|
|
|
def main():
|
|
args = parse_args()
|
|
|
|
# Prompt for version if not provided
|
|
version = args.save_version or input("Enter version: ")
|
|
version_date = args.save_version_date or input("Enter version date (YYYY-MM-DD): ")
|
|
|
|
directory = args.directory
|
|
if not os.path.isdir(directory):
|
|
print(f"Error: '{directory}' is not a valid directory.")
|
|
sys.exit(1)
|
|
|
|
collected = []
|
|
|
|
for fname in sorted(os.listdir(directory)):
|
|
full_path = os.path.join(directory, fname)
|
|
if not os.path.isfile(full_path):
|
|
print(f"Warning: Skipping '{fname}' (not a file)")
|
|
continue
|
|
|
|
if not fname.endswith('.sc'):
|
|
print(f"Warning: Skipping '{fname}' (does not end with .sc)")
|
|
continue
|
|
|
|
if not fname.endswith('.lite.sc'):
|
|
print(f"Warning: '{fname}' ends with .sc but not .lite.sc, skipping")
|
|
continue
|
|
|
|
collected.append(fname)
|
|
|
|
output = {
|
|
"info": {
|
|
"version": version,
|
|
"versionDate": version_date
|
|
},
|
|
"list": collected
|
|
}
|
|
|
|
try:
|
|
with open(args.output_file, 'w', encoding='utf-8') as f:
|
|
json.dump(output, f, ensure_ascii=False, indent=4)
|
|
print(f"Successfully wrote {len(collected)} entries to '{args.output_file}'")
|
|
except Exception as e:
|
|
print(f"Error writing to output file: {e}")
|
|
sys.exit(1)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|