modules/eval.py

   1 # Erebus IRC bot - Author: Erebus Team
   2 # vim: fileencoding=utf-8
   3 # !EVAL and !EXEC commands
   4 # This file is released into the public domain; see http://unlicense.org/
   5
   6 # module info
   7 modinfo = {
   8         'author': 'Erebus Team',
   9         'license': 'public domain',
  10         'compatible': [0],
  11         'depends': [],
  12         'softdeps': ['help'],
  13 }
  14
  15 # preamble
  16 import modlib
  17 lib = modlib.modlib(__name__)
  18 modstart = lib.modstart
  19 modstop = lib.modstop
  20
  21 # module code
  22 import sys
  23 import ctlmod
  24
  25
  26 def module(name):
  27         return lib.mod(name)
  28
  29 @lib.hook('eval', needchan=False, wantchan=True, glevel=lib.OWNER)
  30 @lib.help("<python>", "eval")
  31 @lib.argsGE(1)
  32 def cmd_eval(bot, user, chan, realtarget, *args):
  33         if chan is not None: replyto = chan
  34         else: replyto = user
  35
  36         try: ret = eval(' '.join(args))
  37         except Exception: bot.msg(replyto, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1]))
  38         else: bot.msg(replyto, "Done: %r" % (ret,))
  39
  40
  41 @lib.hook('exec', needchan=False, wantchan=True, glevel=lib.OWNER)
  42 @lib.help("<python>", "exec")
  43 @lib.argsGE(1)
  44 def cmd_exec(bot, user, chan, realtarget, *args):
  45         if chan is not None: replyto = chan
  46         else: replyto = user
  47
  48         try: exec(' '.join(args))
  49         except Exception: bot.msg(replyto, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1]))
  50         else: bot.msg(replyto, "Done.")
  51
  52 @lib.hook('exception', needchan=False, glevel=lib.OWNER)
  53 @lib.help(None, "cause an exception")
  54 def cmd_exception(*args, **kwargs):
  55         raise Exception()
  56
  57 @lib.hook(needchan=False, wantchan=True, glevel=lib.ADMIN)
  58 @lib.help("<nick> <message>", "inject a line as though it came from <nick>", "note that this injects lines, not commands", "ex: INJECT DimeCadmium !WHOAMI")
  59 def inject(bot, user, chan, realtarget, *args):
  60         targetuser = bot.parent.user(args[0], create=False)
  61         if targetuser is None:
  62                 bot.msg(user, "User is unknown.")
  63                 return
  64         if targetuser.glevel > user.glevel:
  65                 bot.msg(user, "That user has a higher access level than you.")
  66                 return
  67
  68         if chan is not None:
  69                 bot.parsemsg(bot.parent.user(args[0], create=False), str(chan), ' '.join(args[1:]))
  70         else:
  71                 bot.parsemsg(bot.parent.user(args[0], create=False), str(bot), ' '.join(args[1:]))