From ac8c38592da00562b353390637311870f9095760 Mon Sep 17 00:00:00 2001 From: Pim Nelissen Date: Fri, 13 Feb 2026 14:58:11 +0100 Subject: [PATCH] Add CLI entry point with --test flag for building test landscape using LandscapeDirector --- src/pg_rad/main.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 src/pg_rad/main.py diff --git a/src/pg_rad/main.py b/src/pg_rad/main.py new file mode 100644 index 0000000..19f4161 --- /dev/null +++ b/src/pg_rad/main.py @@ -0,0 +1,33 @@ +import argparse + +from pg_rad.logger import setup_logger +from pg_rad.landscape import LandscapeDirector + + +def main(): + parser = argparse.ArgumentParser( + prog="pg-rad", + description="Primary Gamma RADiation landscape tool" + ) + + parser.add_argument( + "--test", + action="store_true", + help="Load and run the test landscape" + ) + parser.add_argument( + "--loglevel", + default="INFO", + choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], + ) + + args = parser.parse_args() + setup_logger(args.loglevel) + + if args.test: + landscape = LandscapeDirector().build_test_landscape() + print(landscape.name) + + +if __name__ == "__main__": + main()