Improve lattice, model selection error reportingv0.1.2

1 files changed, 10 insertions, 3 deletions

diff --git a/boltzgen.py b/boltzgen.py
index 4317c68..57b3ed0 100755
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -25,12 +25,19 @@ args = argparser.parse_args()
 if args.model is None:
     args.model = "BGK"
 
-lattice = eval("lbm.lattice.%s" % args.lattice)
-model   = eval("lbm.model.%s"   % args.model)
-
 if args.index is None:
     args.index = 'XYZ'
 
+try:
+    lattice = eval("lbm.lattice.%s" % args.lattice)
+except AttributeError:
+    raise Exception("There is no lattice type called '%s'" % args.lattice) from None
+
+try:
+    model = eval("lbm.model.%s" % args.model)
+except AttributeError:
+    raise Exception("There is no LBM model called '%s'" % args.model) from None
+
 generator = Generator(
     model      = model(lattice, tau = float(args.tau), optimize = not args.disable_cse),
     target     = args.target,