From 687511d69e335571fdd77e43c0563252f62e0362 Mon Sep 17 00:00:00 2001 From: Eryn Wells Date: Sun, 8 May 2022 09:54:08 -0700 Subject: [PATCH] Add an ai attribute to Entity --- roguebasin/object.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/roguebasin/object.py b/roguebasin/object.py index 255eaf0..b6473eb 100644 --- a/roguebasin/object.py +++ b/roguebasin/object.py @@ -17,12 +17,14 @@ class Entity: def __init__(self, symbol: str, *, position: Optional[Point] = None, + ai: Optional[Type['AI']] = None, fg: Optional[Tuple[int, int, int]] = None, bg: Optional[Tuple[int, int, int]] = None): self.position = position if position else Point() self.foreground = fg if fg else (255, 255, 255) self.background = bg self.symbol = symbol + self.ai = ai def print_to_console(self, console: tcod.Console) -> None: console.print(x=self.position.x, y=self.position.y, string=self.symbol, fg=self.foreground, bg=self.background)