Activity

1. atenonai commented on Nov 27, 2024

   

   atenonai

   on Nov 27, 2024

   Last edited by atenonai

   Author

   More actions

   I figured out the first parts!

   How can I include not only moves but also the move numbers and comments/annotations from the PGN? If anyone could provide some pointers that would be great.

   import chess
   import chess.pgn
   
   pgn_file = '/path'
   repertoire_color = 'black'
   
   if repertoire_color == 'black':
       setup_positions = chess.WHITE  # FEN positions will be before opponents move
   elif repertoire_color == 'white':
       setup_positions = chess.BLACK  # FEN positions will be before opponents move
   
   with open(pgn_file) as pgn:
       parsed_pgn = chess.pgn.read_game(pgn)
   
   def extract_positions(node, color):
       if node.board().turn == color:
           fen = node.board().fen()
           
           if node.variations:
               # Get opponent move
               for variation in node.variations:
                   opponent_move = variation.move
                   # Convert move to algebraic notation
                   algebraic_opponent_move = node.board().san(opponent_move)
                   
                   if variation.variations:
                   # For each found move, check if there's a response in the repertoire
                       response_move = variation.variations[0].move
                       if response_move:
                       # Convert move to algebraic notation
                           algebraic_response_move = variation.board().san(response_move)
                           
                           # Print the position and move pair
                           print(f'[FEN "{fen}"]\n{algebraic_opponent_move} {algebraic_response_move}\n')
       
       # Recursively process nodes to extract all positions (both main line and sidelines)
       for variation in node.variations:
           extract_positions(variation, color)
   
   extract_positions(parsed_pgn, setup_positions)