mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-12-25 04:02:46 +01:00
220 lines
7.4 KiB
Python
Executable File
220 lines
7.4 KiB
Python
Executable File
#!/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.
|
|
###
|
|
|
|
"""
|
|
This is the template for bots. supybot-wizard.py uses this file to make
|
|
customized startup files for bots.
|
|
"""
|
|
|
|
import re
|
|
import os
|
|
import sys
|
|
|
|
if sys.version_info < (2, 3, 0):
|
|
sys.stderr.write('This program requires Python >= 2.3.0\n')
|
|
sys.exit(-1)
|
|
|
|
if os.name == 'posix':
|
|
if os.getuid() == 0 or os.geteuid() == 0:
|
|
sys.stderr.write('Dude, don\'t even try to run this as root.\n')
|
|
sys.exit(-1)
|
|
|
|
import time
|
|
import optparse
|
|
|
|
started = time.time()
|
|
|
|
import supybot
|
|
|
|
import conf
|
|
|
|
defaultNick = "%%nick%%"
|
|
defaultUser = "%%user%%"
|
|
defaultIdent = "%%ident%%"
|
|
defaultServer = "%%server%%"
|
|
defaultPassword = "%%password%%"
|
|
|
|
conf.commandsOnStart = "%%onStart%%"
|
|
|
|
afterConnect = "%%afterConnect%%"
|
|
|
|
debugVariables = "%%debugVariables%%"
|
|
configVariables = "%%configVariables%%"
|
|
|
|
if not isinstance(configVariables, basestring):
|
|
for (name, value) in configVariables.iteritems():
|
|
setattr(conf, name, value)
|
|
|
|
|
|
def main():
|
|
import debug
|
|
|
|
if not isinstance(debugVariables, basestring):
|
|
for (name, value) in debugVariables.iteritems():
|
|
setattr(debug, name, value)
|
|
|
|
import world
|
|
import drivers
|
|
import schedule
|
|
schedule.addPeriodicEvent(world.upkeep, 300)
|
|
world.startedAt = started
|
|
try:
|
|
while world.ircs:
|
|
drivers.run()
|
|
except:
|
|
try:
|
|
debug.recoverableException()
|
|
except: # It must've been deadly for a reason :)
|
|
sys.exit(0)
|
|
|
|
if __name__ == '__main__':
|
|
###
|
|
# Options:
|
|
# -p (profiling)
|
|
# -O (optimizing)
|
|
# -n, --nick (nick)
|
|
# -s, --server (server)
|
|
# --startup (commands to run onStart)
|
|
# --connect (commands to run afterConnect)
|
|
# --config (configuration values)
|
|
parser = optparse.OptionParser(usage='Usage: %prog [options]',
|
|
version='supybot %s' % conf.version)
|
|
parser.add_option('-P', '--profile', action='store_true', dest='profile',
|
|
help='enables profiling')
|
|
parser.add_option('-O', action='count', dest='optimize',
|
|
help='-O optimizes asserts out of the code; ' \
|
|
'-OO optimizes asserts and uses psyco.')
|
|
parser.add_option('-n', '--nick', action='store',
|
|
dest='nick', default=defaultNick,
|
|
help='nick the bot should use')
|
|
parser.add_option('-s', '--server', action='store',
|
|
dest='server', default=defaultServer,
|
|
help='server to connect to')
|
|
parser.add_option('-u', '--user', action='store',
|
|
dest='user', default=defaultUser,
|
|
help='full username the bot should use')
|
|
parser.add_option('-i', '--ident', action='store',
|
|
dest='ident', default=defaultIdent,
|
|
help='ident the bot should use')
|
|
parser.add_option('-p', '--password', action='store',
|
|
dest='password', default=defaultPassword,
|
|
help='server password the bot should use')
|
|
parser.add_option('--startup', action='append', dest='onStart',
|
|
help='file of additional commands to run at startup.')
|
|
parser.add_option('--connect', action='append', dest='afterConnect',
|
|
help='file of additional commands to run after connect')
|
|
parser.add_option('--config', action='append', dest='conf',
|
|
help='file of configuration variables to set')
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
if options.optimize:
|
|
__builtins__.__debug__ = False
|
|
if options.optimize > 1:
|
|
import psyco
|
|
psyco.full()
|
|
|
|
if options.onStart:
|
|
for filename in options.onStart:
|
|
fd = file(filename)
|
|
for line in fd:
|
|
conf.commandsOnStart.append(line.rstrip())
|
|
fd.close()
|
|
|
|
if options.afterConnect:
|
|
for filename in options.afterConnect:
|
|
fd = file(filename)
|
|
for line in fd:
|
|
afterConnect.append(line.rstrip())
|
|
fd.close()
|
|
|
|
assignmentRe = re.compile('\s*[:=]\s*')
|
|
if options.conf:
|
|
for filename in options.conf:
|
|
fd = file(filename)
|
|
for line in fd:
|
|
(name, valueString) = assignmentRe.split(line.rstrip(), 1)
|
|
try:
|
|
value = eval(valueString)
|
|
except Exception, e:
|
|
sys.stderr.write('Invalid configuration value: %r' % \
|
|
valueString)
|
|
sys.exit(-1)
|
|
|
|
nick = options.nick
|
|
user = options.user
|
|
ident = options.ident
|
|
password = options.password
|
|
|
|
if ':' in options.server:
|
|
serverAndPort = options.server.split(':', 1)
|
|
serverAndPort[1] = int(serverAndPort[1])
|
|
server = tuple(serverAndPort)
|
|
else:
|
|
server = (options.server, 6667)
|
|
|
|
import irclib
|
|
import ircmsgs
|
|
import drivers
|
|
import callbacks
|
|
import OwnerCommands
|
|
|
|
class ConfigAfterConnect(irclib.IrcCallback):
|
|
public = False
|
|
def __init__(self, commands):
|
|
self.commands = commands
|
|
def do376(self, irc, msg):
|
|
for command in self.commands:
|
|
msg = ircmsgs.privmsg(irc.nick, command, prefix=irc.prefix)
|
|
irc.queueMsg(msg)
|
|
do377 = do376
|
|
do422 = do376 # MOTD File is missing.
|
|
|
|
# We pre-tokenize the commands just to save on significant amounts of work.
|
|
conf.commandsOnStart = map(callbacks.tokenize, conf.commandsOnStart)
|
|
|
|
irc = irclib.Irc(nick, user, ident, password)
|
|
callback = OwnerCommands.Class()
|
|
callback.configure(irc)
|
|
irc.addCallback(callback)
|
|
irc.addCallback(ConfigAfterConnect(afterConnect))
|
|
driver = drivers.newDriver(server, irc)
|
|
|
|
if options.profile:
|
|
import profile
|
|
profile.run('main()')
|
|
else:
|
|
main()
|
|
|
|
|
|
|
|
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78:
|