naja/utils.py

   1 import pygame
   2
   3 from naja.constants import BITS
   4
   5
   6 def convert_colour(colour):
   7     if isinstance(colour, pygame.Color):
   8         return colour
   9     if isinstance(colour, tuple):
  10         return pygame.Color(*colour)
  11     if isinstance(colour, basestring):
  12         return pygame.Color(colour)
  13     raise ValueError()
  14
  15
  16 def bit_glyphs(bits):
  17     bit_names = dict((v, k) for k, v in BITS.items())
  18     return '{%s}' % ','.join(bit_names[bit] for bit in reversed(range(8))
  19                              if bit in bits)
  20
  21
  22 def move_glyph(move_name):
  23     return {
  24         'CASTLE': u'\u265c',
  25         'BISHOP': u'\u265d',
  26         'KNIGHT': u'\u265e',
  27     }.get(move_name, move_name.lower())