mirror of
https://github.com/vgmstream/vgmstream.git
synced 2024-11-28 08:20:54 +01:00
144 lines
5.2 KiB
Python
144 lines
5.2 KiB
Python
#!/usr/bin/env python3
|
|
import os, glob, argparse
|
|
|
|
#******************************************************************************
|
|
# TXTP DUMPER
|
|
#
|
|
# Creates .txtp from a text list, mainly .m3u with virtual txtp to actual .txtp
|
|
# example:
|
|
#
|
|
# # %TITLE Stage 1
|
|
# bgm.awb#1 .txtp
|
|
# >> creates bgm.awb#1 .txtp
|
|
#
|
|
# data/bgm.fsb #2 : bgm_field.txtp
|
|
# >> creates bgm_field.txtp full txtp, with data/bgm.fsb #2 inside
|
|
#
|
|
#******************************************************************************
|
|
|
|
class Cli(object):
|
|
def _parse(self):
|
|
description = (
|
|
"Makes TXTP from list"
|
|
)
|
|
epilog = (
|
|
"examples:\n"
|
|
" %(prog)s !tags.m3u\n"
|
|
" - make .txtp per line in !tags.m3u\n\n"
|
|
" %(prog)s !tags.m3u -m \n"
|
|
" - make full txtp rather than mini-txtp\n (may overwrite files when using subsongs)\n\n"
|
|
" %(prog)s lines.txt -s sound \n"
|
|
" - make .txtp per line, setting a subdir\n\n"
|
|
"text file example:\n"
|
|
" # creates a mini txtp as-is\n"
|
|
" bgm.fsb #1 .txtp\n"
|
|
" # creates a txtp using the name after ':' with the body before it\n"
|
|
" bgm.fsb #2 : bgm_field.txtp\n"
|
|
)
|
|
|
|
p = argparse.ArgumentParser(description=description, epilog=epilog, formatter_class=argparse.RawTextHelpFormatter)
|
|
p.add_argument('files', help="Files to get (wildcards work)", nargs='+')
|
|
p.add_argument('-s', dest='subdir', help="Create .txtp pointing to subdir")
|
|
p.add_argument('-m', dest='maxitxtp', help="Create regular txtp rather than mini-txtp", action='store_true')
|
|
p.add_argument('-f', dest='force', help="Make .txtp even if line isn't a .txtp", action='store_true')
|
|
p.add_argument('-o', dest='output', help="Output dir (default: current)")
|
|
#p.add_argument('-n', dest='normalize', help="Normalize .txtp formatting")
|
|
return p.parse_args()
|
|
|
|
def start(self):
|
|
args = self._parse()
|
|
if not args.files:
|
|
return
|
|
App(args).start()
|
|
|
|
#******************************************************************************
|
|
|
|
class App(object):
|
|
def __init__(self, args):
|
|
self.args = args
|
|
|
|
def start(self):
|
|
print("TXTP dumper start")
|
|
|
|
filenames = []
|
|
for filename in self.args.files:
|
|
if os.path.isfile(filename): #for files in paths with regex-like format
|
|
filenames += [filename]
|
|
else:
|
|
filenames += glob.glob(filename)
|
|
|
|
for filename in filenames:
|
|
path = self.args.output or '.'
|
|
if self.args.output:
|
|
try:
|
|
os.makedirs(self.args.output)
|
|
except OSError:
|
|
pass
|
|
|
|
count = 0
|
|
with open(filename,'r', encoding='utf-8-sig') as fi:
|
|
for line in fi:
|
|
line = line.strip()
|
|
line = line.rstrip()
|
|
if line.startswith('#'):
|
|
continue
|
|
|
|
if not line.endswith('.txtp'):
|
|
if self.args.force:
|
|
line += '.txtp'
|
|
else:
|
|
continue
|
|
|
|
line.replace('\\', '/')
|
|
|
|
|
|
subdir = self.args.subdir
|
|
if ':' in line:
|
|
index = line.find(':') #internal txtp : txtp name
|
|
|
|
text = line[0:index].strip()
|
|
name = line[index+1:].strip()
|
|
|
|
elif self.args.maxitxtp or subdir:
|
|
index = line.find('.') #first extension
|
|
|
|
if line[index:].startswith('.txtp'): #???
|
|
continue
|
|
|
|
name = line[0:index] + '.txtp'
|
|
|
|
text = line.replace('.txtp', '').strip()
|
|
if subdir:
|
|
subdir.replace('\\', '/')
|
|
if not subdir.endswith('/'):
|
|
subdir = subdir + '/'
|
|
text = subdir + text
|
|
else:
|
|
# should be a mini-txtp, but if name isn't "file.ext.txtp" and just "file.txtp",
|
|
# probably means proper txtp exists and should't be created (when generating from !tags.m3u)
|
|
name = line
|
|
text = ''
|
|
|
|
basename = os.path.basename(name)
|
|
subname, _ = os.path.splitext(basename)
|
|
_, subext = os.path.splitext(subname)
|
|
if not subext:
|
|
print("ignored pre-txtp: %s" % (basename))
|
|
continue
|
|
|
|
outpath = os.path.join(path, name)
|
|
|
|
with open(outpath, 'w') as fo:
|
|
if text:
|
|
fo.write(text)
|
|
pass
|
|
count += 1
|
|
|
|
if not count:
|
|
print("%s: no .txtp found" % (filename))
|
|
else:
|
|
print("%s: total %i .txtp" % (filename, count))
|
|
|
|
if __name__ == "__main__":
|
|
Cli().start()
|