X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/4df64299d7751fb90a9cf5e2964e9bcc365bcc3e..694d91f3ec3f4e73a9c176fffb21d3d42e9db5ee:/modules/resources.py

diff --git a/modules/resources.py b/modules/resources.py
index 1a4e3c5..69a6921 100644
--- a/modules/resources.py
+++ b/modules/resources.py
@@ -19,9 +19,9 @@ modstop = lib.modstop
 # module code
 import resource
 
-@lib.hook('ram', needchan=False, glevel=lib.MANAGER)
-def cmd_ram(bot, user, chan, realtarget, *args):
-	if realtarget == chan.name: replyto = chan
+@lib.hook(needchan=False, glevel=lib.MANAGER)
+def ram(bot, user, chan, realtarget, *args):
+	if chan is not None and realtarget == chan.name: replyto = chan
 	else: replyto = user
 
 	try:
@@ -31,9 +31,9 @@ def cmd_ram(bot, user, chan, realtarget, *args):
 
 	bot.msg(replyto, "Memory usage (MiB): %r" % (res.ru_maxrss/1024.0))
 
-@lib.hook('resources', needchan=False, glevel=lib.MANAGER)
-def cmd_resources(bot, user, chan, realtarget, *args):
-	if realtarget == chan.name: replyto = chan
+@lib.hook(needchan=False, glevel=lib.MANAGER)
+def resources(bot, user, chan, realtarget, *args):
+	if chan is not None and realtarget == chan.name: replyto = chan
 	else: replyto = user
 
 	try: