如何用另一個類來初始化一個python線程的socketserver？

Question

我正在寫一個python irc bot，並且我想添加一個tcp服務器，以便bot能夠在irc通道中迴應發送給它的消息。我使用python socketserver模塊，以便它支持linux和freebsd（我都使用）。我有一個簡短的測試腳本，在這裏：

#! /usr/bin/env python3 
# -*- coding: utf-8 -*- 

import socketserver 
from threading import Thread 


class announce(): 
    """ Return message uppercase """ 
    def uppercase(message): 
     print (message.upper()) 

class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler): 
    """ Echo data back in uppercase """ 
    def handle(self): 
     self.announce = announce 
     data = str(self.request.recv(1024), 'utf-8') 
     if data is not None: 
      self.announce.uppercase(data) 
      self.request.send(bytes("message recieved", 'utf-8')) 
     self.request.close() 

class ThreadedTCPServer(socketserver.ThreadingMixIn,  socketserver.TCPServer): 
    daemon_threads = True 
    allow_reuse_address = True 

    def __init__(self, server_address, RequestHandlerClass): 
     socketserver.TCPServer.__init__(self, server_address, RequestHandlerClass) 

if __name__ == "__main__": 
    HOST = "localhost" 
    PORT = 2000 
    server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler) 
    try: 
     server.serve_forever() 
    except KeyboardInterrupt: 
     server.shutdown() 
     server.server_close() 

我希望能夠待辦事項什麼是有IRC機器人啓動TCP服務器，並具備TCP服務器發送接收到機器人的任何數據。

全（非工作）機器人代碼是在這裏：https://github.com/meskarune/autobot/blob/master/src/autobot.py

下面是截斷版本與相關作品：

#!/usr/bin/env python 
# -*- coding: utf-8 -*- 

class AutoBot(irc.bot.SingleServerIRCBot): 
    """Create the single server irc bot""" 
    def __init__(self): 
     self.config = configparser.ConfigParser() 
     self.config.read("autobot.conf") 

     self.nick = self.config.get("irc", "nick") 
     self.nickpass = self.config.get("irc", "nickpass") 
     self.name = self.config.get("irc", "name") 
     self.network = self.config.get("irc", "network") 
     self.port = int(self.config.get("irc", "port")) 
     self._ssl = self.config.getboolean("irc", "ssl") 
     self.channel_list = [channel.strip() for channel in self.config.get("irc", "channels").split(",")] 
     self.prefix = self.config.get("bot", "prefix") 

     if self._ssl: 
      factory = irc.connection.Factory(wrapper=ssl.wrap_socket) 
     else: 
      factory = irc.connectionFactory() 
     try: 
      irc.bot.SingleServerIRCBot.__init__(self, [(self.network, self.port)], 
               self.nick, self.name, 
               reconnection_interval=120, 
               connect_factory = factory) 
     except irc.client.ServerConnectionError: 
      sys.stderr.write(sys.exc_info()[1]) 

     #Listen for data to announce to channels 
     self.listenhost = self.config.get("tcp", "host") 
     self.listenport = int(self.config.get("tcp", "port")) 
     TCPinput(self.connection, self, self.listenhost, self.listenport) 

    def announce(self, connection, text): 
     """Send notice to joined channels""" 
     for channel in self.channel_list: 
      self.connection.notice(channel, text) 
      self.log_message(channel, "-!-", "(notice) {0}: {1}" 
          .format(self.connection.get_nickname(), text)) 
class TCPinput(): 
    """Listen for data on a port and send it to Autobot.announce""" 
    def __init__(self, connection, AutoBot, listenhost, listenport): 
     self.connection = connection 
     self.AutoBot = AutoBot 
     self.listenhost = listenhost 
     self.listenport = listenport 
     server = ThreadedTCPServer((self.listenhost, self.listenport), ThreadedTCPRequestHandler) 
     try: 
      server.serve_forever() 
     except: 
      server.shutdown() 
      server.server_close() 
    def send(self, message): 
     self.AutoBot.announce(self.connection, message.strip()) 

class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler): 
    """ Echo data back in uppercase """ 
    def handle(self): 
     self.TCPinput = TCPinput 
     data = str(self.request.recv(1024), 'utf-8') 
     if data is not None: 
      self.TCPinput.send(data.strip()) 
     self.request.close() 

class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer): 
    daemon_threads = True 
    allow_reuse_address = True 

    def __init__(self, server_address, RequestHandlerClass): 
     socketserver.TCPServer.__init__(self, server_address, RequestHandlerClass) 

def main(): 
    bot = AutoBot() 
    bot.start() 

if __name__ == "__main__": 
    main() 

隨着當前代碼機器人掛一旦運行，當我做ctrl-c它連接，然後崩潰。我真的很想知道如何才能使這項工作。謝謝。

Answer 1

在TCPinput裏面，你在調用主線程中調用serve_forever，導致它被阻塞。相反，您需要在單獨的線程中運行服務器。您可以通過從threading.Thread推導TCPinput，在初始化對象時將serve_forever設置爲線程目標來實現此目的。然後，在初始化機器人的同時，你可以使用Thread.start()那個線程。