mirror of
https://github.com/kastdeur/pipeband-music.git
synced 2024-10-31 17:43:32 +01:00
179 lines
4.6 KiB
Python
Executable file
179 lines
4.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# vim: fdm=marker noet ts=4
|
|
|
|
# Generate a JSON array of all the pipeband scores
|
|
#+ see bottom for script execution
|
|
|
|
#{{{ Functions
|
|
import os
|
|
import re
|
|
|
|
def walk_through_files(path, file_extension='.ly', exclude_dirs=[]):
|
|
for (dirpath, dirnames, filenames) in os.walk(path):
|
|
if os.path.basename(dirpath) in exclude_dirs:
|
|
continue
|
|
|
|
if '.ignore_scores2json' in filenames:
|
|
dirnames = []
|
|
continue
|
|
|
|
if file_extension:
|
|
for filename in filenames:
|
|
if filename.endswith(file_extension):
|
|
yield os.path.join(dirpath, filename)
|
|
else:
|
|
yield os.path.join(dirpath, filename)
|
|
|
|
def rreplace(s, old, new, occurrence):
|
|
li = s.rsplit(old, occurence)
|
|
return new.join(li)
|
|
|
|
|
|
def search_file_for_tuneinfo(fname, textsearches = [], keywords = []):
|
|
data = {}
|
|
with open(fname, 'r') as fh:
|
|
for line in fh:
|
|
if line.startswith('\\version'):
|
|
continue
|
|
|
|
if line.startswith('%'):
|
|
continue
|
|
|
|
for t in textsearches:
|
|
if not t in data:
|
|
result = re.search(t +'\s*=\s*(?P<quote>[\'\"])(.*?)(?P=quote)', line)
|
|
if result:
|
|
data[t] = result.group(2)
|
|
|
|
for keyword in keywords:
|
|
if not keyword in data:
|
|
result = re.search(keyword+'\s*(.*)', line)
|
|
|
|
if result:
|
|
data[keyword.replace('\\','')] = result.group(1).strip().replace('\\', '')
|
|
return data
|
|
|
|
def read_config(config_path):
|
|
''' return a dict created from the configfile '''
|
|
tuneinfo = search_file_for_tuneinfo(
|
|
config_path,
|
|
textsearches = ['title', 'meter', 'instrument'],
|
|
keywords = ['\\\\tempo', '\\\\time', '\\\\partial', '\\\\key'],
|
|
)
|
|
|
|
if 'partial' in tuneinfo:
|
|
tuneinfo['partial'] = tuneinfo['partial'].split(' ')[0]
|
|
if 'time' in tuneinfo:
|
|
tuneinfo['time'] = tuneinfo['time'].split(' ')[0]
|
|
|
|
return tuneinfo
|
|
|
|
def read_lilypond(fname):
|
|
''' return a dict from file'''
|
|
data = read_config(fname)
|
|
|
|
return data
|
|
|
|
def analyse_tune_ly(tune_file):
|
|
''' return a dict created from the tune file'''
|
|
data = read_lilypond(tune_file)
|
|
|
|
if 'instrument' not in data:
|
|
data['instrument'] = os.path.splitext(os.path.basename(tune_file))[0]
|
|
elif data['instrument']:
|
|
candidate = os.path.splitext(os.path.basename(tune_file))[0]
|
|
if data['instrument'].lower() in candidate.lower():
|
|
low_candidate = candidate.lower().split(sep=data['instrument'].lower(), maxsplit=1)
|
|
if len(low_candidate) > 1:
|
|
data['instrument'] += candidate[len(low_candidate[0]) + len(data['instrument']):]
|
|
|
|
data['instrument'] = data['instrument'].lower()
|
|
|
|
data['path'] = tune_file
|
|
|
|
return data
|
|
|
|
|
|
def merge_data_and_config(data={}, config={}):
|
|
"""
|
|
Merge two dictionaries.
|
|
data is from the file itself, config is from the configfile
|
|
|
|
replaces data['title'] = 'My \\title' with 'My '+config['title']
|
|
"""
|
|
|
|
for k in config.keys():
|
|
if k in data and '\\'+k in data[k]:
|
|
data[k].replace('\\'+k, config[k])
|
|
|
|
# overwrite config values with data
|
|
newdata = { **config, **data }
|
|
|
|
return newdata
|
|
|
|
def config_from_lypath(lypath, default_name='config.ily'):
|
|
''' return the configpath belonging to the lypath '''
|
|
li = lypath.rsplit("/", 1)[0]
|
|
|
|
if False:
|
|
# determine config path from file
|
|
pass
|
|
else:
|
|
configpath = os.path.join(li, default_name)
|
|
return configpath
|
|
|
|
def read_path(music_paths, exclude_dirs=[]):
|
|
"""
|
|
Return a dictionary of tunes with tune info from music_paths
|
|
"""
|
|
tunes = {}
|
|
|
|
# Music Directory to work on
|
|
if isinstance(music_paths, str):
|
|
music_paths = [music_paths]
|
|
|
|
for path in music_paths:
|
|
for f in walk_through_files(path, '.ly', exclude_dirs=exclude_dirs):
|
|
|
|
data = analyse_tune_ly(f)
|
|
config_path = config_from_lypath(f)
|
|
|
|
if os.path.exists(config_path):
|
|
config = read_config(config_path)
|
|
# merge them
|
|
data = merge_data_and_config(data, config)
|
|
|
|
data['files'] = {}
|
|
data['files']['lilypond'] = f.replace(path,'',1)
|
|
for ext in ['pdf', 'midi', 'preview.pdf', 'preview.png']:
|
|
sfile = os.path.splitext(f)[0] + '.' + ext
|
|
|
|
if os.path.isfile(sfile):
|
|
data['files'][ext] = sfile.replace(path,'',1)
|
|
|
|
if 'title' in data:
|
|
tune_key = data['title']
|
|
else:
|
|
tune_key = data['path']
|
|
|
|
if not tune_key in tunes:
|
|
tunes[tune_key] = [data]
|
|
else:
|
|
tunes[tune_key].append(data)
|
|
|
|
return tunes
|
|
#}}} End of functions
|
|
|
|
# ================================================================
|
|
# Alright, lets see if it works...
|
|
# ================================================================
|
|
if __name__ == "__main__":
|
|
import argparse
|
|
import json
|
|
|
|
parser = argparse.ArgumentParser(description="Read a music directory into a JSON file")
|
|
parser.add_argument('music_paths', nargs="*", type=str, default=".")
|
|
|
|
args = parser.parse_args()
|
|
|
|
print(json.dumps(read_path( args.music_paths, exclude_dirs=['template'] ), indent=0))
|