squirrel-battle/dungeonbattle/interfaces.py

76 lines
1.7 KiB
Python

#!/usr/bin/env python
from enum import Enum
class Map:
width: int
height: int
tiles: list
def __init__(self, width: int, height: int, tiles: list, entities: list):
self.width = width
self.height = height
self.tiles = tiles
self.entities = entities
@staticmethod
def load(filename: str):
with open(filename, "r") as f:
file = f.read()
return Map.load_from_string(file)
@staticmethod
def load_from_string(content: str):
lines = content.split("\n")
lines = [line for line in lines if line]
height = len(lines)
width = len(lines[0])
tiles = [[Tile(c)
for x, c in enumerate(line)] for y, line in enumerate(lines)]
return Map(width, height, tiles)
def draw_string(self) -> str:
return "\n".join("".join(tile.value for tile in line)
for line in self.tiles)
class Tile(Enum):
EMPTY = ' '
WALL = '#'
FLOOR = '.'
def is_wall(self) -> bool:
return self == Tile.WALL
def can_walk(self) -> bool:
return not self.is_wall()
class Entity:
x: int
y: int
def move(self, x: int, y: int) -> None:
self.x = x
self.y = y
class FightingEntity(Entity):
maxhealth: int
health: int
strength: int
def __init__(self):
self.health = self.maxhealth
def hit(self, opponent) -> None:
opponent.take_damage(self, self.strength)
def take_damage(self, attacker, amount:int) -> None:
self.health -= amount
if self.health <= 0:
self.die()
def die(self) -> None:
pass