我正在尝试使用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作为支持?
答案 0 :(得分:0)
我通过不使用Twisted框架来解决这个问题。这是一个很棒的框架,但我认为这是一个错误的工具。相反,我使用了telnetlib
,cmd
和readline
模块。
我的服务器是异步的,但这并不意味着我的客户端需要这样,所以我使用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。