sh.py

from twisted.internet.protocol import Protocol
import re

class ShellProtocol(Protocol):
    """Define a generic shell-like protocol:
        [login-prompt [input-echo]]
        [password-prompt] [input-echo]]
        [greeting]
        [prompt]
        [input-echo]
        <[output]...>
        <prompt>
    """

    prompts = {'(?:[L|l]ogin|[U|u]ser):\s+$': 'username',
               '[P|p]assword:\s+$': 'password',
               }

    def __init__(self):
        self._buffer = ''

    def dataReceived(self, data):
        self._buffer += data
        re.search(self._re_prompt, self._buffer)
        mo = self.re_prompt.search(self._buffer)
        if mo:
            response, self._buffer = self._buffer[:mo.start()], self._buffer[mo.end():]
            output = mo.groups('output')
            prompt = mo.groups('prompt')
            self._buffer = self._buffer[mo.end():]
            self.responseReceived(response)
            self.dispatchPrompt(prompt, output)

    def responseReceived(self, response):
        pass

    def dispatchPrompt(self, prompt, output):
        if

    def shell_USER(self, message):
        pass

    def shell_PASSWORD(self, message):
        pass

    def shell_DEFAULT(self, message):
        pass