modules/userinfo.py

   1 # Erebus IRC bot - Author: Erebus Team
   2 # trivia module
   3 # This file is released into the public domain; see http://unlicense.org/
   4
   5 # module info
   6 modinfo = {
   7         'author': 'Erebus Team',
   8         'license': 'public domain',
   9         'compatible': [1], # compatible module API versions
  10         'depends': [], # other modules required to work properly?
  11 }
  12
  13 # preamble
  14 import modlib
  15 lib = modlib.modlib(__name__)
  16 def modstart(parent_arg, *args, **kwargs):
  17         global parent
  18         parent = parent_arg
  19         gotParent()
  20         return lib.modstart(parent, *args, **kwargs)
  21 def modstop(*args, **kwargs):
  22         closeshop()
  23         return lib.modstop(*args, **kwargs)
  24
  25 # module code
  26 import json
  27
  28 #setup
  29 def gotParent():
  30         global jsonfile, db
  31         jsonfile = parent.cfg.get('userinfo', 'jsonpath', default="./modules/userinfo.json")
  32         try:
  33                 db = json.load(open(jsonfile, "r"))
  34         except:
  35                 db = {}
  36 def closeshop():
  37         if json is not None and json.dump is not None:
  38                 json.dump(db, open(jsonfile, "w"))#, indent=4, separators=(',', ': '))
  39
  40 #functions
  41 def getauth(thing):
  42         if isinstance(thing, parent.User):
  43                 if thing.auth is not None:
  44                         return "#"+thing.auth
  45         elif isinstance(thing, basestring):
  46                 if thing[0] == "#":
  47                         return thing
  48                 else:
  49                         if parent.user(thing).auth is not None:
  50                                 return "#"+parent.user(thing).auth
  51         return None
  52
  53 def _has(user, key):
  54         return (
  55                 key in db.get(getauth(user), {}) or
  56                 key in db.get(str(user).lower(), {})
  57         )
  58 def _get(user, key, default=None):
  59         return (
  60                 db.get(getauth(user), {}). #try to get the auth
  61                         get(key, #try to get the info-key by auth
  62                         db.get(str(user).lower(), {}). #fallback to using the nick
  63                         get(key, #and try to get the info-key from that
  64                         default #otherwise throw out whatever default
  65         )))
  66 def _set(user, key, value):
  67         if getauth(user) is not None:
  68                 db.setdefault(getauth(user), {})[key] = value #use auth if we can
  69         db.setdefault(str(user).lower(), {})[key] = value #but set nick too
  70
  71 #commands
  72 @lib.hook(needchan=False)
  73 @lib.help("[<user>] <item>", "gets an info item about you or someone else")
  74 @lib.argsGE(1)
  75 def getinfo(bot, user, chan, realtarget, *args):
  76         if chan is not None and realtarget == chan.name: replyto = chan
  77         else: replyto = user
  78
  79         if len(args) > 1:
  80                 target = args[0]
  81                 item = args[1]
  82         else:
  83                 target = user
  84                 item = args[0]
  85
  86         value = _get(target, item, None)
  87         if value is None:
  88                 bot.msg(replyto, "%(user)s: %(item)s on %(target)s is not set." % {'user':user,'item':item,'target':target})
  89         else:
  90                 bot.msg(replyto, "%(user)s: %(item)s on %(target)s: %(value)s" % {'user':user,'item':item,'target':target,'value':value})
  91
  92 @lib.hook(needchan=False)
  93 @lib.help("<item> <value>", "sets an info item about you")
  94 @lib.argsGE(2)
  95 def setinfo(bot, user, chan, realtarget, *args):
  96         _set(user, args[0], ' '.join(args[1:]))
  97         bot.msg(user, "Done.")
  98
  99 @lib.hook(glevel=lib.STAFF, needchan=False)
 100 @lib.help("<user> <item> <value>", "sets an info item about someone else")
 101 @lib.argsGE(3)
 102 def osetinfo(bot, user, chan, realtarget, *args):
 103         _set(args[0], args[1], ' '.join(args[2:]))
 104         bot.msg(user, "Done.")