battle.py