findPositionsToAnalyze.py

import os
import argparse
import sys
import chess
import chess.pgn
from timeit import default_timer as timer



def parse_args():
    """
    Define an argument parser and return the parsed arguments
    """
    parser = argparse.ArgumentParser(
        prog='analyze',
        description='takes a list of PGN files and analyzes last position'
        'of each PGN')
    parser.add_argument("--file", "-f",
                        help="PGN file with repertoir",
                        required=True,
                        metavar="list_FILE")
    parser.add_argument("--side", "-s",
                        help="white/black (1/0)",
                        type=int,
                        required=True,
                        default="1")
    return parser.parse_args()

def printPath(node):
    moves = []
    while 1:
        if node.move is None:
            break
        moves.append(node.move)
        node = node.parent
    moves.reverse()
    board = chess.Board()
    print( board.variation_san(moves) )


def traverseTree(node, side):
    if node.board().turn == side:
        if not node.variations:
            printPath(node)
            return

        if len(node.variations) > 1:
            printPath(node)

    for next_node in node.variations:
        traverseTree(next_node, side)

def main():
    args = parse_args()

    game = chess.pgn.read_game(open(args.file))
    traverseTree(game, args.side)

    sys.exit()

if __name__ == "__main__":
    main()