python3で外部プログラムとstdin/stdoutで双方向通信を行う

shell_communication.py

import subprocess

class Execute(object):
    """外部プログラムを実行。stdinとstdoutで対話する。"""
    
    def __init__(self, *args_1, **args_2):
        if 'encoding' in args_2:
            self.encoding = args_2.pop('encoding')
        else:
            self.encoding = 'utf-8'
        self.popen = subprocess.Popen(*args_1, stdin=subprocess.PIPE, stdout=subprocess.PIPE, **args_2)
    def __enter__(self):
        return self
    
    def __exit__(self, type, value, traceback):
        self.close()
        
    def send(self, message, recieve=True, incr=False):
        message = message.rstrip('\n')
        if not incr and '\n' in message:
            raise ValueError("message in \\n!")
        self.popen.stdin.write((message + '\n').encode(self.encoding))
        self.popen.stdin.flush()
        if recieve:
            return self.recieve()
        return None
        
    def recieve(self):
        return self.popen.stdout.readline().decode(self.encoding)
    
    def close(self):
        self.popen.stdin.close()