Python与asynchat聊天获取会话数据

时间:2013-04-18 12:51:06

标签: python sockets asyncore

我有来自here的代码:

#!/usr/bin/python

from asyncore import dispatcher
from asynchat import async_chat
import socket, asyncore

PORT = 55555
NAME = 'ChatLine'

class ChatSession(async_chat):
    def __init__(self,server,sock):
        async_chat.__init__(self, sock)
        self.server = server
        self.set_terminator("\r\n")
        self.data = []

    def collect_incoming_data(self, data):
        self.data.append(data)

    def found_terminator(self):
        line = "".join(self.data)
        self.data = []
        self.server.broadcast(line)

    def handle_close(self):
        async_chat.handle_close(self)
        self.server.disconnect(self)

class ChatServer(dispatcher):
    def __init__(self, port, name):
        dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.set_reuse_addr()
        self.bind(("",port))
        self.listen(5)
        self.name = name
        self.sessions = []

    def disconnect(self, sessions):
        self.sessions.remove(session)

    def broadcast(self, line):
        for session in self.sessions:
            session.push('>>' + line + '\r\n')

    def handle_accept(self):
        conn, addr = self.accept()
        print "Connected from " + str(addr)
        self.sessions.append(ChatSession(self, conn))

if __name__ == '__main__':
    s = ChatServer(PORT, NAME)
    try: asyncore.loop()
    except KeyboardInterrupt: print

如果我使用putty / telnet或任何其他软件连接它,我得到这个输出:

test
>>test

第一个test是我发送的内容,第二个>>>test是服务器广播的内容。

如何删除我发送的同一邮件的广播,因此我不会获得第二个>>>test?也许我可以注册会话ID或任何其他数据,当服务器广播消息时,我可以将数据发送到除以前记录的会话之外的所有会话?

或者我可以添加用户名并使用它来删除重复的消息吗?

我没有找到任何关于asyncore或asychat的教程或文档,所以任何帮助都将不胜感激。

1 个答案:

答案 0 :(得分:2)

sessions上的ChatServer列表包含代表服务器连接的所有ChatSession个实例。

这包括向服务器发送特定消息的ChatSession。如果要将该消息发送到所有其他连接,则只需跳过ChatSession内循环中的broadcast个实例之一。要知道要跳过哪一个,可以在广播中添加一个参数:

def broadcast(self, line, skip):

然后在循环中使用它 not 调用push一次:

    for session in self.sessions:
        if session is skip:
            session.push('Alternate pedagogical output\r\n')
        else:
            session.push('>>' + line + '\r\n')

这里我实际上没有跳过会话,只是给了它不同的处理方式,这样你就可以轻松地告诉它发生了不同的事情(而不是看到没有输出,这也可能是由其他一些bug造成的......)

现在您只需更改调用者即可传递此参数。幸运的是,调用者是ChatSession的一种方法,可以轻松获得正确的ChatSession跳过 - self

所以只需将found_terminate的{​​{1}}方法更改为......

ChatSession

正如您所看到的,这并没有真正利用任何asyncore或asynchat功能。它只是应用程序代码中的一些逻辑。

顺便提一下,Twisted比asynchat更具特色 - 所以我建议你花时间学习它。为了品味,这是用Twisted编写的聊天服务器:

def found_terminator(self):
    line = "".join(self.data)
    self.data = []
    self.server.broadcast(line, skip=self)