다음으로 공격받아 피해를 입는 동작을 정의해 보겠습니다. 적군의 공격 유닛은 종류별로 공격력이 다르고 상황에 따라 피해 규모가 달라질 수 있으므로 피해량에 해당하는 damage를 전달값으로 받겠습니다. 그리고 유닛의 현재 체력 정보에서 damage의 값만큼 빼 줍니다. 그런데 공격받은 후 남은 체력이 0 이하라면 유닛을 사용할 수 없으므로 아쉽지만 유닛을 파괴 처리합니다. 여기서는 편의상 파괴됐다는 문구만 출력하겠습니다.
class AttackUnit: # 공격 유닛
def __init__(self, name, hp, damage):
self.name = name
self.hp = hp
self.damage = damage
def attack(self, location): # 전달받은 방향으로 공격
print("{0} : {1} 방향 적군을 공격합니다. [공격력 {2}]" \
.format(self.name, location, self.damage)) # 공간이 좁아서 2줄로 나눔
def damaged(self, damage): # damage만큼 유닛 피해
# 피해 정보 출력
print("{0} : {1}만큼 피해를 입었습니다.".format(self.name, damage))
self.hp -= damage # 유닛의 체력에서 전달받은 damage만큼 감소
# 남은 체력 출력
print("{0} : 현재 체력은 {1}입니다.".format(self.name, self.hp))
if self.hp <= 0: # 남은 체력이 0 이하이면
print("{0} : 파괴됐습니다.".format(self.name)) # 유닛 파괴 처리