Explorar el Código

Try / catch for parsing str to int

Foppe Hemminga hace 6 años
padre
commit
16d8ed8d65
Se han modificado 1 ficheros con 5 adiciones y 1 borrados
  1. 5 1
      main.py

+ 5 - 1
main.py

@@ -8,7 +8,11 @@ import sys
 if __name__ == '__main__':
     row = 0
     if len(sys.argv) > 1:
-        row = sys.argv[1]
+        try:
+            row = int(sys.argv[1])
+        except ValueError:
+            sys.exit(1)
+
     start = time.time()
 
     db = database.db