48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import argparse
|
|
|
|
from bemani.format import IIDXMusicDB
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="A utility to patch a IIDX music database.")
|
|
parser.add_argument(
|
|
"infile",
|
|
help="Music DB to work with.",
|
|
type=str,
|
|
)
|
|
parser.add_argument(
|
|
"outfile",
|
|
help="Music DB to overwrite.",
|
|
type=str,
|
|
)
|
|
parser.add_argument(
|
|
"--hide-leggendarias",
|
|
help="Hide leggendarias in normal folders.",
|
|
action="store_true",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
rfp = open(args.infile, 'rb')
|
|
data = rfp.read()
|
|
rfp.close()
|
|
|
|
db = IIDXMusicDB(data)
|
|
if args.hide_leggendarias:
|
|
for song in db.songs:
|
|
if song.title[-1:] == '†' or (
|
|
song.difficulties[0] == 0 and
|
|
song.difficulties[1] == 0 and
|
|
song.difficulties[2] == 12
|
|
):
|
|
print(f'Patching \'{song.title}\' to only appear in leggendaria folder!')
|
|
song.folder = 0x5C
|
|
|
|
print('Generating new database file...')
|
|
wfp = open(args.outfile, 'wb')
|
|
wfp.write(db.get_new_db())
|
|
wfp.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|