X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/fb20be7c5c80f13474786337d69005f3497d21c6..0d93d7b47b642e1c5705b3c75fb7f5cfcbe0fe41:/modules/userinfo.py

diff --git a/modules/userinfo.py b/modules/userinfo.py
index b77f4d0..be6e33b 100644
--- a/modules/userinfo.py
+++ b/modules/userinfo.py
@@ -1,13 +1,14 @@
 # Erebus IRC bot - Author: Erebus Team
-# trivia module
+# userinfo 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?
+	'compatible': [1,2],
+	'depends': [],
+	'softdeps': ['help'],
 }
 
 # preamble
@@ -29,9 +30,12 @@ import json
 def gotParent():
 	global jsonfile, db
 	jsonfile = parent.cfg.get('userinfo', 'jsonpath', default="./modules/userinfo.json")
-	db = json.load(open(jsonfile, "r"))
+	try:
+		db = json.load(open(jsonfile, "r"))
+	except:
+		db = {}
 def closeshop():
-	if json is not None and json.dump is not None:
+	if json is not None and json.dump is not None and db != {}:
 		json.dump(db, open(jsonfile, "w"))#, indent=4, separators=(',', ': '))
 
 #functions
@@ -47,6 +51,8 @@ def getauth(thing):
 				return "#"+parent.user(thing).auth
 	return None
 
+def _keys(user):
+	return list(set(db.get(getauth(user), {}).keys() + db.get(str(user).lower(), {}).keys())) #list-to-set-to-list to remove duplicates
 def _has(user, key):
 	return (
 		key in db.get(getauth(user), {}) or
@@ -57,8 +63,8 @@ def _get(user, key, default=None):
 		db.get(getauth(user), {}). #try to get the auth
 			get(key, #try to get the info-key by auth
 			db.get(str(user).lower(), {}). #fallback to using the nick
-			get(key, #and try to get the info-key from that
-			default #otherwise throw out whatever default
+				get(key, #and try to get the info-key from that
+				default #otherwise throw out whatever default
 	)))
 def _set(user, key, value):
 	if getauth(user) is not None:
@@ -67,6 +73,21 @@ def _set(user, key, value):
 
 #commands
 @lib.hook(needchan=False)
+@lib.help("[<target>]", "lists info items known about someone", "<target> may be a nick, or an auth in format '#auth'", "it defaults to yourself")
+def getitems(bot, user, chan, realtarget, *args):
+	if chan is not None and realtarget == chan.name: replyto = chan
+	else: replyto = user
+
+	if len(args) > 0:
+		target = args[0]
+	else:
+		target = user
+
+	bot.msg(replyto, "%(user)s: %(target)s has the following info items: %(items)s" % {'user':user,'target':target,'items':(', '.join(_keys(target)))})
+
+@lib.hook(needchan=False)
+@lib.help("[<target>] <item>", "gets an info item about someone", "<target> may be a nick, or an auth in format '#auth'", "it defaults to yourself")
+@lib.argsGE(1)
 def getinfo(bot, user, chan, realtarget, *args):
 	if chan is not None and realtarget == chan.name: replyto = chan
 	else: replyto = user
@@ -85,13 +106,17 @@ def getinfo(bot, user, chan, realtarget, *args):
 		bot.msg(replyto, "%(user)s: %(item)s on %(target)s: %(value)s" % {'user':user,'item':item,'target':target,'value':value})
 
 @lib.hook(needchan=False)
+@lib.help("<item> <value>", "sets an info item about you")
 @lib.argsGE(2)
 def setinfo(bot, user, chan, realtarget, *args):
 	_set(user, args[0], ' '.join(args[1:]))
+	closeshop()
 	bot.msg(user, "Done.")
 
 @lib.hook(glevel=lib.STAFF, needchan=False)
+@lib.help("<target> <item> <value>", "sets an info item about someone else", "<target> may be a nick, or an auth in format '#auth'")
 @lib.argsGE(3)
 def osetinfo(bot, user, chan, realtarget, *args):
 	_set(args[0], args[1], ' '.join(args[2:]))
+	closeshop()
 	bot.msg(user, "Done.")