save project

diy.py

@@ -5,6 +5,11 @@ def new_input():
         if unit == 'q':
         if unit == 'q':
             break
             break
         else:
         else:
+            try:
+                unit = int(unit) 
+            except:
+                print('请输入数字')
+            else:
                 total.append(unit)
                 total.append(unit)
     print(total)
     print(total)
 new_input()
 new_input()
\ No newline at end of file