Dungeons-and-Pythons/utils.py at master · TsvetiGenkova/Dungeons-and-Pythons · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
import sys
from Person import Person
from hero import Hero
from enemy import Enemy
from random import randint
from weapon_and_spells import Spell
from weapon_and_spells import Weapon


def check_for_stuff(dungeon_map, x, y, stuff, ran):
    assert type(x) is int and x >= 0, 'X must be integer and positive'
    assert type(y) is int and y >= 0, 'Y must be integer and positive'
    assert type(ran) is int, 'ran must be integer'
    assert ran > 0, 'ran must be positive'
    assert type(stuff) is str, 'Stuff must be string'
    assert stuff == 'H' or stuff == 'E', 'Stuff must be H or E'
    assert type(dungeon_map) is list, 'dungeon_map must be list'
    for i in range(0, ran + 1):
        try:
            dungeon_map[x + i][y]
            if dungeon_map[x + i][y] == stuff:
                return (x + i, y)
            tmp = False
        except IndexError:
            tmp = False
        try:
            dungeon_map[x - i][y]
            if dungeon_map[x - i][y] == stuff and x - i >= 0:
                return (x - i, y)
            tmp = False
        except IndexError:
            tmp = False
        try:
            dungeon_map[x][y + i]
            if dungeon_map[x][y + i] == stuff:
                return (x, y + i)
            tmp = False
        except IndexError:
            tmp = False
        try:
            dungeon_map[x][y - i]
            if dungeon_map[x][y - i] == stuff and y - i >= 0:
                return (x, y - i)
            tmp = False
        except IndexError:
            tmp = False
    return tmp


def check_for_wall(hero_x, hero_y, enemy_x, enemy_y, dungeon_map):

    assert type(hero_x) is int, 'hero_x must be int'
    assert type(hero_y) is int, ' hero_y must be int'
    assert type(enemy_x) is int, 'enemy_x must be int'
    assert type(enemy_y) is int, ' enemy_y must be int'
    assert type(dungeon_map) is list, 'dungeon must be list'
    if (hero_x - enemy_x) == 0 and (hero_y - enemy_y) < 0:
        for i in range(abs(hero_y - enemy_y)):
            if dungeon_map[hero_x][hero_y + i] == '#':
                return True
    if (hero_x - enemy_x) == 0 and (hero_y - enemy_y) > 0:
        for i in range(abs(hero_y - enemy_y)):
            if dungeon_map[hero_x][hero_y - i] == '#':
                return True

    if (hero_x - enemy_x) > 0 and (hero_y - enemy_y) == 0:
        for i in range(abs(hero_x - enemy_x)):
            if dungeon_map[hero_x - i][hero_y] == '#':
                return True
    if (hero_x - enemy_x) < 0 and (hero_y - enemy_y) == 0:
        for i in range(abs(hero_x - enemy_x)):
            if dungeon_map[hero_x + i][hero_y] == '#':
                return True
    return False


class Move():

    def __init__(self, inst):
        assert isinstance(inst, Person), 'Must be instance of Person class.'
        self.inst = inst
        self.cleared = False

    def pick_treasure(self):
        t = []
        with open("loot.txt", 'r') as f:
            for i in f.readlines():
                t.append(i)
        treasure = t[randint(0, len(t) - 1)].split(",")
        if treasure[0] == "weapon":
            tmp = float(treasure[2]) if '.' in treasure[2] else int(
                treasure[2])
            tmp_w = Weapon(name=treasure[1], damage=tmp)
            self.inst.equip(tmp_w)
            tr = f"the weapon \"{tmp_w.name}\" with {tmp_w.get_damage()} dmg"
        elif treasure[0] == "spell":
            tmp = float(treasure[2]) if '.' in treasure[2] else int(
                treasure[2])
            tmp1 = float(treasure[3]) if '.' in treasure[3] else int(
                treasure[3])
            tmp_s = Spell(name=treasure[1], damage=tmp,
                       mana_cost=tmp1, cast_range=int(treasure[4]))
            self.inst.learn(tmp_s)
            tr = f"the spell \"{tmp_s.name}\" with {tmp_s.get_damage()} dmg, {tmp_s.get_mana_cost()} mana cost and range {tmp_s.get_cast_range()}"
        else:
            if isinstance(self.inst, Hero):
                if treasure[0] == "Mana potion":
                    self.inst.take_mana(int(treasure[1]))
                    tr = f"Mana potion with {int(treasure[1])} mana"
                elif treasure[0] == "Health potion":
                    self.inst.take_healing(int(treasure[1]))
                    tr = f"Health potion with {int(treasure[1])} health"
            elif isinstance(self.inst, Enemy):
                print("The enemy can\'t heal themself.")

        return tr

    def is_safe(self, dungeon_map, x, y):
        assert type(x) is int, 'x must be int'
        assert type(y) is int, 'y must be int'
        try:
            dungeon_map[x][y]
            if dungeon_map[x][y] == '#':
                return False
            return True
        except IndexError:
            return False

    def move_util(self, abrv, dungeon_map, curr_x, curr_y, x, y):

        if dungeon_map[curr_x + x][curr_y + y] == ".":
            pass
        elif dungeon_map[curr_x + x][curr_y + y] == "S":
            pass
        elif dungeon_map[curr_x + x][curr_y + y] == "G":
            print("You have cleared the dungeon!")
            self.cleared = True
        elif dungeon_map[curr_x + x][curr_y + y] == "T":
            print(f"Found {self.pick_treasure()}!")
        if isinstance(self.inst, Hero) and dungeon_map[curr_x + x][curr_y + y] == "E":
            pass
            #enemy = Enemy.generate_enemy()
            #enemy_coords = (curr_x + x, curr_y + y)
            #f = Fight(self.inst, enemy, enemy_coords, dungeon_map)
            #f.start_fight()
        dungeon_map[curr_x][curr_y] = '.'
        dungeon_map[curr_x + x][curr_y + y] = abrv
        curr_y += y
        curr_x += x
        return (curr_x, curr_y, dungeon_map)

    def move(self, dungeon_map, curr_x, curr_y, direction):
        assert type(direction) is str, 'Direction must be string'
        "up", "down", "left" and "right"
        assert direction == 'up' or direction == 'down' \
            or direction == 'left' or direction == 'right',\
            'Direction must be up,down, left or right'
        assert type(curr_x) is int and type(
            curr_y) is int, 'curr_x and curr_y must be integers'
        assert curr_x >= 0 and curr_y >= 0, 'curr_x and curr_y must be positive'
        if isinstance(self.inst, Hero):
            abrv = "H"
        elif isinstance(self.inst, Enemy):
            abrv = "E"

        if direction == 'up' and self.is_safe(dungeon_map, curr_x - 1, curr_y):
            return self.move_util(abrv, dungeon_map, curr_x, curr_y, -1, 0)

        elif direction == 'down' and self.is_safe(dungeon_map, curr_x + 1, curr_y):
            return self.move_util(abrv, dungeon_map, curr_x, curr_y, 1, 0)

        elif direction == 'left' and self.is_safe(dungeon_map, curr_x, curr_y - 1):
            return self.move_util(abrv, dungeon_map, curr_x, curr_y, 0, -1)

        elif direction == 'right' and self.is_safe(dungeon_map, curr_x, curr_y + 1):
            return self.move_util(abrv, dungeon_map, curr_x, curr_y, 0, 1)

        return False