save project

diy2.py

@@ -8,17 +8,13 @@ class Hero:
 #xiaowangba=Hero('小王八',250,15)
 #print(wangba.hp)
 #print(xiaowangba.hp)
-def upgrade1():
-    wangba.level=wangba.level+1
-    wangba.hp=wangba.hp+50
-    wangba.attack=wangba.attack+5
-def upgrade2():
-    xiaowangba.level=xiaowangba.level+1
-    xiaowangba.hp=xiaowangba.hp+50
-    xiaowangba.attack=xiaowangba.attack+5
+    def upgrade(self):
+        self.level=self.level+1
+        self.hp=self.hp+50
+        self.attack=self.attack+5
 wangba=Hero('王八',300,20)
 xiaowangba=Hero('小王八',250,15)
-upgrade1()
-upgrade2()
+wangba.upgrade()
+xiaowangba.upgrade()
 print(wangba.hp)
 print(xiaowangba.hp)
\ No newline at end of file