From: zonidjan <redacted>
Date: Mon, 14 Dec 2015 14:02:05 +0000 (-0600)
Subject: added userinfo module
X-Git-Url: https://jfr.im/git/erebus.git/commitdiff_plain/36f91d215514e5f06e1d811e2a24b6ae5a9271f1?ds=inline

added userinfo module
---

diff --git a/modules/contrib/userinfo/userinfo.json.example b/modules/contrib/userinfo/userinfo.json.example
new file mode 100644
index 0000000..0967ef4
--- /dev/null
+++ b/modules/contrib/userinfo/userinfo.json.example
@@ -0,0 +1 @@
+{}
diff --git a/modules/userinfo.py b/modules/userinfo.py
new file mode 100644
index 0000000..29eb59f
--- /dev/null
+++ b/modules/userinfo.py
@@ -0,0 +1,94 @@
+# Erebus IRC bot - Author: Erebus Team
+# trivia module
+# This file is released into the public domain; see http://unlicense.org/
+
+# module info
+modinfo = {
+	'author': 'Erebus Team',
+	'license': 'public domain',
+	'compatible': [1], # compatible module API versions
+	'depends': [], # other modules required to work properly?
+}
+
+# preamble
+import modlib
+lib = modlib.modlib(__name__)
+def modstart(parent_arg, *args, **kwargs):
+	global parent
+	parent = parent_arg
+	gotParent()
+	return lib.modstart(parent, *args, **kwargs)
+def modstop(*args, **kwargs):
+	closeshop()
+	return lib.modstop(*args, **kwargs)
+
+# module code
+import json
+
+#setup
+def gotParent():
+	global jsonfile, db
+	jsonfile = parent.cfg.get('userinfo', 'jsonpath', default="./modules/userinfo.json")
+	db = json.load(open(jsonfile, "r"))
+def closeshop():
+	if json is not None and json.dump is not None:
+		json.dump(db, open(jsonfile, "w"))#, indent=4, separators=(',', ': '))
+
+#functions
+def getauth(thing):
+	if isinstance(thing, parent.User):
+		if thing.auth is not None:
+			return "#"+thing.auth
+	elif isinstance(thing, basestring):
+		if thing[0] == "#":
+			return thing
+		else:
+			if parent.user(thing).auth is not None:
+				return "#"+parent.user(thing).auth
+	return None
+
+def has(user, key):
+	return (
+		key in db.get(getauth(user), {}) or
+		key in db.get(str(user).lower(), {})
+	)
+def get(user, key, default=None):
+	return (
+		db.get(getauth(user), {}).get(key,
+		db.get(str(user).lower(), {}).get(key,
+		default
+	)))
+def set(user, key, value):
+	if getauth(user) is not None: db.setdefault(getauth(user), {})[key] = value
+	db.setdefault(str(user).lower(), {})[key] = value
+
+#commands
+@lib.hook('get', needchan=False)
+def cmd_get(bot, user, chan, realtarget, *args):
+	if realtarget == chan.name: replyto = chan
+	else: replyto = user
+
+	if len(args) > 1:
+		target = args[0]
+		item = args[1]
+	else:
+		target = user
+		item = args[0]
+
+	value = get(target, item, None)
+	if value is None:
+		bot.msg(replyto, "%(user)s: %(item)s on %(target)s is not set." % {'user':user,'item':item,'target':target})
+	else:
+		bot.msg(replyto, "%(user)s: %(item)s on %(target)s: %(value)s" % {'user':user,'item':item,'target':target,'value':value})
+
+@lib.hook('set', needchan=False)
+@lib.argsGE(2)
+def cmd_set(bot, user, chan, realtarget, *args):
+	set(user, args[0], ' '.join(args[1:]))
+	bot.msg(user, "Done.")
+
+@lib.hook('oset', glevel=lib.STAFF, needchan=False)
+@lib.argsGE(3)
+def cmd_oset(bot, user, chan, realtarget, *args):
+	set(args[0], args[1], ' '.join(args[2:]))
+	bot.msg(user, "Done.")