def shieldCommand(self, game):
'''Command to change the shields '''
currentShield = game.TheEnterprise.Shield
currentEnergy = game.TheEnterprise.Energy
Util.displayShieldCommand(currentShield, currentEnergy)
try:
shield = int(Util.prompt())
if shield < (-currentShield) or shield > currentEnergy:
Util.printMessage('Invalid amount of energy.')
return
except ValueError:
Util.printMessage('Invalid amount of energy.')
return
game.TheEnterprise.Shield += shield
game.TheEnterprise.Energy -= shield
Util.printMessage('Shield strength is now {0}. Energy level is now {1}', game.TheEnterprise.Shield, game.TheEnterprise.Energy)
Util.prompt()
def shieldCommand(self, game):
'''Command to change the shields '''
currentShield = game.TheEnterprise.Shield
currentEnergy = game.TheEnterprise.Energy
Util.displayShieldCommand(currentShield, currentEnergy)
try:
shield = int(Util.prompt())
if shield < (-currentShield) or shield > currentEnergy:
Util.printMessage('Invalid amount of energy.')
return
except ValueError:
Util.printMessage('Invalid amount of energy.')
return
game.TheEnterprise.Shield += shield
game.TheEnterprise.Energy -= shield
Util.printMessage(
'Shield strength is now {0}. Energy level is now {1}',
game.TheEnterprise.Shield, game.TheEnterprise.Energy)
Util.prompt()
