BLACKED
This commit is contained in:
louis
parent
03e51318b0
commit
b6c3c6a454
10 changed files with 819 additions and 654 deletions
224
divatool.py
224
divatool.py
|
|
@ -10,101 +10,147 @@ import db
|
|||
|
||||
LANGUAGES = ["jpn", "eng", "rus"]
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Unpack and process XDU data")
|
||||
parser = argparse.ArgumentParser(description="Unpack and process XDU data")
|
||||
|
||||
subparsers = parser.add_subparsers(metavar="<command>", title="subcommands", dest="command")
|
||||
subparsers = parser.add_subparsers(
|
||||
metavar="<command>", title="subcommands", dest="command"
|
||||
)
|
||||
|
||||
# adx subcommand
|
||||
parser_adx = subparsers.add_parser("adx", help="ADX stuff")
|
||||
parser_adx_subparsers = parser_adx.add_subparsers(metavar="<command>", title="subcommands", dest="subcommand")
|
||||
# adx loop
|
||||
parser_adx_loop = parser_adx_subparsers.add_parser("loop", help="Extract loop data")
|
||||
parser_adx_loop.add_argument("--output", "-o",
|
||||
help="JSON output file for loop data",
|
||||
type=str, dest="JSON_OUT")
|
||||
parser_adx_loop.add_argument("ADX_DIR",
|
||||
help="Input directory containing ADX files",
|
||||
type=str)
|
||||
# adx subcommand
|
||||
parser_adx = subparsers.add_parser("adx", help="ADX stuff")
|
||||
parser_adx_subparsers = parser_adx.add_subparsers(
|
||||
metavar="<command>", title="subcommands", dest="subcommand"
|
||||
)
|
||||
# adx loop
|
||||
parser_adx_loop = parser_adx_subparsers.add_parser("loop", help="Extract loop data")
|
||||
parser_adx_loop.add_argument(
|
||||
"--output",
|
||||
"-o",
|
||||
help="JSON output file for loop data",
|
||||
type=str,
|
||||
dest="JSON_OUT",
|
||||
)
|
||||
parser_adx_loop.add_argument(
|
||||
"ADX_DIR", help="Input directory containing ADX files", type=str
|
||||
)
|
||||
|
||||
# utage subcommand
|
||||
parser_utage = subparsers.add_parser("utage", help="UTAGE stuff")
|
||||
parser_utage_subparsers = parser_utage.add_subparsers(metavar="<command>", title="subcommand", dest="subcommand")
|
||||
# utage crypt
|
||||
parser_utage_crypt = parser_utage_subparsers.add_parser("crypt", help="Encrypt/decrypt utage tsv files")
|
||||
parser_utage_crypt.add_argument("--encrypt", "-e",
|
||||
help="Encrypt (default: Decrypt)",
|
||||
dest="encrypt", action="store_true", default=False)
|
||||
parser_utage_crypt.add_argument("--no-compression", "-n",
|
||||
help="Do not compress/decompress. Only applies to tsv, png will never be compressed",
|
||||
dest="ncomp", action="store_true", default=False)
|
||||
parser_utage_crypt.add_argument("--key", "-k",
|
||||
help="Encryption key (Default: SampleSecretKey)",
|
||||
dest="key", type=str, default="SampleSecretKey")
|
||||
parser_utage_crypt.add_argument("--hex", "-x",
|
||||
help="KEY is hexadecimal (Default: False)",
|
||||
dest="hex", action="store_true", default=False)
|
||||
parser_utage_crypt.add_argument("TARGET",
|
||||
help="Input", type=str)
|
||||
# utage translate
|
||||
parser_utage_translate = parser_utage_subparsers.add_parser("translate", help="Generate keyed tsv and json files")
|
||||
parser_utage_translate.add_argument("INPUT",
|
||||
help="Input", type=str)
|
||||
parser_utage_translate.add_argument("TSVDIR", nargs='?',
|
||||
help="_t.tsv output directory", type=str, default=".")
|
||||
parser_utage_translate.add_argument("JSONDIR", nargs='?',
|
||||
help="json output directory", type=str, default=".")
|
||||
# utage names
|
||||
parser_utage_names = parser_utage_subparsers.add_parser("names", help="Generate and update name files")
|
||||
parser_utage_names.add_argument("INPUT",
|
||||
help="Input", type=str)
|
||||
parser_utage_names.add_argument("OLD", nargs='?',
|
||||
help="Directory with old files", type=str, default=".")
|
||||
# utage subcommand
|
||||
parser_utage = subparsers.add_parser("utage", help="UTAGE stuff")
|
||||
parser_utage_subparsers = parser_utage.add_subparsers(
|
||||
metavar="<command>", title="subcommand", dest="subcommand"
|
||||
)
|
||||
# utage crypt
|
||||
parser_utage_crypt = parser_utage_subparsers.add_parser(
|
||||
"crypt", help="Encrypt/decrypt utage tsv files"
|
||||
)
|
||||
parser_utage_crypt.add_argument(
|
||||
"--encrypt",
|
||||
"-e",
|
||||
help="Encrypt (default: Decrypt)",
|
||||
dest="encrypt",
|
||||
action="store_true",
|
||||
default=False,
|
||||
)
|
||||
parser_utage_crypt.add_argument(
|
||||
"--no-compression",
|
||||
"-n",
|
||||
help="Do not compress/decompress. Only applies to tsv, png will never be compressed",
|
||||
dest="ncomp",
|
||||
action="store_true",
|
||||
default=False,
|
||||
)
|
||||
parser_utage_crypt.add_argument(
|
||||
"--key",
|
||||
"-k",
|
||||
help="Encryption key (Default: SampleSecretKey)",
|
||||
dest="key",
|
||||
type=str,
|
||||
default="SampleSecretKey",
|
||||
)
|
||||
parser_utage_crypt.add_argument(
|
||||
"--hex",
|
||||
"-x",
|
||||
help="KEY is hexadecimal (Default: False)",
|
||||
dest="hex",
|
||||
action="store_true",
|
||||
default=False,
|
||||
)
|
||||
parser_utage_crypt.add_argument("TARGET", help="Input", type=str)
|
||||
# utage translate
|
||||
parser_utage_translate = parser_utage_subparsers.add_parser(
|
||||
"translate", help="Generate keyed tsv and json files"
|
||||
)
|
||||
parser_utage_translate.add_argument("INPUT", help="Input", type=str)
|
||||
parser_utage_translate.add_argument(
|
||||
"TSVDIR", nargs="?", help="_t.tsv output directory", type=str, default="."
|
||||
)
|
||||
parser_utage_translate.add_argument(
|
||||
"JSONDIR", nargs="?", help="json output directory", type=str, default="."
|
||||
)
|
||||
# utage names
|
||||
parser_utage_names = parser_utage_subparsers.add_parser(
|
||||
"names", help="Generate and update name files"
|
||||
)
|
||||
parser_utage_names.add_argument("INPUT", help="Input", type=str)
|
||||
parser_utage_names.add_argument(
|
||||
"OLD", nargs="?", help="Directory with old files", type=str, default="."
|
||||
)
|
||||
|
||||
# diva subcommand
|
||||
parser_diva = subparsers.add_parser("diva", help="Diva stuff")
|
||||
parser_diva_subparsers = parser_diva.add_subparsers(metavar="<command>", title="subcommand", dest="subcommand")
|
||||
parser_diva_quest = parser_diva_subparsers.add_parser("quest", help="Generate quest JSON files")
|
||||
parser_diva_quest.add_argument("INPUT", help="Directory of databases containing Quest*.db and ResourceEntry.db",
|
||||
type=str)
|
||||
parser_diva_quest.add_argument("OLD", help="Directory with old quest files",
|
||||
nargs='?', type=str, default=".")
|
||||
# diva subcommand
|
||||
parser_diva = subparsers.add_parser("diva", help="Diva stuff")
|
||||
parser_diva_subparsers = parser_diva.add_subparsers(
|
||||
metavar="<command>", title="subcommand", dest="subcommand"
|
||||
)
|
||||
parser_diva_quest = parser_diva_subparsers.add_parser(
|
||||
"quest", help="Generate quest JSON files"
|
||||
)
|
||||
parser_diva_quest.add_argument(
|
||||
"INPUT",
|
||||
help="Directory of databases containing Quest*.db and ResourceEntry.db",
|
||||
type=str,
|
||||
)
|
||||
parser_diva_quest.add_argument(
|
||||
"OLD", help="Directory with old quest files", nargs="?", type=str, default="."
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.command == "adx":
|
||||
if args.subcommand == "loop":
|
||||
adx.extract_loop_data_from_dir(args.ADX_DIR, args.JSON_OUT)
|
||||
elif args.command == "utage":
|
||||
if args.subcommand == "crypt":
|
||||
if not args.hex:
|
||||
key = bytearray(args.key, "utf-8")
|
||||
else:
|
||||
key = bytearray.fromhex(args.key)
|
||||
if os.path.isfile(args.TARGET):
|
||||
utage.crypt.crypt_file(args.TARGET, key, args.encrypt, args.ncomp)
|
||||
elif os.path.isdir(args.TARGET):
|
||||
utage.crypt.crypt_dir(args.TARGET, key, args.encrypt, args.ncomp)
|
||||
else:
|
||||
raise FileNotFoundError("Could not find {}".format(args.TARGET))
|
||||
elif args.subcommand == "translate":
|
||||
if os.path.isfile(args.INPUT):
|
||||
utage.translate.translate_file(args.INPUT, args.TSVDIR, args.JSONDIR)
|
||||
elif os.path.isdir(args.INPUT):
|
||||
utage.translate.translate_dir(args.INPUT, args.TSVDIR, args.JSONDIR)
|
||||
else:
|
||||
raise FileNotFoundError("Could not find {}".format(args.INPUT))
|
||||
elif args.subcommand == "names":
|
||||
if os.path.isdir(args.INPUT):
|
||||
utage.names.update_names(args.INPUT, args.OLD, LANGUAGES)
|
||||
else:
|
||||
raise FileNotFoundError(args.INPUT)
|
||||
elif args.command == "diva":
|
||||
if args.subcommand == "quest":
|
||||
diva.quest.update_missions(args.INPUT, args.OLD, LANGUAGES)
|
||||
if args.command == "adx":
|
||||
if args.subcommand == "loop":
|
||||
adx.extract_loop_data_from_dir(args.ADX_DIR, args.JSON_OUT)
|
||||
elif args.command == "utage":
|
||||
if args.subcommand == "crypt":
|
||||
if not args.hex:
|
||||
key = bytearray(args.key, "utf-8")
|
||||
else:
|
||||
key = bytearray.fromhex(args.key)
|
||||
if os.path.isfile(args.TARGET):
|
||||
utage.crypt.crypt_file(args.TARGET, key, args.encrypt, args.ncomp)
|
||||
elif os.path.isdir(args.TARGET):
|
||||
utage.crypt.crypt_dir(args.TARGET, key, args.encrypt, args.ncomp)
|
||||
else:
|
||||
raise FileNotFoundError("Could not find {}".format(args.TARGET))
|
||||
elif args.subcommand == "translate":
|
||||
if os.path.isfile(args.INPUT):
|
||||
utage.translate.translate_file(args.INPUT, args.TSVDIR, args.JSONDIR)
|
||||
elif os.path.isdir(args.INPUT):
|
||||
utage.translate.translate_dir(args.INPUT, args.TSVDIR, args.JSONDIR)
|
||||
else:
|
||||
raise FileNotFoundError("Could not find {}".format(args.INPUT))
|
||||
elif args.subcommand == "names":
|
||||
if os.path.isdir(args.INPUT):
|
||||
utage.names.update_names(args.INPUT, args.OLD, LANGUAGES)
|
||||
else:
|
||||
raise FileNotFoundError(args.INPUT)
|
||||
elif args.command == "diva":
|
||||
if args.subcommand == "quest":
|
||||
diva.quest.update_missions(args.INPUT, args.OLD, LANGUAGES)
|
||||
|
||||
return 0
|
||||
return 0
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
sys.exit()
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
sys.exit()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue