diff --git a/plugins/coin.py b/plugins/coin.py index d9929fe..b50f329 100644 --- a/plugins/coin.py +++ b/plugins/coin.py @@ -2,38 +2,18 @@ from util import hook import random -## NOTES: -## Need to make a function to return a number of flips - -# this produces a string full of comma seperated coin flips - should output this as a list -def flip_list(count): - out = "" - for i in range(count): - flip = random.randint(0,1) - if flip == 1: - if i == 0: - out += "Heads" - else: - out += ", Heads" - else: - if i == 0: - out += "Tails" - else: - out += ", Tails" - return out - -# this doesn't work (yet) +# used for tals: x heads: y def flip_simple(count): - out = "" heads = 0 tails = 0 - for i in range(count): - flip = random.randint(0,1) - if flip == 1: - heads = heads + 1 + for x in range(count): + c = random.randint(0,1) + if c == 0: + heads += 1 else: - tails = tails + 1 - return msg + tails += 1 + return [heads,tails] + @hook.command(autohelp=False) def coin(inp): @@ -47,17 +27,15 @@ def coin(inp): return "Invalid Input :(" else: count = 1 - - # depending on the count, we use three different methods to get the output - if count <= 10 and count > 1: - msg = "You flip " + str(count) + " coins and get the following results: " + flip_list(count) - return msg - elif count == 1: + # depending on the count, we use two different methods to get the output + if count == 1: flip = random.randint(0,1) if flip == 1: - return "You flip a coin and get heads." + sidename = "heads" else: - return "You flip a coin and get tails." + sidename = "tails" + return "You flip a coin and get "+sidename+"." else: - return "Amounts over ten are not yet implemented" + flips = flip_simple(count) + return "You flip "+str(count)+" coins and get " +str(flips[0])+" heads and "+str(flips[1])+" tails."