Limnoria/src/asyncoreDrivers.py

140 lines
4.5 KiB
Python
Raw Normal View History

2003-03-12 07:26:59 +01:00
###
# Copyright (c) 2002-2004, Jeremiah Fincher
2003-03-12 07:26:59 +01:00
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
2003-11-25 09:38:19 +01:00
__revision__ = "$Id$"
2004-07-24 07:18:26 +02:00
import supybot.fix as fix
2003-03-12 07:26:59 +01:00
import sys
import time
import socket
import asyncore
import asynchat
2004-07-24 07:18:26 +02:00
import supybot.conf as conf
import supybot.utils as utils
2004-07-24 07:18:26 +02:00
import supybot.world as world
import supybot.drivers as drivers
import supybot.schedule as schedule
2003-03-12 07:26:59 +01:00
class AsyncoreRunnerDriver(drivers.IrcDriver):
2004-08-31 22:19:48 +02:00
def name(self):
return self.__class__.__name__
2003-03-12 07:26:59 +01:00
def run(self):
try:
timeout = conf.supybot.drivers.poll()
2004-04-16 10:17:24 +02:00
if not asyncore.socket_map:
# asyncore should take care of this... but it doesn't?
time.sleep(timeout)
else:
asyncore.poll(timeout)
2003-03-12 07:26:59 +01:00
except:
drivers.log.exception('Uncaught exception:')
2003-03-12 07:26:59 +01:00
class AsyncoreDriver(asynchat.async_chat, drivers.ServersMixin):
2004-07-30 21:05:57 +02:00
def __init__(self, irc, servers=()):
2003-03-12 07:26:59 +01:00
asynchat.async_chat.__init__(self)
drivers.ServersMixin.__init__(self, irc, servers=servers)
self.irc = irc
self.irc.driver = self # Necessary because of the way we reconnect.
2003-03-12 07:26:59 +01:00
self.buffer = ''
2004-08-31 21:05:20 +02:00
self.scheduled = False
2003-03-12 07:26:59 +01:00
self.set_terminator('\n')
2003-04-03 08:56:22 +02:00
try:
2004-07-30 21:05:57 +02:00
server = self._getNextServer()
2004-07-31 02:46:26 +02:00
sock = utils.getSocket(server[0])
self.set_socket(sock)
drivers.log.connect(self.currentServer)
2004-07-30 21:05:57 +02:00
self.connect(server)
2004-01-22 00:35:34 +01:00
except socket.error, e:
drivers.log.connectError(self.currentServer, e)
2004-01-22 00:35:34 +01:00
self.reconnect(wait=True)
2003-04-03 08:56:22 +02:00
2004-07-24 04:29:06 +02:00
def _scheduleReconnect(self, at=60):
when = time.time() + at
if not world.dying:
drivers.log.reconnect(self.irc.network, when)
def makeNewDriver():
self.irc.reset()
self.scheduled = None
2004-07-30 21:05:57 +02:00
driver = self.__class__(self.irc, servers=self.servers)
self.scheduled = schedule.addEvent(makeNewDriver, when)
2003-03-12 07:26:59 +01:00
def writable(self):
2003-04-09 17:13:51 +02:00
while self.connected:
m = self.irc.takeMsg()
2003-03-12 07:26:59 +01:00
if m:
self.push(str(m))
else:
break
return asynchat.async_chat.writable(self)
def handle_error(self):
self.handle_close()
2003-03-12 07:26:59 +01:00
def collect_incoming_data(self, s):
self.buffer += s
def found_terminator(self):
2003-04-15 16:36:58 +02:00
start = time.time()
msg = drivers.parseMsg(self.buffer)
2003-04-15 16:36:58 +02:00
self.buffer = ''
2004-08-23 15:41:04 +02:00
if msg is not None:
self.irc.feedMsg(msg)
2003-03-12 07:26:59 +01:00
2004-08-25 07:07:23 +02:00
def handle_close(self, wait=True):
if not wait:
self._scheduleReconnect(at=0)
else:
self._scheduleReconnect()
if self.socket is not None:
self.close()
2003-04-21 06:32:42 +02:00
reconnect = handle_close
def handle_connect(self):
pass
2003-04-03 10:52:41 +02:00
def die(self):
if self.scheduled:
schedule.removeEvent(self.scheduled)
drivers.log.die(self.irc)
2004-08-25 07:07:23 +02:00
self.reconnect()
2003-03-12 07:26:59 +01:00
try:
ignore(poller)
except NameError:
poller = AsyncoreRunnerDriver()
2003-04-20 09:15:39 +02:00
Driver = AsyncoreDriver
2003-03-12 07:26:59 +01:00
2004-01-18 08:58:26 +01:00
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: