X-Git-Url: https://plomlompom.com/repos/%7B%7Bprefix%7D%7D?a=blobdiff_plain;f=client.py;h=308a15f1c3ef4df8b619668f11f362baecefa0b1;hb=cea107a6abfd38546d6c41a392ba2c89c1331604;hp=5e91828c7bcb1f01926b7fc5c14ed3404a5f081b;hpb=b05efdc0d63180090a1c945235e47991e10218fa;p=plomrogue2-experiments diff --git a/client.py b/client.py index 5e91828..308a15f 100755 --- a/client.py +++ b/client.py @@ -39,6 +39,7 @@ class UrwidSetup: self.server_output = [] input_handler = getattr(self.InputHandler(self.reply_widget, self.map_widget, + self.turn_widget, self.server_output), 'handle_input') self.urwid_pipe_write_fd = self.main_loop.watch_pipe(input_handler) @@ -54,10 +55,12 @@ class UrwidSetup: - self.reply_widget, a urwid.Text widget printing self.socket replies """ edit_widget = self.EditToSocketWidget(self.socket, 'SEND: ') - self.reply_widget = self.LogWidget('') self.map_widget = self.MapWidget('', wrap='clip') + self.turn_widget = self.TurnWidget('') + self.reply_widget = self.LogWidget('') map_box = urwid.Padding(self.map_widget, width=50) - widget_pile = urwid.Pile([edit_widget, map_box, self.reply_widget]) + widget_pile = urwid.Pile([edit_widget, map_box, self.turn_widget, + self.reply_widget]) return urwid.Filler(widget_pile, valign='top') class EditToSocketWidget(urwid.Edit): @@ -74,11 +77,20 @@ class UrwidSetup: plom_socket_io.send(self.socket, self.edit_text) self.edit_text = '' + class TurnWidget(urwid.Text): + """Displays turn number.""" + + def set_turn(self, turn_string): + turn_string = turn_string.strip() + if not turn_string.isdigit(): + raise ArgumentError('Argument must be non-negative integer.') + self.set_text('TURN: ' + turn_string) + class LogWidget(urwid.Text): - """Display client log, newest message on top.""" + """Displays client log, newest message on top.""" def add(self, text): - """Add text to (top of) log.""" + """Add text plus newline to (top of) log.""" self.set_text(text + '\n' + self.text) class MapWidget(urwid.Text): @@ -86,14 +98,21 @@ class UrwidSetup: map_size = (5, 5) terrain_map = ' ' * 25 position = (0, 0) + things = [] + + class Thing: + def __init__(self, position, symbol): + self.position = position + self.symbol = symbol def draw_map(self): """Draw map view from .map_size, .terrain_map, .position.""" whole_map = [] for c in self.terrain_map: whole_map += [c] - pos_i = self.position[0] * (self.map_size[1] + 1) + self.position[1] - whole_map[pos_i] = '@' + for t in self.things: + pos_i = t.position[0] * (self.map_size[1] + 1) + t.position[1] + whole_map[pos_i] = t.symbol self.set_text(''.join(whole_map)) def get_yx(self, yx_string): @@ -133,22 +152,30 @@ class UrwidSetup: self.terrain_map = terrain_map self.draw_map() - def update_position(self, position_string): - """Update self.position, ensure it's within map bounds.""" - - def get_axis_position_from_argument(axis, token): - if len(token) < 3 or token[:2] != axis + ':' or \ - not token[2:].isdigit(): - raise ArgumentError('Bad arg for ' + axis + ' position.') - return int(token[2:]) - - new_position = self.get_yx(position_string) - if new_position[0] >= self.map_size[0] or \ - new_position[1] >= self.map_size[1]: + def update_things(self, thing_description): + """Append thing of thing_description to self.things.""" + thing_types = {'human': '@', 'monster': 'M'} + tokens = thing_description.split() + if len(tokens) != 2: + raise ArgumentError('Wrong number of tokens.') + yx = self.get_yx(tokens[1]) + if yx[0] >= self.map_size[0] or yx[1] >= self.map_size[1]: raise ArgumentError('Position outside of map size bounds.') - self.position = new_position + type_token = tokens[0] + prefix = 'TYPE:' + type_ = '?' + if len(type_token) <= len(prefix) or \ + type_token[:len(prefix)] != prefix: + raise ArgumentError('Invalid type token.') + type_ = type_token[len(prefix):] + if type_ not in thing_types: + raise ArgumentError('Unknown thing type.') + self.things += [self.Thing(yx, thing_types[type_])] self.draw_map() + def clear_things(self, _): + self.things = [] + class InputHandler: """Delivers data from other thread to widget via message_container. @@ -159,9 +186,11 @@ class UrwidSetup: urwid-external thread. """ - def __init__(self, log_widget, map_widget, message_container): + def __init__(self, log_widget, map_widget, turn_widget, + message_container): self.log_widget = log_widget self.map_widget = map_widget + self.turn_widget = turn_widget self.message_container = message_container def handle_input(self, trigger): @@ -186,15 +215,24 @@ class UrwidSetup: return True return False + def turndraw_command(prefix, func): + n = len(prefix) + if len(msg) > n and msg[:n] == prefix: + m = getattr(self.turn_widget, func) + m(msg[n:]) + return True + return False + msg = self.message_container[0] if msg == 'BYE': raise urwid.ExitMainLoop() return found_command = False try: - found_command = ( + found_command = turndraw_command('NEW_TURN ', 'set_turn') or ( + mapdraw_command('NEW_TURN ', 'clear_things') or mapdraw_command('TERRAIN\n', 'update_terrain') or - mapdraw_command('POSITION ', 'update_position') or + mapdraw_command('THING ', 'update_things') or mapdraw_command('MAP_SIZE ', 'update_map_size')) except ArgumentError as e: self.log_widget.add('ARGUMENT ERROR: ' + msg + '\n' + str(e))