Fix trying to roll a zero-sided die error

2011-06-20 16:43:39 -05:00 · 2011-06-20 16:43:39 -05:00 · a767b3d6e2
commit a767b3d6e2
parent c7b7e6e3ad
1 changed files with 6 additions and 0 deletions
--- a/modules/Dice.py
+++ b/modules/Dice.py
@ -119,6 +119,12 @@ class Dice(Module):
                    size = m[1]
                    if keep > dice or keep == 0:
                        keep = dice
+                    if size < 1:
+                        output = "# of sides for die is incorrect: %d" % size
+                        return output
+                    if dice < 1:
+                        output = "# of dice is incorrect: %d" % dice
+                        return output
                    res = self.roll_dice(keep, dice, size)
                    curr_str += "%d%s" % (res[0], res[1])
                    res = res[0]