如何使用readline功能构建一个扭曲的python客户端

时间:2012-12-14 02:14:01

标签: python twisted readline

我正在尝试使用Python Twisted为简单的TCP服务器编写客户端。当然我对Python很陌生,刚开始看Twisted,所以我可能做错了。

服务器很简单,您打算使用nc或telnet。没有身份验证。您只需连接并获得一个简单的控制台。我想写一个添加了一些readline功能的客户端(历史记录和电子邮件,比如ctrl-a / ctrl-e就是我追求的)

下面是我编写的代码,与在命令行中使用netcat一样好nc localhost 4118

from twisted.internet import reactor, protocol, stdio
from twisted.protocols import basic
from sys import stdout

host='localhost'
port=4118
console_delimiter='\n'

class MyConsoleClient(protocol.Protocol):
    def dataReceived(self, data):
        stdout.write(data)
        stdout.flush()

    def sendData(self,data):
        self.transport.write(data+console_delimiter)

class MyConsoleClientFactory(protocol.ClientFactory):
    def startedConnecting(self,connector):
        print 'Starting connection to console.'

    def buildProtocol(self, addr):
        print 'Connected to console!'
        self.client = MyConsoleClient()
        self.client.name = 'console'
        return self.client

    def clientConnectionFailed(self, connector, reason):
        print 'Connection failed with reason:', reason

class Console(basic.LineReceiver):
    factory = None
    delimiter = console_delimiter

    def __init__(self,factory):
        self.factory = factory

    def lineReceived(self,line):
        if line == 'quit':
            self.quit()
        else:
            self.factory.client.sendData(line)

    def quit(self):
        reactor.stop()

def main():
    factory = MyConsoleClientFactory()
    stdio.StandardIO(Console(factory))
    reactor.connectTCP(host,port,factory)
    reactor.run()

if __name__ == '__main__':
    main()

输出:

$ python ./console-console-client.py 
Starting connection to console.
Connected to console!
console> version
d305dfcd8fc23dc6674a1d18567a3b4e8383d70e
console> number-events
338
console> quit

我看了

Python Twisted integration with Cmd module

这对我来说真的没用。示例代码工作得很好但是当我介绍网络时,我似乎与stdio有竞争条件。这个较旧的链接似乎提倡类似的方法(在一个单独的线程中运行readline)但我没有做到这一点。

我也研究过扭曲的海螺侮辱,但除了演示示例之外,我没有任何运气能够得到任何工作。

使基于终端的客户端提供readline支持的最佳方法是什么?

http://twistedmatrix.com/documents/current/api/twisted.conch.stdio.html

看起来很有希望,但我很困惑如何使用它。

http://twistedmatrix.com/documents/current/api/twisted.conch.recvline.HistoricRecvLine.html

似乎也提供了对例如处理向上和向下箭头的支持,但我无法切换控制台继承自HistoricRecVLine而不是LineReceiver来运行。

也许扭曲是错误的框架使用或我应该使用所有海螺类。我只是喜欢它的事件驱动风格。是否有一种更好/更简单的方法可以在扭曲的客户端中使用readline或readline作为支持?

1 个答案:

答案 0 :(得分:0)

我通过不使用Twisted框架来解决这个问题。这是一个很棒的框架,但我认为这是一个错误的工具。相反,我使用了telnetlibcmdreadline模块。

我的服务器是异步的,但这并不意味着我的客户端需要这样,所以我使用telnetlib来与服务器进行通信。这样就可以轻松创建ConsoleClient类,其中包含cmd.Cmd子类并获取历史记录和类似emacs的快捷方式。

#! /usr/bin/env python

import telnetlib
import readline
import os
import sys
import atexit
import cmd
import string

HOST='127.0.0.1'
PORT='4118'

CONSOLE_PROMPT='console> '

class ConsoleClient(cmd.Cmd):
    """Simple Console Client in Python.  This allows for readline functionality."""

    def connect_to_console(self):
        """Can throw an IOError if telnet connection fails."""
        self.console = telnetlib.Telnet(HOST,PORT)
        sys.stdout.write(self.read_from_console())
        sys.stdout.flush()

    def read_from_console(self):
        """Read from console until prompt is found (no more data to read)
        Will throw EOFError if the console is closed.
        """
        read_data = self.console.read_until(CONSOLE_PROMPT)
        return self.strip_console_prompt(read_data)

    def strip_console_prompt(self,data_received):
        """Strip out the console prompt if present"""
        if data_received.startswith(CONSOLE_PROMPT):
            return data_received.partition(CONSOLE_PROMPT)[2]
        else:
            #The banner case when you first connect
            if data_received.endswith(CONSOLE_PROMPT):
                return data_received.partition(CONSOLE_PROMPT)[0]
            else:
                return data_received

    def run_console_command(self,line):
        self.write_to_console(line + '\n')
        data_recved = self.read_from_console()        
        sys.stdout.write(self.strip_console_prompt(data_recved))        
        sys.stdout.flush()

    def write_to_console(self,line):
        """Write data to the console"""
        self.console.write(line)
        sys.stdout.flush()

    def do_EOF(self, line): 
        try:
            self.console.write("quit\n")
            self.console.close()
        except IOError:
            pass
        return True

    def do_help(self,line):
        """The server already has it's own help command.  Use that"""
        self.run_console_command("help\n")

    def do_quit(self, line):        
        return self.do_EOF(line)

    def default(self, line):
        """Allow a command to be sent to the console."""
        self.run_console_command(line)

    def emptyline(self):
        """Don't send anything to console on empty line."""
        pass


def main():
    histfile = os.path.join(os.environ['HOME'], '.consolehistory') 
    try:
        readline.read_history_file(histfile) 
    except IOError:
        pass
    atexit.register(readline.write_history_file, histfile) 

    try:
        console_client = ConsoleClient()
        console_client.prompt = CONSOLE_PROMPT
        console_client.connect_to_console()
        doQuit = False;
        while doQuit != True:
            try:
                console_client.cmdloop()
                doQuit = True;
            except KeyboardInterrupt:
                #Allow for ^C (Ctrl-c)
                sys.stdout.write('\n')
    except IOError as e:
        print "I/O error({0}): {1}".format(e.errno, e.strerror)
    except EOFError:
        pass

if __name__ == '__main__':
    main()

我做的一项更改是删除从服务器返回的提示,并使用Cmd.prompt向用户显示。我的理由是支持Ctrl-c更像shell。