class Dungeon:
def __init__(self, path="", rows=5, cols=8):
self.__hero = Hero()
self.__hero_pos = (0, 0)
self.__rows = rows
self.__cols = cols
self.__map = []
self.__last_spawn = (0, 0)
if path != "":
with open(path) as f:
lst = f.readlines()
self.__rows = len(lst)
self.__cols = len(lst[0]) - 1
self.__map = [list(elems[:-1]) for elems in lst]
def set_map(self, curr_map):
self.__map = curr_map
def get_rows(self):
return self.__rows
def get_hero(self):
return self.__hero
def get_cols(self):
return self.__cols
def adj(self, row, col):
result = []
if row - 1 >= 0 and self.__map[row - 1][col] == 0:
result.append((row - 1, col))
if row + 1 < self.__rows and self.__map[row + 1][col] == 0:
result.append((row + 1, col))
if col - 1 >= 0 and self.__map[row][col - 1] == 0:
result.append((row, col - 1))
if col + 1 < self.__cols and self.__map[row][col + 1] == 0:
result.append((row, col + 1))
return result
@staticmethod
def generate_map(rows, cols):
dungeon = Dungeon(rows=rows, cols=cols)
# defaulting the self.__map
dun_rows = dungeon.get_rows()
dun_cols = dungeon.get_cols()
curr_map = [[0 for x in range(dun_cols)] for y in range(dun_rows)]
dungeon.set_map(curr_map)
# making the road
lower_end = dun_rows + 2 * dun_cols
upper_end = int(0.6 * dun_rows * dun_cols)
road_len = random.randint(lower_end, upper_end)
start_pos = random.choice(
[(row, col) for row in range(dun_rows) for col in range(dun_cols)])
# start drawing
path_to_curr_elem = []
path_to_curr_elem.append((0, 0))
path_elems = ['.', 'E', '.', 'T', '.', '.']
row = 0
col = 0
to_choose_from = []
curr_map[row][col] = 'S'
while road_len >= 0:
to_choose_from = dungeon.adj(row, col)
if to_choose_from != []:
next_pos = random.choice(to_choose_from)
row = next_pos[0]
col = next_pos[1]
path_to_curr_elem.append((row, col))
if road_len > 0:
curr_map[row][col] = random.choice(path_elems)
else:
curr_map[row][col] = 'G'
road_len -= 1
to_choose_from = []
else:
curr_map[row][col] = '.'
dungeon.set_map(curr_map)
to_choose_from = dungeon.adj(row, col)
while to_choose_from == []:
elem = path_to_curr_elem.pop()
dungeon.set_map(curr_map)
to_choose_from = dungeon.adj(elem[0], elem[1])
row = elem[0]
col = elem[1]
# adding obsticles(such English, much fail)
curr_map = [['#' if elem == 0 else elem for elem in row]
for row in curr_map]
dungeon.set_map(curr_map)
dungeon.spawn(dungeon.get_hero())
return dungeon
def print_map(self):
printable = ""
for row in range(self.__rows):
for col in range(self.__cols):
printable += (self.__map[row][col] + ' ')
printable += '\n'
printable = printable[:-1]
print(printable)
def spawn(self, hero):
weapon = Weapon("Fart gun", 50)
self.__hero.equip(weapon)
self.__hero_pos = 'E'
self.__hero_pos = self.__last_spawn
self.__hero_pos = (0, 0)
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = 'H'
self.print_map()
def hero_atack(self):
enemy = Enemy(health=random.randrange(50, 150),
mana=random.randrange(20, 100),
damage=random.randrange(20, 80))
# finding the enemy
distance = 0
curr_vec = 1
row = self.__hero_pos[0]
col = self.__hero_pos[1]
choices = []
range_to_enemy = 0
atack_range = self.__hero.get_cast_range()
if atack_range:
while row - curr_vec >= 0 and curr_vec <= atack_range:
if self.__map[row - curr_vec][col] == '#':
break
choices.append((row - curr_vec, col))
curr_vec -= 1
curr_vec = 1
while col + curr_vec < self.__cols and curr_vec <= atack_range:
if self.__map[row][col + curr_vec] == '#':
break
choices.append((row, col + curr_vec))
curr_vec += 1
curr_vec = 1
while row + curr_vec < self.__rows and curr_vec <= atack_range:
if self.__map[row + curr_vec][col] == '#':
break
choices.append((row + curr_vec, col))
curr_vec += 1
curr_vec = 1
while col - \
curr_vec >= 0 and curr_vec <= atack_range and self.__map[row][col - curr_vec] != '#':
if self.__map[row][col - curr_vec] == '#':
break
choices.append(row, col + curr_vec)
curr_vec -= 1
fight = Fight(self.__hero, enemy)
fight_pos = 0
for elem in choices:
if self.__map[elem[0]][elem[1]] == 'E':
fight_pos = elem
if fight_pos != 0:
range_to_enemy = max(row - fight_pos[0], col - fight_pos[1])
for i in range(range_to_enemy):
fight.moving_fight()
result = fight.static_fight()
if result:
print("Hero wins")
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = 'H'
return True
else:
print("Hero is dead! Respawning...")
self.spawn(self.__hero)
return True
def __movement(self, new_hero_pos):
if self.__map[new_hero_pos[0]][new_hero_pos[1]] in ['#', 'T', 'E']:
if self.__map[new_hero_pos[0]][new_hero_pos[1]] == '#':
return False
if self.__map[new_hero_pos[0]][new_hero_pos[1]] == 'T':
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = '.'
self.__hero_pos = new_hero_pos
treasure = self.spawn_treasure()
if isinstance(treasure, ManaPotion):
print("Mana pot found")
mana = treasure.get_mana()
self.__hero.take_mana(mana)
print("Hero healed with {} mana".format(mana))
print("Hero's mana is: {}".format(self.__hero.get_mana()))
if isinstance(treasure, HealthPotion):
print("Healing pot found!")
health = treasure.get_health()
self.__hero.take_healing(health)
print("Hero healed with {} health".format(health))
print(
"Hero;s health is: {}".format(
self.__hero.get_health()))
if isinstance(treasure, Weapon):
print("Weapon found!")
self.__hero.equip(treasure)
if isinstance(treasure, Spell):
print("Spell found!")
self.__hero.learn(treasure)
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = 'H'
return True
if self.__map[new_hero_pos[0]][new_hero_pos[1]] == 'E':
print("Enemy found")
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = '.'
self.__hero_pos = new_hero_pos
# enemy spawner
enemy = Enemy(health=random.randrange(50, 150),
mana=random.randrange(20, 100),
damage=random.randrange(20, 80))
# initiate fight
fight = Fight(self.__hero, enemy)
result = fight.static_fight()
if result:
print("Hero wins")
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = 'H'
return True
else:
print("Hero is dead! Respawning...")
self.spawn(self.__hero)
return False
return True
else:
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = '.'
self.__hero_pos = new_hero_pos
self.__map[self.__hero_pos[0]][self.__hero_pos[1]] = 'H'
return True
def move_hero(self, direction):
if direction in ['up', 'down', 'left', 'right']:
if direction == 'up':
if self.__hero_pos[0] >= 1:
new_hero_pos = (self.__hero_pos[0] - 1,
self.__hero_pos[1])
if self.__movement(new_hero_pos):
self.print_map()
return True
else:
return False
if direction == 'down':
if self.__hero_pos[0] + 1 <= self.__rows - 1:
new_hero_pos = (self.__hero_pos[0] + 1,
self.__hero_pos[1])
if self.__movement(new_hero_pos):
self.print_map()
return True
else:
return False
if direction == 'left':
if self.__hero_pos[1] >= 1:
new_hero_pos = (self.__hero_pos[0],
self.__hero_pos[1] - 1)
if self.__movement(new_hero_pos):
self.print_map()
return True
else:
return False
if direction == 'right':
if self.__hero_pos[1] + 1 <= self.__cols - 1:
new_hero_pos = (self.__hero_pos[0],
self.__hero_pos[1] + 1)
if self.__movement(new_hero_pos):
self.print_map()
return True
else:
return False
else:
print("Wrong command!!!")
return True
def spawn_treasure(self):
weapon_damage = random.randrange(10, 130)
weap = Weapon("random", weapon_damage)
spell_damage = random.randrange(5, 90)
spell_cost = random.randrange(2, 60)
spell_range = random.randrange(1, self.__cols - 1)
spell = Spell("random", spell_damage, spell_cost, spell_range)
health_potion = HealthPotion()
mana_potion = ManaPotion()
to_choose_from = [mana_potion,
health_potion,
weap,
spell]
return random.choice(to_choose_from)
