larigira/larigira/audiogen.py
boyska d1a1202552
Some checks failed
Build documentation / build (push) Failing after 2s
Install and run tests / test (3.10) (push) Failing after 2s
Install and run tests / test (3.11) (push) Failing after 2s
Install and run tests / test (3.12) (push) Failing after 2s
Install and run tests / test (3.13) (push) Failing after 3s
Install and run tests / test (3.8) (push) Failing after 2s
Install and run tests / test (3.9) (push) Failing after 3s
Build documentation / deploy (push) Has been skipped
remove obsolete __future__
2025-03-01 01:25:26 +01:00

65 lines
1.6 KiB
Python

import sys
import argparse
from .entrypoints_utils import get_one_entrypoint
import json
from logging import getLogger, basicConfig
log = getLogger("audiogen")
def get_audiogenerator(kind):
"""Messes with entrypoints to return an audiogenerator function"""
return get_one_entrypoint("larigira.audiogenerators", kind)
def get_parser():
parser = argparse.ArgumentParser(description="Generate audio and output paths")
parser.add_argument("--log-level", choices=['DEBUG', 'INFO', 'WARNING', 'DEBUG'], default='WARNING')
parser.add_argument(
"audiospec",
metavar="AUDIOSPEC",
type=str,
nargs=1,
help="filename for audiospec, formatted in json",
)
return parser
def read_spec(fname):
try:
if fname == "-":
return json.load(sys.stdin)
with open(fname) as buf:
return json.load(buf)
except ValueError:
sys.stderr.write("Error: invalid JSON\n")
sys.exit(1)
def check_spec(spec):
if "kind" not in spec:
yield "Missing field 'kind'"
def audiogenerate(spec):
gen = get_audiogenerator(spec["kind"])
return tuple(gen(spec))
def main():
"""Main function for the "larigira-audiogen" executable"""
args = get_parser().parse_args()
basicConfig(level=args.log_level)
spec = read_spec(args.audiospec[0])
errors = tuple(check_spec(spec))
if errors:
log.error("Errors in audiospec")
for err in errors:
sys.stderr.write("Error: {}\n".format(err))
sys.exit(1)
for path in audiogenerate(spec):
print(path)
if __name__ == "__main__":
main()