config.py

   1 # This file is part of Acrobat.
   2 #
   3 # Acrobat is free software; you can redistribute it and/or modify
   4 # it under the terms of the GNU General Public License as published
   5 # by the Free Software Foundation; either version 2 of the License,
   6 # or (at your option) any later version.
   7 #
   8 # Acrobat is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 # GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License
  14 # along with Acrobat; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
  16 # USA.
  17
  18 # Andrew Walkingshaw <andrew@lexical.org.uk>
  19 # Peter Corbett <ptc24@cam.ac.uk>
  20 # Matthew Vernon <matthew@debian.org>
  21 # Stephen Early <steve@greenend.org.uk>
  22
  23 # Acrobat configuration file
  24
  25 # The following definitions are required to be present in this module:
  26 server = "rapun"
  27 port = 6667
  28 nickname = "Testbot"
  29 channel = "#test"
  30 owner = "Emperor"
  31 # Also a function called "command"; see later.
  32
  33 # Everything else in this file is configuration-specific.
  34
  35 # Most command implementations are stored in a separate module.
  36 import commands as c
  37
  38 # This fishpond is shared between trouts and flirts.  It doesn't have to be;
  39 # you can define as many ponds as you like.
  40 class fish:
  41         cur_fish=5
  42         max_fish=5
  43         nofish_time=60
  44         fish_time_inc=60
  45         fish_inc=2
  46         DoS=0
  47         Boring_Git='Nobody'
  48         quotatime=0
  49
  50 # load a file full of flirts or trouts
  51 def __load(filename):
  52     try:
  53         f = open(filename, "r")
  54         r = [l.strip() for l in f.readlines() if l.find("%s") != -1]
  55         f.close()
  56     except IOError:
  57         r = [ "doesn't know what to do about %s." ]
  58     return r
  59
  60 # (troutlist,selftroutmsg,DoSmsg,notargetmsg,nofishmsg,fishpond,selftroutprob)
  61 troutcfg = (
  62         __load("trouts"),
  63         __load("trouts-local"),
  64         ' (at the instigation of %s)',
  65         "Sorry, but %s is being a spoilsport.",
  66         "Who do you wish me to trout?",
  67         "Fish stocks exhausted.",
  68         fish,
  69         0.1)
  70
  71 flirtcfg = (
  72         __load("flirts"),
  73         ' (but %s is their secret admirer)',
  74         "Sorry, but %s made me take Holy Orders.",
  75         "Who do you wish me to flirt with?",
  76         "My libido is over-used!",
  77         fish,
  78         0.1)
  79
  80 # Hacky command to output the current fishpond state
  81 def fishq(bot, cmd, nick, conn, public,f):
  82         from irclib import irc_lower
  83         if not public and irc_lower(nick) == irc_lower(bot.owner):
  84                 state=("Fishpond state: cur_fish=%d, max_fish=%d, nofish_time=%d, "
  85                        +"fish_time_inc=%d, fish_inc=%d, DoS=%d, Boring_Git=%s, "
  86                        +"quotatime=%d")%(f.cur_fish,f.max_fish,f.nofish_time,
  87                                          f.fish_time_inc,f.fish_inc,f.DoS,f.Boring_Git,
  88                                          f.quotatime)
  89                 bot.automsg(public,nick,state)
  90
  91 # Karma implementation
  92 import cPickle
  93 karmafilename = "karmadump"
  94 # load the karma db
  95 try:
  96     f = open(karmafilename, "r")
  97     karmadb = cPickle.load(f)
  98     f.close()
  99 except IOError:
 100     karmadb = {}
 101 # Modify karma
 102 def karma(cmd, amount):
 103     thing=cmd.split()[0][:-2].lower()
 104     if karmadb.has_key(thing):
 105         karmadb[thing] += amount
 106     else:
 107         karmadb[thing] = amount
 108 def savekarma():
 109     try:
 110         f = open(karmafilename, "w")
 111         cPickle.dump(karmadb, f)
 112         f.close()
 113     except IOError:
 114         sys.stderr.write("Problems dumping karma: probably lost :(")
 115
 116 # When the bot exits we should save the karma db
 117 def quit(bot,cmd,nick,conn,public):
 118     savekarma()
 119     c.quitq(bot,cmd,nick,conn,public)
 120 def reload(bot,cmd,nick,conn,public):
 121     savekarma()
 122     c.reloadq(bot,cmd,nick,conn,public)
 123
 124 # Command processing: whenever something is said that the bot can hear,
 125 # "command" is invoked and must decide what to do.  This configuration
 126 # defines a couple of special cases (for karma) but is otherwise driven
 127 # by a dictionary of commands.
 128
 129 commands = {"karma": (c.karmaq,karmadb),
 130             "karmalist": (c.listkeysq,karmadb),
 131             "karmadel": (c.karmadelq,karmadb),
 132             "info": (c.infoq,karmadb),
 133             "trout": (c.troutq,troutcfg),
 134             "flirt": (c.troutq,flirtcfg),
 135             "fish": (fishq,fish),
 136             "quiet": (c.nofishq,fish),
 137             "reload": reload,
 138             "quit": quit,
 139             "google": c.googleq,
 140             "define": c.defineq,
 141             "say": c.sayq,
 142             "do": c.doq }
 143 # disconnect and hop annoy people
 144 #            "disconnect": c.disconnq,
 145 #            "hop": c.disconnq }
 146 commands["list"]=(c.listkeysq,commands)
 147
 148 triggers = ("!", "~") # what character should the bot be invoked by:
 149                       # eg !trout, ~trout etc.
 150
 151 def command(bot, cmd, nick, conn, public):
 152     ours=0
 153     try:
 154             if public and cmd[0] in triggers:
 155                     ours=1
 156                     cmd=cmd[1:]
 157             if not public:
 158                     ours=1
 159             command = cmd.split()[0]
 160     except IndexError:
 161             command=""
 162     # karma: up
 163     if command.endswith("++"):
 164         karma(cmd,1)
 165     # karma: down
 166     if command.endswith("--"):
 167         karma(cmd,-1)
 168
 169     if ours and command.lower() in commands.keys():
 170         e=commands[command]
 171         if callable(e):
 172             e(bot,cmd,nick,conn,public)
 173         else:
 174             e[0](bot,cmd,nick,conn,public,*e[1:])