실행하니 화염방사병이 2번 공격받고 각각 25씩 피해를 입었습니다. 체력이 50에서 0이 돼 아쉽게도 파괴됐다는 메시지를 받았습니다.
새로운 AttackUnit 클래스를 정의하고 __init__() 생성자를 제외한 2개 메서드를 추가로 정의한 후 각 메서드를 호출하는 부분까지 다뤘습니다. AttackUnit 클래스와 관련 있는 전체 코드는 다음과 같습니다.
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)) # 유닛 파괴 처리
# 화염방사병: 공격 유닛, 화염방사기를 사용함
flamethrower1 = AttackUnit("화염방사병", 50, 16) # 객체 생성, 체력 50, 공격력 16
flamethrower1.attack("5시") # 5시 방향으로 공격 명령
# 25만큼의 공격을 2번 받음
flamethrower1.damaged(25) # 남은 체력 25
flamethrower1.damaged(25) # 남은 체력 0