다음으로 두 유닛을 사용해 공격하는 내용을 구현해 보겠습니다. 공격 부분은 두 유닛이 공통으로 사용합니다. 그래서 함수로 정의해 보겠습니다. 앞에서 작성한 코드에 이어서 다음 내용을 추가합니다.

# 공격 함수
def attack(name, location, damage):
    print("{0} : {1} 방향 적군을 공격합니다. [공격력 {2}]".format(name, location, damage))

보병과 탱크가 1시 방향을 공격하도록 attack() 함수로 명령을 내려 보겠습니다.

attack(name, "1시", damage) # 보병 공격 명령
attack(tank_name, "1시", tank_damage) # 탱크 공격 명령

보병 : 1시 방향 적군을 공격합니다. [공격력 5]
탱크 : 1시 방향 적군을 공격합니다. [공격력 35]

실행해 보니 두 유닛 모두 명령한 대로 공격하는 것을 볼 수가 있습니다. 그런데 인원이 부족해서 공격력이 약해 보입니다. 탱크를 하나 더 만들어 봅시다. 앞에서 tank_를 붙여 tank_name, tank_hp, tank_damage 변수를 정의했으니 이번에는 tank2_를 붙여 tank2_name, tank2_hp, tank2_damage라고 변수를 만들겠습니다. 새로운 탱크를 만들고 공격도 추가한 다음 지금까지 작성한 전체 코드를 실행해 봅시다.