Servus-chiark.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 # Richard Kettlewell <rjk@greenend.org.uk
  23
  24 # Acrobat configuration file
  25
  26 # The following definitions are required to be present in this module:
  27 server = "rapun"
  28 port = 6667
  29 nickname = "Servus"
  30 channel = "#chiark"
  31 owner = "Emperor"
  32 # Also a function called "command"; see later.
  33
  34 # Everything else in this file is configuration-specific.
  35
  36 import os, time, re
  37
  38 # Most command implementations are stored in a separate module.
  39 import commands as c
  40
  41 # This fishpond is shared between trouts and flirts.  It doesn't have to be;
  42 # you can define as many ponds as you like.
  43 class fish:
  44         cur_fish=5
  45         max_fish=5
  46         nofish_time=60
  47         fish_time_inc=60
  48         fish_inc=2
  49         DoS=0
  50         Boring_Git='Nobody'
  51         quotatime=0
  52
  53 # load a file full of flirts or trouts
  54 def __load(filename):
  55     try:
  56         f = open(filename, "r")
  57         r = [l.strip() for l in f.readlines() if l.find("%s") != -1]
  58         f.close()
  59     except IOError:
  60         r = [ "doesn't know what to do about %s." ]
  61     return r
  62
  63 # (troutlist,selftroutmsg,DoSmsg,notargetmsg,nofishmsg,fishpond,selftroutprob)
  64 troutcfg = (
  65         __load("trouts"),
  66         ' (at the instigation of %s)',
  67         "Sorry, but %s is being a spoilsport.",
  68         "Who do you wish me to trout?",
  69         "Fish stocks exhausted.",
  70         fish,
  71         0.1)
  72
  73 flirtcfg = (
  74         __load("flirts"),
  75         ' (but %s is their secret admirer)',
  76         "Sorry, but %s made me take Holy Orders.",
  77         "Who do you wish me to flirt with?",
  78         "My libido is over-used!",
  79         fish,
  80         0.1)
  81
  82 slashcfg= (
  83         __load("slashes"),
  84         ' (while %s watches)',
  85         "Sorry, but %s stole my pen.",
  86         "Who do you want to slash?",
  87         "I have writer's block!",
  88         fish,
  89         0.1)
  90
  91 # Hacky command to output the current fishpond state
  92 def fishq(bot, cmd, nick, conn, public,f):
  93         from irclib import irc_lower
  94         if not public and irc_lower(nick) == irc_lower(bot.owner):
  95                 state=("Fishpond state: cur_fish=%d, max_fish=%d, nofish_time=%d, "
  96                        +"fish_time_inc=%d, fish_inc=%d, DoS=%d, Boring_Git=%s, "
  97                        +"quotatime=%d")%(f.cur_fish,f.max_fish,f.nofish_time,
  98                                          f.fish_time_inc,f.fish_inc,f.DoS,f.Boring_Git,
  99                                          f.quotatime)
 100                 bot.automsg(public,nick,state)
 101
 102 # Karma implementation
 103 import cPickle
 104 karmafilename = "chiark-karma"
 105 # load the karma db
 106 try:
 107     f = open(karmafilename, "r")
 108     karmadb = cPickle.load(f)
 109     f.close()
 110 except IOError:
 111     karmadb = {}
 112 # Modify karma
 113 def karma(cmd, amount):
 114     thing=cmd.split()[0][:-2].lower()
 115     if karmadb.has_key(thing):
 116         karmadb[thing] += amount
 117     else:
 118         karmadb[thing] = amount
 119     savekarma()
 120 def savekarma():
 121     tmp = "%s.tmp" % karmafilename
 122     try:
 123         f = open(tmp, "w")
 124         cPickle.dump(karmadb, f)
 125         f.close()
 126         os.rename(tmp, karmafilename)
 127     except IOError, e:
 128         sys.stderr.write("error writing karma: %s" % e)
 129
 130 def quit(bot,cmd,nick,conn,public):
 131     c.quitq(bot,cmd,nick,conn,public)
 132 def reload(bot,cmd,nick,conn,public):
 133     c.reloadq(bot,cmd,nick,conn,public)
 134
 135 #The game...
 136 class game:
 137         trigger="Servus"
 138         grace=time.time()
 139         minlose=24*60*60 #1 day
 140         maxlose=14*minlose #2 weeks
 141         losetime=time.time()+300000
 142
 143 # initialise the urldb on startup
 144 urldb={}
 145 lastexp=time.time()
 146 #expire urls if not asked about or seen for >71 hours
 147 expirelen=71*60*60
 148 #do an expiry run every hour
 149 expirevery=60*60
 150
 151 # Command processing: whenever something is said that the bot can hear,
 152 # "command" is invoked and must decide what to do.  This configuration
 153 # defines a couple of special cases (for karma) but is otherwise driven
 154 # by a dictionary of commands.
 155
 156 commands = {"karma": (c.karmaq,karmadb),
 157             "karmalist": (c.listkeysq,karmadb),
 158             "karmadel": (c.karmadelq,karmadb),
 159             "info": (c.infoq,karmadb),
 160             "trout": (c.troutq,troutcfg),
 161             "slash": (c.slashq, slashcfg),
 162             "rot13": c.rot13q,
 163             "fish": (fishq,fish),
 164             "flirt": (c.troutq,flirtcfg),
 165             "quiet": (c.nofishq,fish),
 166             "reload": reload,
 167             "quit": quit,
 168             "die": quit,
 169             "define": c.defineq,
 170             "google": c.googleq,
 171             "url": (c.urlq,urldb),
 172             "units": c.unitq,
 173             "help": c.helpq,
 174 #           "game": (c.gameq,game),
 175             "say": c.sayq,
 176             "do": c.doq }
 177 # disconnect and hop annoy people
 178 #            "disconnect": c.disconnq,
 179 #            "hop": c.disconnq }
 180 commands["list"]=(c.listkeysq,commands,True)
 181
 182 triggers = ("!", "~") # what character should the bot be invoked by:
 183                       # eg !trout, ~trout etc.
 184
 185 def command(bot, cmd, nick, conn, public):
 186     global urldb,lastexp,expirelen,expirevery
 187     ours=0
 188     try:
 189             if public and cmd[0] in triggers:
 190                     ours=1
 191                     cmd=cmd[1:]
 192             if not public:
 193                     ours=1
 194             command = cmd.split()[0]
 195     except IndexError:
 196             command=""
 197
 198     t=time.time()
 199     if t - lastexp > expirevery:
 200             c.urlexpire(urldb,expirelen)
 201             lastexp=t
 202
 203     if public:
 204       if c.urlre.search(cmd) and command.lower()!="url":
 205         c.dourl(bot,conn,nick,cmd,urldb)
 206
 207     # karma: up
 208     if command.endswith("++"):
 209         karma(cmd,1)
 210     # karma: down
 211     if command.endswith("--"):
 212         karma(cmd,-1)
 213
 214     if ours and command.lower() in commands.keys():
 215         e=commands[command.lower()]
 216         if callable(e):
 217             e(bot,cmd,nick,conn,public)
 218         else:
 219             e[0](bot,cmd,nick,conn,public,*e[1:])
 220 #    elif public:
 221 #        if cmd.find("GAME")!=-1: #someone else lost
 222 #           grace.grace=time.time()+60*20
 223 #       elif cmd.find(game.trigger)!=-1 and len(game.trigger)>2: #we lost!
 224 #           c.gameq(bot,"pad "+game.trigger,bot.owner,conn,False,game)
 225 #       elif time.time()>game.losetime: #we randomly lost, take new trigger
 226 #           c.gameq(bot,cmd,bot.owner,conn,False,game)
 227 #