Add check for integer when -d is used

Make sure that the difficulty parameter is an integer.
Mithil467 · Mar 22, 2020 · 7d23f88 · 7d23f88
1 parent 0d6ad86
commit 7d23f88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mitype/app.py b/mitype/app.py
@@ -238,7 +238,11 @@ def parse(self):
                 print("Cannot open file -", args[0])
                 sys.exit(0)
         elif "-d" in opts:
-            limit = int(args[0])
+            try:
+                limit = int(args[0])
+            except:
+                print("Expected an integer")
+                sys.exit(0)
             if limit not in range(1, 6):
                 print("Please enter a difficulty level between 1 and 5")
                 sys.exit(0)