45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
# # Lukeroge
|
|
from util import hook
|
|
import random
|
|
|
|
# used for tals: x heads: y
|
|
def flip_simple(count):
|
|
heads = 0
|
|
tails = 0
|
|
for x in range(count):
|
|
c = random.randint(0,1)
|
|
if c == 0:
|
|
heads += 1
|
|
else:
|
|
tails += 1
|
|
return [heads,tails]
|
|
|
|
|
|
@hook.command(autohelp=False)
|
|
def coin(inp):
|
|
".coin [amount] -- flips some coins and shares the result."
|
|
|
|
# checking for valid input. if valid input [count=inp], if invalid [return error], if no input [count=1]
|
|
if inp.isdigit():
|
|
count = int(inp)
|
|
else:
|
|
if inp:
|
|
return "Invalid input."
|
|
else:
|
|
count = 1
|
|
|
|
if count > 200:
|
|
return "Too many coins! Maximum is 200."
|
|
# depending on the count, we use two different methods to get the output
|
|
if count == 1:
|
|
flip = random.randint(0,1)
|
|
if flip == 1:
|
|
sidename = "heads"
|
|
else:
|
|
sidename = "tails"
|
|
return "You flip a coin and get "+sidename+"."
|
|
else:
|
|
flips = flip_simple(count)
|
|
return "You flip "+str(count)+" coins and get " +str(flips[0])+" heads and "+str(flips[1])+" tails."
|
|
|