TalkChess.com

Code: Select all

FEN "rnbqkb1r/pp2pppp/3p1n2/8/3NP3/8/PPP2PPP/RNBQKB1R w KQkq - 0 1"
FEN "rnbqkbnr/ppp1pppp/8/3p4/2PP4/8/PP2PPPP/RNBQKBNR b KQkq - 0 1"
[...]

Code: Select all

:: (1) extract games when all pos in a game is not found in all pos from epd file
pgn-extract-17-21 -s -ttag.txt -nsrc-out-notfound-in-epd.pgn src.pgn


:: (2) extract games when at least a pos in a game is found in any pos found in epd file
pgn-extract-17-21 -s -ttag.txt -osrc-out-found-in-epd.pgn src.pgn

Code: Select all

FEN "rnbqkb1r/pp2pppp/3p1n2/8/3NP3/8/PPP2PPP/RNBQKB1R w KQkq - 0 1"

Code: Select all

import sys
import os

    
def delete_file(fn):
    """ Delete file fn """
    if os.path.isfile(fn):
        os.remove(fn)


def creat_tag(in_epd_fn, out_tag):
    """ read epd file and create out_tag """
    # Delete existing tag file
    delete_file(out_tag)
    
    with open(in_epd_fn, 'r') as f:
        for lines in f:
            line = lines.strip()
            splitted_epd = line.split(' ')

            # Take the first 4 fields only
            epd = ' '.join(splitted_epd[:4])

            # Save the epd line with different format
            with open(out_tag, 'a') as fo:
                fo.write('FEN "%s 0 1"\n' % epd)

    
def main(argv):
    """ starts at main """    
    # Vars
    epd_fn = '1.epd'
    tag_fn = 'tag.txt'

    # check epd file 1.epd is existing
    if not os.path.isfile(epd_fn):
        print 'process stopped, %s file is missing' % epd_fn
        raw_input('Press enter key to exit')
        sys.exit(1)

    # Create tag.txt
    creat_tag(epd_fn, tag_fn)

    print 'done!!'
    raw_input('Press enter key to exit')
    

if __name__ == "__main__":
    main(sys.argv[1:])