fix python isprime

2019-10-18 22:55:26 +02:00
parent 4e59fa8048
commit 5adde814c9
4 changed files with 9 additions and 9 deletions
--- a/isprime-python/app.py
+++ b/isprime-python/app.py
@@ -4,10 +4,16 @@ from flask import Flask

 app = Flask(__name__)

+def safe_cast(val, to_type, default=107107):
+    try:
+        return to_type(val)
+    except (ValueError, TypeError):
+        return default

@app.route('/')
 def isprime():
-    num = os.environ.get('TARGET', '107107')
+    num = safe_cast(os.environ.get('TARGET', 107107),int)
+
    if num > 1:
        for i in range(2, num):
            if (num % i) == 0: