Acesso Python ao BasequestHandler
-
21-09-2019 - |
Pergunta
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?
Solução
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.
Outras dicas
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()