Merge branch 'gonzobot' of github.com:snoonetIRC/CloudBot into gonzobot

edwardslabs · Jan 4, 2018 · 640429f · 640429f
2 parents 7ba6581 + 27615fe
commit 640429f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/plugins/duckhunt.py b/plugins/duckhunt.py
@@ -281,8 +281,10 @@ def update_score(nick, chan, db, conn, shoot=0, friend=0):
                        .where(table.c.name == nick.lower())).fetchone()
     if score:
         dbupdate(nick, chan, db, conn, score[0] + shoot, score[1] + friend)
+        return {'shoot': score[0] + shoot, 'friend': score[1] + friend}
     else:
         dbadd_entry(nick, chan, db, conn, shoot, friend)
+        return {'shoot': shoot, 'friend': friend}
 
 
 def attack(nick, chan, message, db, conn, notice, attack):
@@ -355,13 +357,12 @@ def attack(nick, chan, message, db, conn, notice, attack):
 
         status['duck_status'] = 2
         try:
-            score = 1
             args = {
-                attack_type: score
+                attack_type: 1
             }
 
-            update_score(nick, chan, db, conn, **args)
-        except DatabaseError:
+            score = update_score(nick, chan, db, conn, **args)[attack_type]
+        except Exception:
             status['duck_status'] = 1
             raise