add error checking on mysql DataErrors

[erebus.git] / modules / channel_admin.py

diff --git a/modules/channel_admin.py b/modules/channel_admin.py
index da9d22863f9db5228c0b8942dd73ad481a90cc74..b8bebc6e9922bee5136ee2809c78c6c4588cd9aa 100644 (file)
--- a/modules/channel_admin.py
+++ b/modules/channel_admin.py
@@ -75,8 +75,10 @@ def clevel(bot, user, chan, realtarget, *args):
                if user_clevel <= level:
                        return "I'm afraid I can't let you do that. Your current access level is not higher than you are trying to set."
 
-       chan.setlevel(target.auth, level)
-       return "Set #%s channel level to %s" % (target.auth, args[1])
+       if chan.setlevel(target.auth, level):
+               return "Set #%s channel level to %s" % (target.auth, args[1])
+       else:
+               return "An error occurred!"
 
 
 @lib.hook(('forceclevel','fclevel'), glevel=lib.MANAGER)
@@ -85,5 +87,7 @@ def clevel(bot, user, chan, realtarget, *args):
 def forceclevel(bot, user, chan, realtarget, *args):
        target = user.auth
        level = _resolve_level(args[0])
-       chan.setlevel(target, level)
-       return 'Your level on %s has been set to %d' % (chan, level)
+       if chan.setlevel(target, level):
+               return 'Your level on %s has been set to %d' % (chan, level)
+       else:
+               return "An error occurred!"