지금까지 작성한 전체 코드는 다음과 같습니다.
# 일반 유닛
class Unit:
def __init__(self, name, hp, speed): # speed 추가
self.name = name
self.hp = hp
self.speed = speed # 지상 이동 속도
def move(self, location): # 이동 동작 정의
print("[지상 유닛 이동]")
print("{0} : {1} 방향으로 이동합니다. [속도 {2}]" \
.format(self.name, location, self.speed))
# 공격 유닛
class AttackUnit(Unit): # Unit 클래스 상속
def __init__(self, name, hp, damage, speed): # speed 추가
Unit.__init__(self, name, hp, speed) # speed 추가
self.damage = damage
def attack(self, location):
print("{0} : {1} 방향 적군을 공격합니다. [공격력 {2}]" \
.format(self.name, location, self.damage))
def damaged(self, damage):
print("{0} : {1}만큼 피해를 입었습니다.".format(self.name, damage))
self.hp -= damage
print("{0} : 현재 체력은 {1}입니다.".format(self.name, self.hp))
if self.hp <= 0:
print("{0} : 파괴됐습니다.".format(self.name))
# 비행 기능
class Flyable:
def __init__(self, flying_speed): # 비행 속도
self.flying_speed = flying_speed
def fly(self, name, location):
print("{0} : {1} 방향으로 날아갑니다. [속도 {2}]" \
.format(name, location, self.flying_speed))
# 공중 공격 유닛
class FlyableAttackUnit(AttackUnit, Flyable):
def __init__(self, name, hp, damage, flying_speed):
AttackUnit.__init__(self, name, hp, damage, 0)
Flyable.__init__(self, flying_speed)
def move(self, location): # Unit 클래스의 move() 메서드를 오버라이딩
print("[공중 유닛 이동]")
self.fly(self.name, location)
# 호버 바이크: 지상 유닛, 기동성 좋음
hoverbike = AttackUnit("호버 바이크", 80, 20, 10) # 지상 이동 속도 10
# 우주 순양함: 공중 유닛, 체력도 굉장히 좋음, 공격력도 좋음
spacecruiser = FlyableAttackUnit("우주 순양함", 500, 25, 3) # 비행 속도 3
hoverbike.move("11시")
spacecruiser.move("9시") # 오버라이딩한 move() 메서드 호출