1
0
Fork 0
mirror of https://github.com/kastdeur/pipeband-music.git synced 2024-10-31 17:43:32 +01:00
pipeband-music/ly_expander.py

102 lines
3.3 KiB
Python

#!/usr/bin/env python3
import io, os, sys
# Rewrite relative includes ('./', '../') and copy the files into one big file.
# This file is then compilable even after it has been moved around.
def rewrite_local_lilypond_includes( out_file, fname, recursive = True ):
with io.open(fname, 'r', encoding='utf8') as in_file:
for i, line in enumerate(in_file.readlines()):
if line.startswith(u'\ufeff'): continue
if '\\include' in line:
# Rewrite includes to absolute location of file
incline = line.replace('\\include', '').strip('"\'\n ')
# Only copy if it starts with a "./"
if incline.startswith('./') or incline.startswith('../'):
# Make it an absolute location first
inc_fname = os.path.join(os.path.abspath(os.path.dirname(fname)), incline)
try:
with io.open( inc_fname, 'r', encoding='utf8') as inc_file:
out_file.write("% Including Relative File: " + incline + "\n")
if recursive:
rewrite_local_lilypond_includes( out_file, inc_fname, False )
else:
for subline in inc_file.readlines():
out_file.write( subline )
out_file.write("% End of Relative File: " + incline + "\n")
# We are done here
continue
except IOError:
print("{}:{} Error opening File: '{}'".format(fname, i, incline), file=sys.stderr)
out_file.write(line)
if __name__ == "__main__":
from argparse import ArgumentParser
import os
parser = ArgumentParser()
parser.add_argument("files", default='', nargs='+')
command_group = parser.add_mutually_exclusive_group()
command_group.add_argument("-c", "--command", metavar="COMMAND",
dest="runcommand", default=None,
help="Run COMMAND on the generated file")
command_group.add_argument("-l", "--lilypond",
action="store_true", dest="uselilypond", default=False,
help="Run Lilypond on the generated file")
parser.add_argument("-o", "--options", nargs='?',
action="append", dest="options", default=[],
help="Options to pass to lilypond")
parser.add_argument("-k", "--keep",
action="store_false", dest="delete_tempfile", default=True,
help="Keep Temporary File after running COMMAND on it")
args = parser.parse_args()
if args.uselilypond:
args.runcommand = 'lilypond'
if args.runcommand:
import subprocess
import tempfile
else:
import sys
out_file = sys.stdout
for f in args.files:
if args.runcommand:
out_file = tempfile.NamedTemporaryFile( mode='w+', suffix='.ly', delete=args.delete_tempfile, prefix=os.path.basename(f).rsplit('.', 1)[0], dir=os.path.abspath(os.path.dirname(f)) )
out_file.write('\ufeff')
out_file.write("%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%\n")
out_file.write("% This file was preprocessed by a script '" + os.path.basename(__file__) + "'\n")
out_file.write("% that copies all '\\include \"./\"'d files into one file\n")
out_file.write("% such that it can be moved around and still be compiled\n")
out_file.write("%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%\n")
out_file.write("% Original file: " + f + "\n")
out_file.write("%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%\n\n")
rewrite_local_lilypond_includes( out_file, f )
out_file.flush()
if args.runcommand:
print( args.runcommand + " " + " ".join(args.options) + " " + out_file.name )
subprocess.call( args.runcommand + " " + " ".join(args.options) + " " + out_file.name, shell=True )