Python 2.6 聊天循环问题。不能同时接收和发送

Question

我正在尝试制作一个控制台聊天程序，但我的循环有问题。我无法同时获得输入并接收其他人的输入。如果从一端发送了两条或更多条消息，则另一端在发送一条消息之前无法接收下一条消息。我对 python 相当陌生，并且正在寻找正确的方向。我想到了多线程，但这有点超出我的掌握。还有其他想法吗？

import EncMod
from socket import *

#Get User Info
Ip = raw_input('IP>>>')
Port = int(raw_input('Port>>>'))
User = raw_input('Username>>>')

#Open Socket To Server
EncCon = socket(AF_INET, SOCK_STREAM)
EncCon.connect((Ip, Port))

print '\nStarting Chat....'
print '\n<-------------------------------------------->\n\n'

#Send/Receive Loop
while 1:
   MsgOut = raw_input()
   if MsgOut: EncCon.send(MsgOut)

   MsgIn = EncCon.recv(1024)
   if MsgIn: print MsgIn

EncCon.close()

Answer 1

Twisted framework 可用于帮助完成此任务。下面的代码会启动一个聊天服务器，然后客户端可以连接到该服务器并根据服务器实例设置来回通信。您可以进行适当的修改以满足您的要求：

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

class Chat(LineReceiver):

    def __init__(self, users):
        self.users = users
        self.name = None
        self.state = "GETNAME"

    def connectionMade(self):
        self.sendLine("What's your name?")

    def connectionLost(self, reason):
        if self.users.has_key(self.name):
            del self.users[self.name]

    def lineReceived(self, line):
        if self.state == "GETNAME":
            self.handle_GETNAME(line)
        else:
            self.handle_CHAT(line)

    def handle_GETNAME(self, name):
        if self.users.has_key(name):
            self.sendLine("Name taken, please choose another.")
            return
        self.sendLine("Welcome, %s!" % (name,))
        self.name = name
        self.users[name] = self
        self.state = "CHAT"

    def handle_CHAT(self, message):
        message = "<%s> %s" % (self.name, message)
        for name, protocol in self.users.iteritems():
            if ':' in message:
                self.exc(message.split(':')[0])
            if protocol != self:
                protocol.sendLine(message)

    def exc(self, cmd):
        print cmd
        if cmd == 'who':
            for i in self.users:
                print i


class ChatFactory(Factory):

    def __init__(self):
        self.users = {} # maps user names to Chat instances

    def buildProtocol(self, addr):
        return Chat(self.users)


reactor.listenTCP(8123, ChatFactory())
reactor.run()

Answer 2

问题是您的 recv() 调用会阻塞，直到接收到一些数据，而当 recv() 阻塞时，您的程序不会检查是否有任何来自标准输入的输入。传统的单线程解决方案是将套接字设置为非阻塞 I/O（通过 EncCon.setblocking(False)），然后将程序块放在 select() 中。将 EncCon 和 stdin 都传递给 select() （作为其 read-socket-set 参数的一部分），这样只要它们中的任何一个有一些数据要给你，select() 就会返回。 （请注意，此方法在 Windows 下不起作用，因为 Windows 不允许 select() 阻塞 stdin :P ）

Answer 3

线程并不像您想象的那么难，掌握它是您工具箱的宝贵补充。

只需创建一个作为 Thread 子类的类并确保它具有 run() 方法。然后实例化该类并调用它的 start() 方法。

使线程停止更难正确处理。最好设置一个标志并确保在您的 while 循环中定期检查它，因此您需要阻塞 recv() 的超时时间，例如 1 秒。

from socket import *
from threading import Thread


#Get User Info
Ip = raw_input('IP>>>')
Port = int(raw_input('Port>>>'))
User = raw_input('Username>>>')

#Open Socket To Server
EncCon = socket(AF_INET, SOCK_STREAM)
EncCon.connect((Ip, Port))

print '\nStarting Chat....'
print '\n<-------------------------------------------->\n\n'


class ReceiveThread(Thread):

    def __init__(self, sock):
        Thread.__init__(self)
        self.sock = sock
        self.shouldstop = False

    def run(self):
        self.sock.settimeout(1)
        while not self.shouldstop:
            try:
                data = self.sock.read()
                print data
            except socket.timeout:
                continue

    def stop(self):
        self.shouldstop = True


# start receive loop:
r = ReceiveThread(EncCon).start()


#Send Loop
while 1:
    MsgOut = raw_input()
    if MsgOut: EncCon.send(MsgOut)

    if MsgOut == '.':
        r.stop()
        r.join()


EncCon.close()

现在，该程序仍然存在无法启动两个实例的原始问题，因为您不听，而是立即连接。但是，我相信这不是您问题的主要部分。