Limnoria/src/world.py

186 lines
6.3 KiB
Python
Raw Normal View History

2003-03-12 07:26:59 +01:00
#!/usr/bin/env python
###
# 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-10-04 14:53:09 +02:00
"""
Module for general worldly stuff, like global variables and whatnot.
"""
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 gc
2003-11-02 20:00:15 +01:00
import os
import sys
2003-11-02 20:00:15 +01:00
import sre
2003-03-12 07:26:59 +01:00
import time
import atexit
import threading
2003-03-12 07:26:59 +01:00
2004-07-24 07:18:26 +02:00
import supybot.log as log
import supybot.conf as conf
import supybot.drivers as drivers
import supybot.ircutils as ircutils
import supybot.registry as registry
import supybot.schedule as schedule
2003-03-12 07:26:59 +01:00
2003-12-03 23:17:12 +01:00
startedAt = time.time() # Just in case it doesn't get set later.
2003-04-08 09:23:56 +02:00
starting = False
mainThread = threading.currentThread()
2004-01-15 13:16:55 +01:00
assert 'MainThread' in repr(mainThread)
threadsSpawned = 1 # Starts at one for the initial "thread."
commandsProcessed = 0
2003-03-12 07:26:59 +01:00
2003-12-03 23:17:12 +01:00
ircs = [] # A list of all the IRCs.
2003-03-12 07:26:59 +01:00
2004-04-30 20:24:35 +02:00
def _flushUserData():
userdataFilename = os.path.join(conf.supybot.directories.conf(),
'userdata.conf')
registry.close(conf.users, userdataFilename, annotated=False)
2004-04-30 20:24:35 +02:00
flushers = [_flushUserData] # A periodic function will flush all these.
2003-03-12 07:26:59 +01:00
registryFilename = None
2003-03-12 07:26:59 +01:00
def flush():
2003-10-04 14:53:09 +02:00
"""Flushes all the registered flushers."""
for (i, f) in enumerate(flushers):
try:
f()
except Exception, e:
log.exception('Uncaught exception in flusher #%s (%s):', i, f)
2003-03-12 07:26:59 +01:00
def debugFlush(s=''):
if conf.supybot.debug.flushVeryOften():
if s:
log.debug(s)
flush()
def upkeep(scheduleNext=True):
2003-10-04 14:53:09 +02:00
"""Does upkeep (like flushing, garbage collection, etc.)"""
sys.exc_clear() # Just in case, let's clear the exception info.
if os.name == 'nt':
try:
import msvcrt
msvcrt.heapmin()
except ImportError:
pass
except IOError: # Win98 sux0rs!
pass
if conf.daemonized:
# If we're daemonized, sys.stdout has been replaced with a StringIO
# object, so let's see if anything's been printed, and if so, let's
# log.warning it (things shouldn't be printed, and we're more likely
# to get bug reports if we make it a warning).
assert not type(sys.stdout) == file, 'Not a StringIO object!'
s = sys.stdout.getvalue()
if s:
log.warning('Printed to stdout after daemonization: %s', s)
2004-04-20 12:40:27 +02:00
sys.stdout.reset() # Seeks to 0.
sys.stdout.truncate() # Truncates to current offset.
assert not type(sys.stderr) == file, 'Not a StringIO object!'
s = sys.stderr.getvalue()
if s:
log.error('Printed to stderr after daemonization: %s', s)
sys.stderr.reset() # Seeks to 0.
sys.stderr.truncate() # Truncates to current offset.
flushed = conf.supybot.flush() and not starting
if flushed:
2003-03-12 07:26:59 +01:00
flush()
# This is so registry._cache gets filled.
if registryFilename is not None:
registry.open(registryFilename)
if not dying:
log.debug('Regexp cache size: %s', len(sre._cache))
log.debug('Pattern cache size: %s'%len(ircutils._patternCache))
log.debug('HostmaskPatternEqual cache size: %s' %
len(ircutils._hostmaskPatternEqualCache))
2004-08-07 02:58:45 +02:00
#timestamp = log.timestamp()
if flushed:
2004-08-07 02:58:45 +02:00
log.info('Flushers flushed and garbage collected.')
else:
2004-08-07 02:58:45 +02:00
log.info('Garbage collected.')
if scheduleNext:
schedule.addEvent(upkeep, time.time() + conf.supybot.upkeepInterval())
2004-05-07 13:41:45 +02:00
collected = gc.collect()
if gc.garbage:
2004-08-19 01:15:27 +02:00
log.warning('Noncollectable garbage (file this as a bug on SF.net): %s',
2004-05-07 13:41:45 +02:00
gc.garbage)
return collected
2003-03-12 07:26:59 +01:00
2004-01-15 13:55:19 +01:00
def makeDriversDie():
"""Kills drivers."""
log.info('Killing Driver objects.')
for driver in drivers._drivers.itervalues():
driver.die()
def makeIrcsDie():
2003-12-03 23:17:12 +01:00
"""Kills Ircs."""
2003-12-16 22:06:20 +01:00
log.info('Killing Irc objects.')
for irc in ircs[:]:
irc.die()
2003-09-12 09:48:01 +02:00
def startDying():
2003-12-03 23:17:12 +01:00
"""Starts dying."""
2003-12-16 22:06:20 +01:00
log.info('Shutdown initiated.')
global dying
dying = True
2003-12-16 22:06:20 +01:00
def finished():
log.info('Shutdown complete.')
2004-08-29 06:32:07 +02:00
# These are in order; don't reorder them for cosmetic purposes. The order
# in which they're registered is the reverse order in which they will run.
2003-12-16 22:06:20 +01:00
atexit.register(finished)
atexit.register(upkeep)
atexit.register(makeIrcsDie)
2004-01-15 13:55:19 +01:00
atexit.register(makeDriversDie)
atexit.register(startDying)
##################################################
##################################################
##################################################
## Don't even *think* about messing with these. ##
##################################################
##################################################
##################################################
dying = False
testing = False
starting = False
profiling = False
documenting = False
2003-03-26 00:42:37 +01:00
2004-01-02 23:29:03 +01:00
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: