지금까지 작성한 전체 코드입니다. 게임이 작동하는 데 필요한 클래스를 모두 완성했습니다.
# 일반 유닛
class Unit:
def __init__(self, name, hp, speed):
self.name = name
self.hp = hp
self.speed = speed
print("{0} 유닛을 생성했습니다.".format(name))
def move(self, location):
print("{0} : {1} 방향으로 이동합니다. [속도 {2}]" \
.format(self.name, location, self.speed))
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 AttackUnit(Unit):
def __init__(self, name, hp, damage, speed):
Unit.__init__(self, name, hp, speed)
self.damage = damage
def attack(self, location):
print("{0} : {1} 방향 적군을 공격합니다. [공격력 {2}]" \
.format(self.name, location, self.damage))
# 보병 유닛
class Soldier(AttackUnit):
def __init__(self):
AttackUnit.__init__(self, "보병", 40, 5, 1) # 이름, 체력, 공격력, 이동 속도
# 강화제: 일정 시간 동안 이동 속도와 공격 속도 증가, 체력 10 감소
def booster(self):
if self.hp > 10:
self.hp -= 10 # 체력 10 소모
print("{0} : 강화제를 사용합니다. (HP 10 감소)".format(self.name))
else:
print("{0} : 체력이 부족해 기술을 사용할 수 없습니다".format(self.name))
# 탱크 유닛
class Tank(AttackUnit):
# 시지 모드: 탱크를 지상에 고정, 이동 불가, 공격력 증가
siege_developed = False # 시지 모드 개발 여부
def __init__(self):
AttackUnit.__init__(self, "탱크", 150, 35, 1) # 이름, 체력, 공격력, 이동 속도
self.siege_mode = False # 시지 모드(해제 상태)
# 시지 모드 설정
def set_siege_mode(self):
if Tank.siege_developed == False: # 시지 모드가 개발되지 않았으면 바로 반환
return
if self.siege_mode == False:
print("{0} : 시지 모드로 전환합니다.".format(self.name))
self.damage *= 2 # 공격력 2배 증가
self.siege_mode = True # 시지 모드 설정
else:
print("{0} : 시지 모드를 해제합니다.".format(self.name))
self.damage //= 2 # 공격력 절반으로 감소
self.siege_mode = False # 시지 모드 해제
# 비행 기능
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):
self.fly(self.name, location)
# 전투기 유닛
class Stealth(FlyableAttackUnit): # FlyableAttackUnit 클래스 상속
def __init__(self):
# 부모 클래스 생성자로 기본 정보 설정
FlyableAttackUnit.__init__(self, "전투기", 80, 20, 5)
self.cloaked = False # 은폐 모드(해제 상태), 인스턴스 변수 정의
def cloaking(self): # 은폐 모드를 메서드로 정의
if self.cloaked == True:
print("{0} : 은폐 모드를 해제합니다.".format(self.name))
self.cloaked = False
else:
print("{0} : 은폐 모드를 설정합니다.".format(self.name))
self.cloaked = True