from sys import exit as sysexit
from os.path import splitext
import os
from subprocess import *
import time
import popen2

# интерактив с энжином
class GoEngine:
    def __init__(self):
        self._gtpnr = 1
        self.running=False
    
    def StartEngin1(self,cmd):
        print "starting ", cmd
        self.to_gnugo,self.from_gnugo = os.popen2(cmd)
        self.running=True

    def StartEngin2(self,cmd):
        print "starting ", cmd
        p=Popen(cmd, bufsize=-1, stdin=PIPE, stdout=PIPE)
        #p=Popen(["leela080.exe","--gtp"], shell=True, bufsize=-1, stdin=PIPE, stdout=PIPE)
        self.to_gnugo=p.stdin
        self.from_gnugo=p.stdout
        self.running=True
    
    def gtp(self, command):
        if not self.running : 
            # raise BasicError("OOOOPS..... NO GTP ENGINE!!!!")
            print ("OOOOPS..... NO GTP ENGINE!!!!")
            return
        verbose = True
        cmd = str(self._gtpnr) + " " + command
        if verbose:
            print cmd
            sys.stdout.flush()
        self.to_gnugo.write(cmd + "\n")
        self.to_gnugo.flush()
        status = self.from_gnugo.read(1)
        value = status
        while not status == "\n":
            status = self.from_gnugo.read(1)
            value += status
        assert(self.from_gnugo.read(1) == "\n")
        if verbose:
            print value
        retval=value[1 + len(str(self._gtpnr)):]
        #todo - тут надо бы вылавливать ошибки хотя бы так
        self.is_ok=value[0]=="="
        if not self.is_ok:
            print "gtp error!!! "+retval
        self._gtpnr += 1
        return retval
            
    def time_by_move(self, seconds) :
        # всю партию машина играет в режиме байоми, по нужному числу секунд на ход
        return self.gtp("time_settings 1 "+str(seconds)+" 1")

    def quit(self):
        r=self.gtp('quit')
        self.running=False
        return r
        
    def undo(self):
        return self.gtp('undo')

e=GoEngine()
e.StartEngin1("leela080.exe --g")
e.time_by_move(3)
e.quit()
e.StartEngin2("leela080.exe --g")
e.time_by_move(3)
e.quit()