snake_curses_view.py

import curses

from snake import Snake

class SnakeCursesView:

    KEY_MAP = {
	curses.KEY_UP: Snake.UP,
	curses.KEY_DOWN: Snake.DOWN,
	curses.KEY_LEFT: Snake.LEFT,
	curses.KEY_RIGHT: Snake.RIGHT
    }

    def __init__(self, window, game):
	self.w = window
	self.w.erase() # clear the window
	self.game = game
	self.listeners = []

    def draw(self, ch = '*'):
	self.draw_numbers(ch)
	self.draw_snakes(ch)

    def draw_numbers(self, ch):
	num = self.game.get_numbers()
	if num.is_valid():
	    nch = ' '
	    if ch != ' ':
		nch = chr(ord('0') + num.get_number())
	    self.w.addch(num.get_y(), num.get_x(), nch)

    def draw_snakes(self, ch):
	for snake in self.game.get_snakes():
	    for cell in snake.get_cells():
		self.w.addch(cell[1], cell[0], ch)

    def undraw(self):
	self.draw(' ')

    def add_action_listener(self, listener):
	self.listeners.append(listener)

    def got_key(self, ch):
	direction = SnakeCursesView.KEY_MAP[ch]
	for listener in self.listeners:
	    listener.turn_action(direction)