# RPG STAGE 2
# Generates some random enemies.
# Still no input validation.
import random

enemies = ("imp","goblin","orc","troll","DEATH")
numEnemies = int(input("How many enemies would you like to generate? "))

for count in range(1, numEnemies+1):

    enemyType = random.choice(enemies)
    enemyVitality = 4
    enemyStrength = 5
    enemyAgility = 5
    enemyLuck = 3

    if enemyType == "imp":
        enemyAgility += 1
    elif enemyType == "goblin":
        enemyVitality += 1
        enemyLuck += 2
    elif enemyType == "orc":
        enemyVitality += 2
        enemyStrength += 1
    elif enemyType == "troll":
        enemyVitality *= 2
        enemyStrength += 3
        enemyAgility -= 1
    else:
        print("You have encountered DEATH.")
        print("Goodbye!")
        break

    print("Enemy", count, "has the following stats:")
    print("Type:", enemyType)
    print("Vitality:", enemyVitality)
    print("Strength:", enemyStrength)
    print("Agility:", enemyAgility)
    print("Luck:", enemyLuck)

print("DONE")