from layup_cmdline.layupargumentparser import LayupArgumentParser
#
# layup demo prepare
#
[docs]
def cmd_demo_prepare(args): # pragma: no cover
#
# NOTE: DO NOT MOVE THESE IMPORTS TO THE TOP LEVEL OF THE MODULE !!!
#
# Importing layup from the function and not at the top-level of the module
# allows us to exit quickly and print the help/error message (in case there
# was a mistake on the command line). Importing layup can take 5 seconds or
# more, and making the user wait that long just to print out an erro message
# is poor user experience.
#
import sys
if args.verb not in ["comet", "convert", "orbitfit", "predict", "visualize"]:
sys.exit("not a layup verb try ['comet', 'convert', 'orbitfit', 'predict', 'visualize']")
if args.verb == "orbitfit" or args.verb == "convert" or args.verb == "comet":
print("creating demo files")
from layup.utilities.layup_copy_demo_files import copy_demo_files
import os
copy_location = os.path.abspath(args.path)
copy_demo_files(args.verb, copy_location, args.force)
else:
print(f"Demo not yet created for {args.verb}")
#
# layup demo howto
#
[docs]
def cmd_demo_howto(args): # pragma: no cover
#
# NOTE: DO NOT MOVE THESE IMPORTS TO THE TOP LEVEL OF THE MODULE !!!
#
# Importing layup from the function and not at the top-level of the module
# allows us to exit quickly and print the help/error message (in case there
# was a mistake on the command line). Importing layup can take 5 seconds or
# more, and making the user wait that long just to print out an erro message
# is poor user experience.
#
import sys
if args.verb not in ["comet", "convert", "orbitfit", "predict", "visualize"]:
sys.exit("not a layup verb try ['comet', 'convert', 'orbitfit', 'predict', 'visualize']")
from layup.utilities.layup_demo_commands import print_demo_command
return print_demo_command(args.verb)
#
# layup demo
#
[docs]
def main():
# Create the top-level parser
parser = LayupArgumentParser(prog="layup demo", description="Prepare and explain how to run layup demos")
subparsers = parser.add_subparsers(
title="commands",
description="Available commands",
help="Command to execute must also specify verb with -v --verb",
dest="command",
)
# Add the `prepare` subcommand
demo_prepare_parser = subparsers.add_parser(
"prepare", help="Prepares the files necessary to run the demo"
)
demo_prepare_parser.set_defaults(func=cmd_demo_prepare)
demo_prepare_parser.add_argument(
dest="verb",
type=str,
help="which verb to prepare from ['comet', 'convert', 'demo','orbitfit', 'predict', 'visualize']",
)
demo_prepare_parser.add_argument(
"-p",
"--path",
type=str,
default="./",
help="Filepath where you want to copy the demo files. Default is current working directory.",
)
demo_prepare_parser.add_argument(
"-f",
"--force",
action="store_true",
default=False,
help="Force deletion/overwrite of existing demo file(s). Default False.",
)
# Add the `howto` subcommand
demo_howto_parser = subparsers.add_parser("howto", help="Show the command to run the layup demo")
demo_howto_parser.set_defaults(func=cmd_demo_howto)
demo_howto_parser.add_argument(
dest="verb",
type=str,
help="which verb to prepare from ['comet', 'convert', 'demo','orbitfit', 'predict', 'visualize']",
)
# Parse the command-line arguments
args = parser.parse_args()
# Call the appropriate function based on the subcommand
if hasattr(args, "func"):
args.func(args)
else:
parser.print_help()
if __name__ == "__main__":
main()