save project

diy.py

@@ -4,12 +4,14 @@ class Hero:
         self.level=1
         self.hp=hp
         self.attack=attack
+    def upgrade(self):
+        self.level=self.level+1
+        self.hp=self.hp+50
+        self.attack=self.attack+4
 yase=Hero("亚瑟",300,20)
 sunwukong=Hero("孙悟空",280,30)
-def upgrade():
-    yase.level=yase.level+1
-    yase.hp=yase.hp+50
-    yase.attack=yase.attack+4
-upgrade()
+yase.upgrade()
+sunwukong.upgrade()
+
 print(yase.hp)
 print(sunwukong.hp)