ircbot.py

# -*- coding: utf8 -*-
import settings

import sys
import socket
import threading
import functools
from message_handler import MessageHandler
from irc_messages import JoinMessage

import logging
logging.basicConfig()
logger = logging.getLogger('ircbot')
logger.setLevel(logging.DEBUG)

class IrcBot:
    def __init__(self):
        self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def connect(self, host, port, ircname):
        """ Function that connects bot to irc server and starts listening events"""
        def login_to_server():
            logger.info("Log in to server with username: {user}".format(user=self._ircname))
            self._send_to_server("NICK {nick}".format(nick=self._ircname))
            self._send_to_server("USER {nick} 8 * : Yet another Ircbot".format(nick=self._ircname))

        self._ircname = ircname
        logger.info("connecting to %s:%s" % (host, port))
        self._socket.connect((host, port))
        logger.info("connected")

        login_to_server()

        self._response_loop()

    def _response_loop(self):
        def get_server_response():
            response = ''
            while '\r\n' not in response:
                response += self._socket.recv(1)

            return response

        while True:
            server_response = get_server_response()
            handle_thread = threading.Thread(target=functools.partial(self._handle_server_response, server_response))
            handle_thread.start()

    def _send_to_server(self, data):
        self._socket.sendall(data + '\r\n')
        logger.debug("Sent data: {data}".format(data=data))

    def _handle_server_response(self, response):
        """
        Handles server response. Passes raw response from server to MessageHandler and except that MessageHandler.handle returns instance of IrcEvent class, so ircbot knows what to do next with the event.
        """
        message_handler = MessageHandler()
        event = message_handler.handle(response)
        if event and event.to_server:
            for irc_message in event.to_server:
                self._send_to_server(irc_message.get_command())

if __name__ == '__main__':
    ircbot = IrcBot()
    try:
        host = getattr(settings, 'SERVER')
        port = int(getattr(settings, 'PORT'))
        ircname = getattr(settings, 'IRCNAME')

        ircbot.connect(host, port, ircname)
    except AttributeError:
        logger.error('SERVER NOT CONFIGURED!')
        sys.exit()