save project

diy.py

@@ -2,17 +2,17 @@ def new_input():
     total = []
     while True:
         unit = input('请输入（q退出）:')
-        if unit = 'q':
+        if unit == 'q':
             break
         else:
             try:
                 unit = int(unit)
-                except:
-                    print('请重新输入一个数字')
-                    else:
-                        total.append(unit)
-                        finally:
-                            print('=======================')
+            except:
+                print('请重新输入一个数字')
+            else:
+                total.append(unit)
+            finally:
+                print('=======================')
     return total
 
 def sum(money):