Как общаться с шахматным движком в Python?

на win 7 я могу общаться с шахматным движком через командную строку. Небольшой пример сеанса с Stockfish на Win 7:

C:runStockfish>stockfish-x64.exe
Stockfish 2.2.2 JA SSE42 by Tord Romstad, Marco Costalba and Joona Kiiski
quit

C:runStockfish>

первая строка была выведена двигателем, и "quit" был тем, что я набрал, чтобы выйти из двигателя (есть другие вещи я могу сделать, но мне это ясно).

теперь я хочу общаться с этим движком от python:

import subprocess
engine = subprocess.Popen(
    'stockfish-x64.exe',
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
)
for line in engine.stdout:
    print(line.strip())
engine.stdin.write('quitn')

и я

C:runStockfish>communicate.py
b'Stockfish 2.2.2 JA SSE42 by Tord Romstad, Marco Costalba and Joona Kiiski'

но он не ушел двигателя (нет C:runStockfish > prompt), он продолжает ждать ввода. Я должен закрыть окно вручную. Кажется, что мое сообщение quit (последняя строка скрипта python) не написано в stdin.

другими словами, я могу читать из stdout, но когда я пишу в stdin, ничего не происходит.

что я делаю неправильно и как это правильно делать?


Edit: хорошо, благодаря помощи larsmans я решил это:

Пример сценарий:

import subprocess, time

engine = subprocess.Popen(
    'stockfish-x64.exe',
    universal_newlines=True,
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
)

def put(command):
    print('nyou:nt'+command)
    engine.stdin.write(command+'n')

def get():
    # using the 'isready' command (engine has to answer 'readyok')
    # to indicate current last line of stdout
    engine.stdin.write('isreadyn')
    print('nengine:')
    while True:
        text = engine.stdout.readline().strip()
        if text == 'readyok':
            break
        if text !='':
            print('t'+text)

get()
put('uci')
get()
put('setoption name Hash value 128')
get()
put('ucinewgame')
get()
put('position startpos moves e2e4 e7e5 f2f4')
get()
put('go infinite')
time.sleep(3)
get()
put('stop')
get()
put('quit')

выход:

C:runStockfish>1-communicate.py

engine:
        Stockfish 2.2.2 JA SSE42 by Tord Romstad, Marco Costalba and Joona Kiiski

you:
        uci

engine:
        id name Stockfish 2.2.2 JA SSE42
        id author Tord Romstad, Marco Costalba and Joona Kiiski
        option name Use Search Log type check default false
        option name Search Log Filename type string default SearchLog.txt
        ... etc ...
        uciok

you:
        setoption name Hash value 128

engine:

you:
        ucinewgame

engine:

you:
        position startpos moves e2e4 e7e5 f2f4

engine:

you:
        go infinite

engine:
        info depth 1 seldepth 1 score cp 56 nodes 62 nps 1675 time 37 multipv 1 pv e5f4
        info depth 2 seldepth 2 score cp 48 nodes 804 nps 21157 time 38 multipv 1 pv b8c6 g1h3
        info depth 3 seldepth 3 score cp 64 nodes 1409 nps 37078 time 38 multipv 1 pv b8c6 b1c3 e5f4
        ... etc ...

you:
        stop

engine:
        bestmove e5f4 ponder g1f3

you:
        quit

C:runStockfish>

1 ответов


у вас есть тупик: подпроцесс ждет ввода, в то время как ваша программа ждет, когда она выведет больше строк в

for line in engine.stdout:
    print(line.strip())

этот цикл останавливается только тогда, когда подпроцесс закрывает его stdout.