def removeFigure(self, Figure):
# if Figure.short == 'k' or Figure.short == 'K':
# return None
#print(Figure in self.getFigures(),self.is_the_piece_on_the_board_pos(Figure.pos) )
if Figure in self.getFigures() and self.is_the_piece_on_the_board_pos(
Figure.pos):
self.figures.remove(Figure)
#print('succesful remove', Figure)
self.board[Figure.getPos()[0]][Figure.getPos()[1]] = '_'
else:
#print('no such figure',Figure, Figure.pos)
pass
def addFigure(self, Figure):
for figure in self.figures:
if figure.pos == Figure.pos and figure.short == Figure.short:
#print(figure.pos, Figure,' nije uspio add')
if Figure.short == 'K' or Figure.short == 'k':
if self.board[Figure.getPos()[0]][Figure.getPos()
[1]] != Figure.short:
self.board[Figure.getPos()[0]][Figure.getPos()
[1]] = Figure.short
return None
if self.is_the_piece_on_the_board_pos(Figure.pos):
self.board[Figure.getPos()[0]][Figure.getPos()[1]] = Figure.short
self.figures.append(Figure)
#print('uspio add', Figure)
return Figure
return None
