Newer
Older
from argparse import ArgumentParser, BooleanOptionalAction
from .configuration import (
SfgConfigException, SfgConfigSource,
add_config_args_to_parser, config_from_parser_args, merge_configurations
)
def add_newline_arg(parser):
parser.add_argument("--newline", action=BooleanOptionalAction, default=True,
help="Whether to add a terminating newline to the output.")
def cli_main(program='sfg-cli'):
parser = ArgumentParser(program,
description="pystencilssfg command-line utility for build system integration")
subparsers = parser.add_subparsers(required=True, title="Subcommands")
version_parser = subparsers.add_parser("version", help="Print version and exit.")
add_newline_arg(version_parser)
version_parser.set_defaults(func=version)
outfiles_parser = subparsers.add_parser(
"list-files", help="List files produced by given codegen script.")
outfiles_parser.set_defaults(func=list_files)
add_config_args_to_parser(outfiles_parser)
add_newline_arg(outfiles_parser)
outfiles_parser.add_argument("--sep", type=str, default=" ", dest="sep", help="Separator for list items")
outfiles_parser.add_argument("codegen_script", type=str)
cmake_parser = subparsers.add_parser("cmake", help="Operations for CMake integation")
cmake_subparsers = cmake_parser.add_subparsers(required=True)
modpath = cmake_subparsers.add_parser(
"modulepath", help="Print the include path for the pystencils-sfg cmake module")
add_newline_arg(modpath)
modpath.set_defaults(func=print_cmake_modulepath)
findmod = cmake_subparsers.add_parser("make-find-module",
help="Creates the pystencils-sfg CMake find module as" +
"'FindPystencilsSfg.cmake' in the current directory.")
findmod.set_defaults(func=make_cmake_find_module)
args = parser.parse_args()
args.func(args)
exit(-1) # should never happen
def version(args):
from . import __version__
print(__version__, end=os.linesep if args.newline else '')
exit(0)
def list_files(args):
try:
project_config, cmdline_config = config_from_parser_args(args)
except SfgConfigException as exc:
abort_with_config_exception(exc)
config = merge_configurations(project_config, cmdline_config, None)
_, scriptname = path.split(args.codegen_script)
basename = path.splitext(scriptname)[0]
from .emitters import HeaderSourcePairEmitter
emitter = HeaderSourcePairEmitter(config.get_output_spec(basename))
print(args.sep.join(emitter.output_files), end=os.linesep if args.newline else '')
def print_cmake_modulepath(args):
from .cmake import get_sfg_cmake_modulepath
print(get_sfg_cmake_modulepath(), end=os.linesep if args.newline else '')
def make_cmake_find_module(args):
from .cmake import make_find_module
make_find_module()
def abort_with_config_exception(exception: SfgConfigException):
def eprint(*args, **kwargs):
print(*args, file=sys.stderr, **kwargs)
match exception.config_source:
case SfgConfigSource.PROJECT:
eprint(
f"Invalid project configuration: {exception.message}\nCheck your configurator script.")
case SfgConfigSource.COMMANDLINE:
eprint(
f"Invalid configuration on command line: {exception.message}")
case _: assert False, "(Theoretically) unreachable code. Contact the developers."
exit(1)