diff --git a/dungeonbattle/entities/items.py b/dungeonbattle/entities/items.py index 23a1c6e..3edec91 100644 --- a/dungeonbattle/entities/items.py +++ b/dungeonbattle/entities/items.py @@ -1,33 +1,36 @@ -from ..interfaces import Entity, FightingEntity +from ..interfaces import Entity, FightingEntity, Map + class Item(Entity): - held:bool + held: bool def __init__(self, *args, **kwargs): - super().__init__(self, *args, **kwargs) + super().__init__(*args, **kwargs) self.held = False - - def drop(self, x:int, y:int): + + def drop(self, y: int, x: int) -> None: self.held = False - self.move(x, y) - - def hold(self): + self.move(y, x) + + def hold(self) -> None: self.held = True + class Bomb(Item): - damage:int = 5 - exploding:bool + damage: int = 5 + exploding: bool def __init__(self, *args, **kwargs): - super().__init__(self, *args, **kwargs) + super().__init__(*args, **kwargs) self.exploding = False - def drop(self, x:int, y:int): - super.drop(self, x, y) + def drop(self, x: int, y: int) -> None: + super().drop(x, y) self.exploding = True - - def act(self, map): + + def act(self, m: Map) -> None: if self.exploding: - for e in map.entities: - if abs (e.x - self.x) + abs (e.y - self.y) <= 1 and isinstance(e,FightingEntity): + for e in m.entities: + if abs(e.x - self.x) + abs(e.y - self.y) <= 1 and \ + isinstance(e, FightingEntity): e.take_damage(self, self.damage) diff --git a/dungeonbattle/entities/monsters.py b/dungeonbattle/entities/monsters.py index 9aa6c06..59db0e7 100644 --- a/dungeonbattle/entities/monsters.py +++ b/dungeonbattle/entities/monsters.py @@ -1,9 +1,11 @@ -from ..interfaces import FightingEntity +from ..interfaces import FightingEntity, Map + class Monster(FightingEntity): - def act(self, map): + def act(self, m: Map) -> None: pass + class Squirrel(Monster): maxhealth = 10 strength = 3 diff --git a/dungeonbattle/entities/player.py b/dungeonbattle/entities/player.py index b687366..9037236 100644 --- a/dungeonbattle/entities/player.py +++ b/dungeonbattle/entities/player.py @@ -1,5 +1,6 @@ from ..interfaces import FightingEntity + class Player(FightingEntity): maxhealth = 20 strength = 5 diff --git a/dungeonbattle/game.py b/dungeonbattle/game.py index b793fdd..197f0de 100644 --- a/dungeonbattle/game.py +++ b/dungeonbattle/game.py @@ -1,3 +1,5 @@ +from typing import Any + from .interfaces import Map from .mapdisplay import MapDisplay from .settings import Settings @@ -15,7 +17,7 @@ class Game: with TermManager() as term_manager: self._start_game(term_manager.screen) - def _start_game(self, screen): + def _start_game(self, screen: Any) -> None: # TODO Generate map, or make the possibility to load another one m = Map.load("example_map.txt") player = Player() @@ -26,11 +28,11 @@ class Game: while True: screen.clear() screen.refresh() - d.display(player.getPosY(), player.getPosX()) + d.display(player.y, player.y) key = screen.getkey() self.handle_key_pressed(key) - def handle_key_pressed(self, key): + def handle_key_pressed(self, key: str) -> None: # TODO load keys from settings if key == 'z' or key == 'KEY_UP': self.player.move_up() @@ -47,20 +49,14 @@ class Player: y: int = 0 x: int = 0 - def move_up(self): + def move_up(self) -> None: self.y -= 1 - def move_down(self): + def move_down(self) -> None: self.y += 1 - def move_left(self): + def move_left(self) -> None: self.x -= 1 - def move_right(self): + def move_right(self) -> None: self.x += 1 - - def getPosX(self): - return self.x - - def getPosY(self): - return self.y diff --git a/dungeonbattle/interfaces.py b/dungeonbattle/interfaces.py index f1474e0..6546523 100644 --- a/dungeonbattle/interfaces.py +++ b/dungeonbattle/interfaces.py @@ -11,11 +11,12 @@ class Map: height: int tiles: list - def __init__(self, width: int, height: int, tiles: list, entities = []): + def __init__(self, width: int, height: int, tiles: list, + entities: list = None): self.width = width self.height = height self.tiles = tiles - self.entities = entities + self.entities = entities or [] @staticmethod def load(filename: str): @@ -40,7 +41,6 @@ class Map: return Map(width, height, tiles, []) - def draw_string(self) -> str: """ Draw the current map as a string object that can be rendered @@ -66,31 +66,43 @@ class Tile(Enum): class Entity: - x: int y: int + x: int - def move(self, x: int, y: int) -> None: - self.x = x + def __init__(self): + self.y = 0 + self.x = 0 + + def move(self, y: int, x: int) -> None: self.y = y - - def act(self, m:Map): + self.x = x + + def act(self, m: Map) -> None: + """ + Define the action of the entity that is ran each tick. + By default, does nothing. + """ pass + class FightingEntity(Entity): maxhealth: int health: int strength: int + dead: bool def __init__(self): + super().__init__() self.health = self.maxhealth + self.dead = False - def hit(self, opponent) -> None: + def hit(self, opponent: "FightingEntity") -> None: opponent.take_damage(self, self.strength) - - def take_damage(self, attacker, amount:int) -> None: + + def take_damage(self, attacker: "Entity", amount: int) -> None: self.health -= amount if self.health <= 0: self.die() - + def die(self) -> None: - pass + self.dead = True diff --git a/dungeonbattle/interfaces_test.py b/dungeonbattle/interfaces_test.py deleted file mode 100644 index 849e28b..0000000 --- a/dungeonbattle/interfaces_test.py +++ /dev/null @@ -1,14 +0,0 @@ -import unittest - -from dungeonbattle.interfaces import Map - - -class TestInterfaces(unittest.TestCase): - def test_map(self) -> None: - """ - Create a map and check that it is well parsed. - """ - m = Map.load_from_string(".█\n█.\n") - self.assertEqual(m.width, 2) - self.assertEqual(m.height, 2) - self.assertEqual(m.draw_string(), ".█\n█.") diff --git a/dungeonbattle/mapdisplay.py b/dungeonbattle/mapdisplay.py index 2e3b8cc..404e267 100644 --- a/dungeonbattle/mapdisplay.py +++ b/dungeonbattle/mapdisplay.py @@ -1,30 +1,33 @@ #!/usr/bin/env python import curses +from typing import Any + from dungeonbattle.interfaces import Map -class MapDisplay: - def __init__(self, m: Map, player): +class MapDisplay: + def __init__(self, m: Map, player: Any): + # TODO Type the player field with the good type self.map = m - self.pad = curses.newpad(m.height, m.width+1) + self.pad = curses.newpad(m.height, m.width + 1) self.player = player - def update_pad(self): + def update_pad(self) -> None: self.pad.addstr(0, 0, self.map.draw_string()) for e in self.map.entities: self.pad.addch(e.y, e.x, e.img) self.pad.addstr(self.player.getPosY(), self.player.getPosX(), '🐿') - def display(self, y, x): - deltay, deltax = (curses.LINES // 2) + 1, (curses.COLS //2) + 1 - pminrow, pmincol = y-deltay, x-deltax + def display(self, y: int, x: int) -> None: + deltay, deltax = (curses.LINES // 2) + 1, (curses.COLS // 2) + 1 + pminrow, pmincol = y - deltay, x - deltax sminrow, smincol = max(-pminrow, 0), max(-pmincol, 0) deltay, deltax = curses.LINES - deltay, curses.COLS - deltax - smaxrow = self.map.height - (y + deltay) + curses.LINES -1 - smaxrow = min(smaxrow, curses.LINES-1) - smaxcol = self.map.width - (x + deltax) + curses.COLS -1 - smaxcol = min(smaxcol, curses.COLS-1) - pminrow = max(0, min(self.map.height, pminrow)) + smaxrow = self.map.height - (y + deltay) + curses.LINES - 1 + smaxrow = min(smaxrow, curses.LINES - 1) + smaxcol = self.map.width - (x + deltax) + curses.COLS - 1 + smaxcol = min(smaxcol, curses.COLS - 1) + pminrow = max(0, min(self.map.height, pminrow)) pmincol = max(0, min(self.map.width, pmincol)) self.pad.clear() self.update_pad() diff --git a/dungeonbattle/settings.py b/dungeonbattle/settings.py index 064ad69..5c80b7c 100644 --- a/dungeonbattle/settings.py +++ b/dungeonbattle/settings.py @@ -11,14 +11,22 @@ class Settings: We can define the setting by simply use settings.TEXTURE_PACK = 'new_key' """ def __init__(self): - self.KEY_UP_PRIMARY = ['z', 'Touche principale pour aller vers le haut'] - self.KEY_UP_SECONDARY = ['KEY_UP', 'Touche secondaire pour aller vers le haut'] - self.KEY_DOWN_PRIMARY = ['s', 'Touche principale pour aller vers le bas'] - self.KEY_DOWN_SECONDARY = ['KEY_DOWN', 'Touche secondaire pour aller vers le bas'] - self.KEY_LEFT_PRIMARY = ['q', 'Touche principale pour aller vers la gauche'] - self.KEY_LEFT_SECONDARY = ['KEY_LEFT', 'Touche secondaire pour aller vers la gauche'] - self.KEY_RIGHT_PRIMARY = ['d', 'Touche principale pour aller vers la droite'] - self.KEY_RIGHT_SECONDARY = ['KEY_RIGHT', 'Touche secondaire pour aller vers la droite'] + self.KEY_UP_PRIMARY = \ + ['z', 'Touche principale pour aller vers le haut'] + self.KEY_UP_SECONDARY = \ + ['KEY_UP', 'Touche secondaire pour aller vers le haut'] + self.KEY_DOWN_PRIMARY = \ + ['s', 'Touche principale pour aller vers le bas'] + self.KEY_DOWN_SECONDARY = \ + ['KEY_DOWN', 'Touche secondaire pour aller vers le bas'] + self.KEY_LEFT_PRIMARY = \ + ['q', 'Touche principale pour aller vers la gauche'] + self.KEY_LEFT_SECONDARY = \ + ['KEY_LEFT', 'Touche secondaire pour aller vers la gauche'] + self.KEY_RIGHT_PRIMARY = \ + ['d', 'Touche principale pour aller vers la droite'] + self.KEY_RIGHT_SECONDARY = \ + ['KEY_RIGHT', 'Touche secondaire pour aller vers la droite'] self.TEXTURE_PACK = ['ASCII', 'Pack de textures utilisé'] def __getattribute__(self, item: str) -> Any: diff --git a/dungeonbattle/tests/__init__.py b/dungeonbattle/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/dungeonbattle/tests/entities_test.py b/dungeonbattle/tests/entities_test.py new file mode 100644 index 0000000..f7d75f1 --- /dev/null +++ b/dungeonbattle/tests/entities_test.py @@ -0,0 +1,81 @@ +import unittest + +from dungeonbattle.entities.items import Bomb, Item +from dungeonbattle.entities.monsters import Squirrel +from dungeonbattle.entities.player import Player +from dungeonbattle.interfaces import Entity, Map + + +class TestEntities(unittest.TestCase): + def setUp(self) -> None: + """ + Load example map that can be used in tests. + """ + self.map = Map.load("example_map.txt") + + def test_basic_entities(self) -> None: + """ + Test some random stuff with basic entities. + """ + entity = Entity() + entity.move(42, 64) + self.assertEqual(entity.y, 42) + self.assertEqual(entity.x, 64) + self.assertIsNone(entity.act(self.map)) + + def test_fighting_entities(self) -> None: + """ + Test some random stuff with fighting entities. + """ + entity = Squirrel() + self.assertIsNone(entity.act(self.map)) + self.assertEqual(entity.maxhealth, 10) + self.assertEqual(entity.maxhealth, entity.health) + self.assertEqual(entity.strength, 3) + self.assertIsNone(entity.hit(entity)) + self.assertFalse(entity.dead) + self.assertIsNone(entity.hit(entity)) + self.assertFalse(entity.dead) + self.assertIsNone(entity.hit(entity)) + self.assertFalse(entity.dead) + self.assertIsNone(entity.hit(entity)) + self.assertTrue(entity.dead) + + def test_items(self) -> None: + """ + Test some random stuff with items. + """ + item = Item() + self.assertFalse(item.held) + item.hold() + self.assertTrue(item.held) + item.drop(42, 42) + self.assertEqual(item.y, 42) + self.assertEqual(item.x, 42) + + def test_bombs(self) -> None: + """ + Test some random stuff with bombs. + """ + item = Bomb() + squirrel = Squirrel() + self.map.entities.append(item) + self.map.entities.append(squirrel) + squirrel.health = 2 + squirrel.move(41, 42) + item.act(self.map) + self.assertFalse(squirrel.dead) + item.drop(42, 42) + self.assertEqual(item.y, 42) + self.assertEqual(item.x, 42) + item.act(self.map) + self.assertTrue(squirrel.dead) + + def test_players(self) -> None: + """ + Test some random stuff with players. + """ + player = Player() + self.assertEqual(player.strength, 5) + self.assertEqual(player.health, player.maxhealth) + self.assertEqual(player.maxhealth, 20) diff --git a/dungeonbattle/tests/interfaces_test.py b/dungeonbattle/tests/interfaces_test.py new file mode 100644 index 0000000..d6ab078 --- /dev/null +++ b/dungeonbattle/tests/interfaces_test.py @@ -0,0 +1,33 @@ +import unittest + +from dungeonbattle.interfaces import Map, Tile + + +class TestInterfaces(unittest.TestCase): + def test_map(self) -> None: + """ + Create a map and check that it is well parsed. + """ + m = Map.load_from_string(".█\n█.\n") + self.assertEqual(m.width, 2) + self.assertEqual(m.height, 2) + self.assertEqual(m.draw_string(), ".█\n█.") + + def test_load_map(self) -> None: + """ + Try to load a map from a file. + """ + m = Map.load("example_map.txt") + self.assertEqual(m.width, 52) + self.assertEqual(m.height, 17) + + def test_tiles(self) -> None: + """ + Test some things about tiles. + """ + self.assertFalse(Tile.FLOOR.is_wall()) + self.assertTrue(Tile.WALL.is_wall()) + self.assertFalse(Tile.EMPTY.is_wall()) + self.assertTrue(Tile.FLOOR.can_walk()) + self.assertFalse(Tile.WALL.can_walk()) + self.assertTrue(Tile.EMPTY.can_walk()) diff --git a/dungeonbattle/settings_test.py b/dungeonbattle/tests/settings_test.py similarity index 89% rename from dungeonbattle/settings_test.py rename to dungeonbattle/tests/settings_test.py index 4841b5f..5821521 100644 --- a/dungeonbattle/settings_test.py +++ b/dungeonbattle/tests/settings_test.py @@ -18,8 +18,10 @@ class TestSettings(unittest.TestCase): self.assertEqual(settings.KEY_LEFT_SECONDARY, 'KEY_LEFT') self.assertEqual(settings.KEY_RIGHT_SECONDARY, 'KEY_RIGHT') self.assertEqual(settings.TEXTURE_PACK, 'ASCII') - self.assertEqual(settings.get_comment(settings.TEXTURE_PACK), settings.get_comment('TEXTURE_PACK')) - self.assertEqual(settings.get_comment(settings.TEXTURE_PACK), 'Pack de textures utilisé') + self.assertEqual(settings.get_comment(settings.TEXTURE_PACK), + settings.get_comment('TEXTURE_PACK')) + self.assertEqual(settings.get_comment(settings.TEXTURE_PACK), + 'Pack de textures utilisé') settings.TEXTURE_PACK = 'UNICODE' self.assertEqual(settings.TEXTURE_PACK, 'UNICODE')