浏览代码

Merge branch 'master' of https://git.fop.pe/Foppe/scratchings

Afwas [1337627 6 年之前
父节点
当前提交
b8bdd56539
共有 1 个文件被更改,包括 5 次插入1 次删除
  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