Я не мог найти дерево python, которое я мог бы использовать для создания древовидной структуры для шахматных открытий, поэтому я попытался написать свой собственный. Чтобы продвигаться глубже в дерево, я пытаюсь вернуть подкрот, когда добавляю новую позицию, но кажется, что все позиции добавляются к корню, и мне не дают ссылку на подкрот, как я ожидаю, хотя я сделал проверку, и в корне тоже много внуков.Не передается желаемое значение для дерева python
import chess.pgn
class Node(object):
children = []
score = None
def __init__(self, fen):
self.fen = fen
def add(self, fen):
for c in self.children:
if c.fen == (fen):
print("working")
return c
self.children.append(Node(fen))
return self.children[-1]
root = Node('rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1')
def createTree(fileName):
pgn = open(fileName)
game = chess.pgn.read_game(pgn)
while(game):
next_move = game.variations[0]
fen = next_move.board().fen()
global root
currentRoot = root.add(fen)
while(not next_move.is_end() and next_move.board().fullmove_number <= 5):
next_move = next_move.variations[0]
fen = next_move.board().fen()
currentRoot = currentRoot.add(fen)
print(currentRoot.children)
game = chess.pgn.read_game(pgn)
file = r"C:\all.pgn"
createTree(file)
for n in root.children:
print(n.fen)
Просьба предоставить ссылку для скачивания в библиотеке 'chess.pgn', которую вы используете. –
https://pypi.python.org/pypi/python-chess – Josh