Integer validator (threat9#235)

* Add ability to cast Option's value to integer

* Fix formatting.

routersploit/test/test_validators.py

@@ -136,6 +136,16 @@ def test_boolify_true_7(self):
         value = "t"
         self.assertEqual(validators.boolify(value), True)
 
+    def test_integer_1(self):
+        self.assertEqual(validators.integer('1'), 1)
+
+    def test_integer_2(self):
+        self.assertEqual(validators.integer('123'), 123)
+
+    def test_integer_3(self):
+        with self.assertRaises(OptionValidationError):
+            validators.integer('foobar')
+
 
 if __name__ == '__main__':
     unittest.main()

routersploit/validators.py

@@ -55,3 +55,11 @@ def boolify(value):
             return False
     else:
         return bool(value)
+
+
+def integer(number):
+    """ Cast Option value to the integer using int() """
+    try:
+        return int(number)
+    except ValueError:
+        raise OptionValidationError("Invalid option. can't cast '{}' to integer.".format(number))