Acesso Python ao BasequestHandler

Question

Meu código basicamente precisa iniciar um servidor de bate -papo simples com um cliente. Onde o servidor e o cliente podem conversar um para o outro. Eu consegui tudo para ser implementado corretamente, mas não consigo descobrir como desligar o servidor sempre que terminar. (Eu sei que é ss.shutdown()).

Estou querendo terminar agora com base em uma palavra -chave compartilhada entre os dois (algo como "bye"), mas não sei se posso de alguma forma enviar uma mensagem para o meu SocketServer a partir de BaseRequestHandler para shutdown() sempre que recebe a mensagem.

Eventualmente, meu objetivo é incorporar Tkinter Para fazer uma GUI, mas eu queria que todo o resto funcionasse primeiro, e esta é a primeira vez que lido com soquetes em Python.

from sys import argv, stderr
from threading import Thread
import socket
import SocketServer
import threading
import sys

class ThreadedRecv(Thread):
    def __init__(self,socket):
        Thread.__init__(self)
        self.__socket = socket
        self.__message = ''
        self.__done = False
    def recv(self):
        while self.__message.strip() != "bye" and not self.getStatus():
            self.__message = self.__socket.recv(4096)
            print 'received',self.__message
        self.setStatus(True)

    def run(self):
        self.recv()

    def setStatus(self,status):
        self.__done = status

    def getStatus(self):
        return self.__done

class ThreadedSend(Thread):
    def __init__(self,socket):
        Thread.__init__(self)
        self.__socket = socket
        self.__message = ''
        self.__done = False
    def send(self):
        while self.__message != "bye" and not self.getStatus():
            self.__message = raw_input()
            self.__socket.send(self.__message)
        self.setStatus(True)

    def run(self):
        self.send()

    def setStatus(self,status):
        self.__done = status

    def getStatus(self):
        return self.__done



class HostException(Exception):
    def __init__(self, value):
        self.value = value
    def __str__(self):
        return repr(self.value)

class EchoServer(SocketServer.BaseRequestHandler):
    def setup(self):
        print self.client_address, 'is connected!'
        self.request.send('Hello ' + str(self.client_address) + '\n')
        self.__done = False
    def handle(self):

        sender = ThreadedSend(self.request)
        recver = ThreadedRecv(self.request)
        sender.start()
        recver.start()
        while 1:
            if recver.getStatus():
                sender.setStatus(True)
                break
            if sender.getStatus():
                recver.setStatus(True)
                break         

    def finish(self):
        print self.client_address, 'disconnected'
        self.request.send('bye client %s\n' % str(self.client_address))
        self.setDone(True)

    def setDone(self,done):
        self.__done = done

    def getDone(self):
        return self.__done



def setup(arg1, arg2, arg3):
    server = False
    defaultPort,defaultHost = 2358,"localhost"
    hosts = []
    port = defaultPort
    serverNames = ["TRUE","SERVER","S","YES"]
    arg1 = arg1.upper()
    arg2 = arg2.upper()
    arg3 = arg3.upper()
    if arg1 in serverNames or arg2 in serverNames or arg3 in serverNames:
    server = True
    try:
        port = int(arg1)
        if arg2 != '':
            hosts.append(arg2)
    except ValueError:
        if arg1 != '':
            hosts.append(arg1)
        try:
            port = int(arg2)
            if arg3 != '':
                hosts.append(arg3)
        except ValueError:
            if arg2 != '':
                hosts.append(arg2)
            try:
                port = int(arg3)
            except ValueError:
                if arg3 != '':
                    hosts.append(arg3)
                port = defaultPort

    for sn in serverNames:
        if sn in hosts:
            hosts.remove(sn)

    try:
        if len(hosts) != 1:
            raise HostException("Either more than one or no host "+ \
                                "declared.  Setting host to localhost.")
    except HostException as error:
        print error.value, "Setting hosts to default"
        return (server,defaultHost,port)

    return (server,hosts[0].lower(),port)

def main():
    bufsize = 4096
    while len(argv[1:4]) < 3:
        argv.append('')
    settings = setup(*argv[1:4])
    connections = (settings[1],settings[2])
    print connections
    if not settings[0]:
        try:
            mySocket = socket.socket(socket.AF_INET,\
                                     socket.SOCK_STREAM)
        except socket.error, msg:
            stderr.write("[ERROR] %s\n" % msg[1])
            sys.exit(1)
        try:
            mySocket.connect(connections)
        except socket.error, msg:
            stderr.write("[ERROR] %s\n" % msg[1])
            sys.exit(2)

        message = ""
        print "Enter a message to send to the server. "+\
              "Enter \"bye\" to quit."
        sender = ThreadedSend(mySocket)
        recver = ThreadedRecv(mySocket)
        sender.start()
        recver.start()
        while 1:
            if sender.getStatus():
                recver.setStatus(True)
                break
            if recver.getStatus():
                sender.setStatus(True)
                break    

    else:
        xserverhandler = EchoServer
        serversocket = SocketServer.ThreadedTCPServer(\
            connections,xserverhandler)
        server_thread = Thread(target = serversocket.serve_forever)
        server_thread.setDaemon(True)
        server_thread.start()
        # I would like to shut down this server whenever 
        # I get done talking to it.
        """while 1:
            if xserverhandler.getDone():
                print 'This is now true!'
                serversocket.shutdown()
                break"""

if __name__ == '__main__':
    main()

Sim, eu sei que o setup () é uma função terrível agora com as tentativas e capturas, mas funciona por enquanto, então eu iria consertá -lo mais tarde.

Minha pergunta é basicamente: como posso fazer com que o servidor termine com base em uma mensagem que ele recebe? Se possível, existe uma maneira de acessar o manipulador de solicitação após o início?

Answer 1

Por favor, corrija seu código para que ele funcione e inclua uma maneira de usá -lo. Você precisa adicionar

class ThreadedTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
    pass

Como o SocketServer não inclui essa classe (pelo menos não na minha versão do 2.6 nem 2.7). Em vez disso, é um exemplo do Definição do SocketServer.

Inclua um exemplo de como iniciar/usar o código. Nesse caso, para iniciar o servidor que você precisa fazer:

ss.py SERVER localhost 8001

e o cliente como

ss.py localhost 8001

Se você fizer isso, não poderá fazer server_thread.setDaemon (true) porque não há outros threads em execução, o que significa que o servidor sairá imediatamente.

Uma vez feito, a solução é adicionar uma chamada (ou dois) a self.server.shutdown () insdie do seu método Echoserver.Handle, como:

    while 1:
        if recver.getStatus():
            sender.setStatus(True)
            self.server.shutdown()
            break

No entanto, não consigo fazer isso funcionar, e acho que é porque herdei as coisas erradas ou adivinhei errado no que você fez.

O que você deve fazer é procurar por outra pessoa que tenha feito um servidor de bate -papo no Python. Usando o google eu encontrei http://www.slideshare.net/didip/socket-programming-in-python E certamente existem outros.

Além disso, se você quiser misturar a GUI e a programação encadeada, deve procurar exemplos com base nisso. Há vários hits quando procurei por "tknter bate -papo". Além disso, você pode querer examinar Twisted, que já resolveu muitos desses problemas.

Que problemas? Bem, por exemplo, você provavelmente deseja uma opção de soquete SO_ReUSEADDR.

Answer 2

O objeto de manipulador de solicitação é criado para cada nova solicitação. Portanto, você precisa armazenar o sinalizador "feito" no servidor, não o manipulador. Algo como o seguinte:

class EchoServer(SocketServer.BaseRequestHandler):
    ...
    def setDone(self):
        self.server.setDone() # or even better directly self.server.shutdown()