save project

diy2.py

@@ -4,14 +4,16 @@ class hero:
         self.name=name
         self.name=name
         self.hp=hp
         self.hp=hp
         self.attack=attack
         self.attack=attack
+
+    def upgrade(self):
+        self.level+=1
+        self.hp+=20
+        self.attack+=10
+
 zhouyu=hero('周瑜',240,30)
 zhouyu=hero('周瑜',240,30)
 zhangfei=hero('张飞',300,40)
 zhangfei=hero('张飞',300,40)
-def upgrade():
-    zhouyu.level+=1
-    zhouyu.hp+=20
-    zhouyu.attack+=10
-
-upgrade()
+zhouyu.upgrade()
+zhangfei.upgrade()
 print(zhouyu.name)
 print(zhouyu.name)
 print(zhouyu.hp)
 print(zhouyu.hp)
 print(zhouyu.attack)
 print(zhouyu.attack)