Twisted聊天服务器

如意IT

于 2019-01-24 15:32:51 发布

阅读量301

点赞数 1

分类专栏： Twisted

Twisted 专栏收录该内容

2 篇文章

订阅专栏

本文介绍使用Twisted框架实现一个简单的聊天服务器的过程。通过创建ChatProtocol类继承自LineReceiver，实现用户注册、消息广播等功能。文章详细展示了如何利用Twisted的网络编程特性，构建一个多用户在线聊天应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from twisted.internet.protocol import Factory,connectionDone
from twisted.protocols.basic import LineReceiver
from twisted.internet import reactor

class ChatProtocol(LineReceiver):
    def __init__(self,factory):
        self.factory=factory
        self.name=None
        self.state="REGISTER"
    def connectionMade(self):
        self.sendLine("What is your name?".encode('ascii'))
    def connectionLost(self, reason=connectionDone):
        if self.name in self.factory.users:
            del self.factory.users[self.name]
            self.broadcastMessage(
                "%s has left the channel"%(self.name,)
            )
    def lineReceived(self,line):
        if self.state=='REGISTER':
            self.handle_REGISTER(line)
        else:
            self.handle_CHAT(line)

    def handle_REGISTER(self,name):
        if name in self.factory.users:
            self.sendLine("Name taken,please choose another".encode('ascii'))
            return
        self.sendLine("Welcome %s"%name)
        self.broadcastMessage("%s has joined the channel!"%name)
        self.name=name
        self.factory.users[name]=self
        self.state='CHAT'
    def handle_CHAT(self,msg):
        msg="%s %s"%(self.name,msg)
        self.broadcastMessage(msg)
    def boradcastMessage(self,message):
        for name,protocol in self.factory.users:
            if protocol!=self:
                protocol.sendline(message)

class ChatFctory(Factory):
    def __init__(self):
        self.users={}
    def buildProtocol(self, addr):
        return ChatProtocol(self)
reactor.listenTCP(8000,ChatFctory())
reactor.run()