python twisted TCP通信 脚本

-------------------server-------------------

!/usr/bin/python

coding=utf-8

from twisted.internet.protocol import Protocol
from twisted.internet.protocol import Factory
from twisted.internet.endpoints import TCP4ServerEndpoint
from twisted.internet import reactor

clients=[]

class Spreader(Protocol):

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

def connectionMade(self):
self.factory.numProtocols+=1
self.transport.write("欢迎,你是第{}客户".format(self.factory.numProtocols).encode("utf-8"))
clients.append(self)

def connectionLost(self,reason):
self.factory.numProtocols-=1
clients.remove(self)
print("lost connect: %d" % (self.factory.numProtocols))

def dataReceived(self,data):
if data=="close":
self.transport.lostConnection()
for client in clients:
if client!=self:
client.transport.write(data.encode("utf-8"))
else:
print(data)

class SpreadFactory(Factory):

def init(self):
self.numProtocols=0

def buildProtocol(self,addr):
return Spreader(self)

endpoint=TCP4ServerEndpoint(reactor,8007)
endpoint.listen(SpreadFactory())
reactor.run()

-------------------------------------------------client------------------------

!/usr/bin/python

coding=utf-8

from twisted.internet.protocol import Protocol, ClientFactory
from twisted.internet import reactor
import threading
import time
import sys
import datetime

class Echo(Protocol):
def init(self):
self.connected = False

def connectionMade(self):
    self.connected = True

def connectionLost(self, reason):
    self.connected = False

def dataReceived(self, data):
    print(data.decode("utf-8"))

class EchoClientFactory(ClientFactory):
def init(self):
self.protocol = None

def startedConnecting(self, connector):
    print("Start to Connect...")

def buildProtocol(self, addr):
    print("Connected...")
    self.protocol = Echo()
    return self.protocol

def clientConnectionLost(self, connector, reason):
    print("Lost connection. Reason: ", reason)

def clientConnectionFailed(self, connector, reason):
    print("Connection is failed, Reason: ", reason)

bStop = False

def routine(factory):
while not bStop:
if factory.protocol and factory.protocol.connected:
factory.protocol.transport.write("hello, I'm {} {} ".format(
sys.argv[0], datetime.datetime.now()
).encode("utf-8"))
print(sys.argv[0], datetime.datetime.now())
time.sleep(5)

host = '127.0.0.1'
port = 8007
factory = EchoClientFactory()
reactor.connectTCP(host, port, factory)
threading.Thread(target=routine, args=(factory,)).start()
reactor.run()
bStop = True