ToolboxToolbox/zcpcore.py
2018-03-06 17:25:35 +08:00

73 lines
1.7 KiB
Python

#!/usr/bin/env python3
from subprocess import run, PIPE
from sys import argv
import tempfile
def read_rsrc_path(path):
BADCHARS = b'\t$" '
if '//' not in path:
with open(path, 'rb') as f:
return f.read()
else:
path, _, rest = path.partition('//')
rtype, rid, *_ = rest.split('/')
try:
int(rid)
except ValueError:
rid = '"%s"' % rid
type_expr = '\'%s\' (%s)' % (rtype, rid)
rez_code = run(['DeRez', '-only', type_expr, path], stdout=PIPE, check=True).stdout
if len(rez_code) < 2:
raise FileNotFoundError
accum = bytearray()
for l in rez_code.split(b'\n'):
if len(l) >= 6 and l[1:2] == b'$':
hx = l[:43].lstrip(BADCHARS).rstrip(BADCHARS)
accum.extend(bytes.fromhex(hx.decode('ascii')))
return bytes(accum)
def write_rsrc_path(path, data):
STEP = 32
if '//' not in path:
with open(path, 'wb') as f:
f.write(data)
else:
path, _, rest = path.partition('//')
rtype, rid, *args = rest.split('/')
if args:
rname = ', "%s"' % args[0]
args = args[1:]
else:
rname = ""
rid = int(rid)
args = ''.join(', %s' % x for x in args)
with tempfile.NamedTemporaryFile(mode='w') as f:
print('data \'%s\' (%d%s%s) {\n' % (rtype, rid, rname, args), file=f)
for o in range(0, len(data), STEP):
chunk = data[o:o+STEP].hex()
print('\t$"%s"' % chunk, file=f)
print('};', file=f)
f.flush()
run(['Rez', '-a', '-o', path, f.name], check=True)