Script to find the "densest" Pro Game

density.py

from multiprocessing.dummy import Pool
import os
import re
from subprocess import Popen, PIPE
from threading import RLock

DIR = 'go4go'

def parse(fn):
    fh = open(fn, 'r')
    data = fh.read()
    res = re.search(r'RE\[[BW]\+R]', data)
    count = len(re.findall(r';[BW]\[', data))
    fh.close()
    return bool(res), count

def gnugo(fn):
    fh = Popen(["gnugo", "-l", fn, "--mode", "gtp"], stdout=PIPE, stdin=PIPE)
    fh.stdin.write('final_status_list alive\n')
    fh.stdin.close()
    res = fh.stdout.read()
    return len(res.split()) - 1

highest = 0
lock = RLock()

def handle_file(fn):
    global lock, highest

    resigned, moves = parse(fn)
    if resigned or moves < highest:
        return
    count = gnugo(fn)

    with lock:
        if count > highest:
            print fn, count, "%.3f" % (count / 361.)
            highest = count

Pool(5).map(
    handle_file,
    [os.path.join(DIR, path) for path in os.listdir(DIR)],
)