2016-07-01 5 views
1

私はpython ircボットを書いています。ボットがircチャネルで送信したメッセージをエコーできるように、私はtcpサーバを追加したいと思います。私はPythonソケットサーバモジュールを使用しているので、linuxとfreebsdの両方をサポートしています(私は両方を使用しています)。私はここで短いテストスクリプトを持っている:どのように私は別のクラスでPythonのスレッドソケットサーバーを初期化できますか?

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

私ができるTODOようにしたいことは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() 

ボットは、ランニング時にハングアップしてたときにI ctrl-cを実行して接続し、クラッシュします。私はこの仕事をどうやって得ることができるかを本当に知りたいです。ありがとう。

答えて

1

TCPinputの中には、コントロールのメインスレッド内でserve_foreverが呼び出され、ブロックされています。代わりに、別のスレッド内でサーバーを実行する必要があります。これを達成するには、からTCPinputを派生させ、オブジェクトを初期化するときにスレッドのターゲットとしてserve_foreverを設定します。その後、ボットを初期化している間に、あなたはThread.start()そのスレッドにできます。

+0

ありがとうございました – Dolores

関連する問題