From f148421a3993213cbc005707f0dc679566c3b4e5 Mon Sep 17 00:00:00 2001 From: Florian Festi Date: Tue, 19 Apr 2016 21:02:05 +0200 Subject: [PATCH] Add edge choices to argparser --- boxes/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/boxes/__init__.py b/boxes/__init__.py index 0612af9..d7113de 100755 --- a/boxes/__init__.py +++ b/boxes/__init__.py @@ -229,13 +229,14 @@ class Boxes: elif arg == "bottom_edge": self.argparser.add_argument( "--bottom_edge", action="store", - type=ArgparseEdgeType("Fhs"), default="h", + type=ArgparseEdgeType("Fhs"), choices=list("Fhs"), + default="h", help="edge type for bottom edge") elif arg == "top_edge": self.argparser.add_argument( "--top_edge", action="store", - type=ArgparseEdgeType("eES"), default="e", - help="edge type for top edge") + type=ArgparseEdgeType("eES"), choices=list("eES"), + default="e", help="edge type for top edge") else: raise ValueError("No default for argument", arg)