Wie kann ich einen Python-Thread-Socket-Server mit einer anderen Klasse initialisieren?

Question

Ich schreibe einen Python IRC-Bot, und ich möchte einen TCP-Server hinzufügen, so dass der Bot Nachrichten in IRC-Kanälen gesendet Echo kann. Ich benutze das Python-Socketserver-Modul, so dass es sowohl Linux als auch Freebsd unterstützt (ich benutze beide). Ich habe eine kurze Testskript hier:

#! /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() 

Was ich möchte in der Lage sein, todo ist der IRC-Bot TCP-Server starten müssen, und haben die TCP-Server alle Daten senden Sie es an den Bot erhält.

Der vollständige (ohne Funktion) Bot-Code ist hier: https://github.com/meskarune/autobot/blob/master/src/autobot.py

Unten finden Sie eine verkürzte Version mit den entsprechenden Stücken:

#!/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() 

Mit dem aktuellen Code der Bot beim Laufen hängt und wenn ich Do Ctrl-C verbindet und stürzt dann ab. Ich würde wirklich gerne wissen, wie ich das funktionieren lassen kann. Vielen Dank.

Answer 1

Innerhalb TCPinput, Sie rufen serve_forever innerhalb der Haupt-Thread der Steuerung, wodurch es blockiert. Stattdessen müssen Sie den Server in einem separaten Thread ausführen. Sie können dies erreichen, indem Sie TCPinput von threading.Thread ableiten, serve_forever als das Thread-Ziel beim Initialisieren des Objekts festlegen. Dann, während Sie den Bot initialisieren, können Sie Thread.start() diesen Thread.