Limnoria/src/asyncoreDrivers.py

150 lines
4.9 KiB
Python
Raw Normal View History

2003-03-12 07:26:59 +01:00
#!/usr/bin/env python
###
# Copyright (c) 2002, Jeremiah Fincher
# 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 re
import sys
import time
import socket
import asyncore
import asynchat
2004-07-24 07:18:26 +02:00
import supybot.log as log
import supybot.conf as conf
import supybot.ircdb as ircdb
import supybot.world as world
import supybot.drivers as drivers
import supybot.ircmsgs as ircmsgs
import supybot.schedule as schedule
2003-03-12 07:26:59 +01:00
class AsyncoreRunnerDriver(drivers.IrcDriver):
def run(self):
2004-02-16 09:10:50 +01:00
#log.debug(repr(asyncore.socket_map))
2003-03-12 07:26:59 +01:00
try:
timeout = conf.supybot.drivers.poll()
2004-04-16 10:17:24 +02:00
if not asyncore.socket_map:
# FIXME: 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:
log.exception('Uncaught exception:')
2003-03-12 07:26:59 +01:00
class AsyncoreDriver(asynchat.async_chat, object):
2004-07-30 08:52:21 +02:00
def __init__(self, irc):
2003-03-12 07:26:59 +01:00
asynchat.async_chat.__init__(self)
self.irc = irc
2003-03-12 07:26:59 +01:00
self.buffer = ''
2004-07-30 08:52:21 +02:00
self.servers = ()
self.networkGroup = conf.supybot.networks.get(self.irc.network)
2003-03-12 07:26:59 +01:00
self.set_terminator('\n')
2004-07-30 08:52:21 +02:00
# XXX: Use utils.getSocket.
2003-03-12 07:26:59 +01:00
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
2003-04-03 08:56:22 +02:00
try:
2004-07-30 08:52:21 +02:00
self.connect(self._getNextServer())
2004-01-22 00:35:34 +01:00
except socket.error, e:
2004-07-30 08:52:21 +02:00
log.warning('Error connecting to %s: %s', 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-30 08:52:21 +02:00
def _getServers(self):
# We do this, rather than itertools.cycle the servers in __init__,
# because otherwise registry updates given as setValues or sets
# wouldn't be visible until a restart.
return self.networkGroup.servers()[:] # Be sure to copy!
def _getNextServer(self):
if not self.servers:
self.servers = self._getServers()
assert self.servers, 'Servers value for %s is empty.' % \
self.networkGroup.name
server = self.servers.pop(0)
self.currentServer = '%s:%s' % server
return server
2004-07-24 04:29:06 +02:00
def _scheduleReconnect(self, at=60):
when = time.time() + at
if not world.dying:
whenS = log.timestamp(when)
2004-07-30 08:52:21 +02:00
log.info('Scheduling reconnect to %s at %s',
self.currentServer, whenS)
def makeNewDriver():
self.irc.reset()
2004-07-30 08:52:21 +02:00
driver = self.__class__(self.irc)
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()
2003-03-12 07:26:59 +01:00
msg = ircmsgs.IrcMsg(self.buffer)
2004-02-16 09:10:50 +01:00
#log.debug('Time to parse IrcMsg: %s', time.time()-start)
2003-04-15 16:36:58 +02:00
self.buffer = ''
self.irc.feedMsg(msg)
2003-03-12 07:26:59 +01:00
def handle_close(self):
self._scheduleReconnect()
2003-04-03 10:52:41 +02:00
self.die()
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):
2004-01-04 14:53:33 +01:00
log.info('Driver for %s dying.', self.irc)
2003-03-12 07:26:59 +01:00
self.close()
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: