+from parser import ArgError, Parser
+
+
+class Game:
+ turn = 0
+ log_text = ''
+ map_size = (5, 5)
+ terrain_map = ('?'*5+'\n')*4+'?'*5
+ things = []
+
+ class Thing:
+ def __init__(self, position, symbol):
+ self.position = position
+ self.symbol = symbol
+
+ def log(self, msg):
+ """Prefix msg plus newline to self.log_text."""
+ self.log_text = msg + '\n' + self.log_text
+
+ def cmd_THING(self, type_, yx):
+ """Add to self.things at .position yx with .symbol defined by type_."""
+ symbol = '?'
+ if type_ == 'TYPE:human':
+ symbol = '@'
+ elif type_ == 'TYPE:monster':
+ symbol = 'm'
+ self.things += [self.Thing(yx, symbol)]
+ cmd_THING.argtypes = 'string yx_tuple:nonneg'
+
+ def cmd_MAP_SIZE(self, yx):
+ """Set self.map_size to yx, redraw self.terrain_map as '?' cells."""
+ y, x = yx
+ self.map_size = (y, x)
+ self.terrain_map = ''
+ for y in range(self.map_size[0]):
+ self.terrain_map += '?' * self.map_size[1] + '\n'
+ self.terrain_map = self.terrain_map[:-1]
+ cmd_MAP_SIZE.argtypes = 'yx_tuple:nonneg'
+
+ def cmd_TURN_FINISHED(self, n):
+ """Do nothing. (This may be extended later.)"""
+ pass
+ cmd_TURN_FINISHED.argtypes = 'int:nonneg'
+
+ def cmd_NEW_TURN(self, n):
+ """Set self.turn to n, empty self.things."""
+ self.turn = n
+ self.things = []
+ cmd_NEW_TURN.argtypes = 'int:nonneg'
+
+ def cmd_TERRAIN(self, terrain_map):
+ """Reset self.terrain_map from terrain_map."""
+ lines = terrain_map.split('\n')
+ if len(lines) != self.map_size[0]:
+ raise ArgError('wrong map height')
+ for line in lines:
+ if len(line) != self.map_size[1]:
+ raise ArgError('wrong map width')
+ self.terrain_map = terrain_map
+ cmd_TERRAIN.argtypes = 'string'
+
+
+class WidgetManager:
+
+ def __init__(self, socket, game):
+ """Set up all urwid widgets we want on the screen."""
+ self.game = game
+ edit_widget = self.EditToSocketWidget(socket, 'SEND: ')
+ self.map_widget = urwid.Text('', wrap='clip')
+ self.turn_widget = urwid.Text('')
+ self.log_widget = urwid.Text('')
+ map_box = urwid.Padding(self.map_widget, width=50)
+ widget_pile = urwid.Pile([edit_widget, map_box, self.turn_widget,
+ self.log_widget])
+ self.top = urwid.Filler(widget_pile, valign='top')
+
+ def draw_map(self):
+ """Draw map view from .game.terrain_map, .game.things."""
+ whole_map = []
+ for c in self.game.terrain_map:
+ whole_map += [c]
+ for t in self.game.things:
+ pos_i = t.position[0] * (self.game.map_size[1] + 1) + t.position[1]
+ whole_map[pos_i] = t.symbol
+ return ''.join(whole_map)
+
+ def update(self):
+ """Redraw all non-edit widgets."""
+ self.turn_widget.set_text('TURN: ' + str(self.game.turn))
+ self.log_widget.set_text(self.game.log_text)
+ self.map_widget.set_text(self.draw_map())
+
+ class EditToSocketWidget(urwid.Edit):
+ """Extends urwid.Edit with socket to send input on 'enter' to."""