utils/colors.py

   1 from collections import namedtuple
   2
   3 PLAYER_COLOR_WHITE = 'W'
   4 PLAYER_COLOR_BLACK = 'B'
   5
   6 PLAYER_COLORS = ( PLAYER_COLOR_BLACK, PLAYER_COLOR_WHITE )
   7
   8 class BlackWhite(namedtuple('BlackWhite', 'black white')):
   9     def map_both(self, f):
  10         return BlackWhite(*map( f, self ))
  11
  12     def map_pathway(self, func_list):
  13         bw = self
  14         for f in func_list:
  15             bw = bw.map_both(f)
  16         return bw
  17
  18 def the_other_color(color):
  19     if color == PLAYER_COLOR_BLACK:
  20         return PLAYER_COLOR_WHITE
  21     return PLAYER_COLOR_BLACK