save project

diy1.py

+from random import randint
 class Hero(object):
     def __init__(self, name):
         self.name = name
@@ -42,4 +43,18 @@ while True:
     print('~'*30)
     print("          战斗开始")
     print('~'*30)
-    choice=input("请选择释放英雄技能(1攻击/2治疗)")
\ No newline at end of file
+    choice=input("请选择释放英雄技能(1攻击/2治疗)")
+    computer_choice=randint(1,2)
+    if choice=="q":
+        break
+    elif choice=="2":
+        houyi.cure()
+    elif choice=="1":
+        houyi.combat(yase)
+    else:
+        print("错")
+        continue
+    if computer_choice==1:
+        yase.combat(houyi)
+    elif computer_choice==2:
+        yase.cure()
\ No newline at end of file