73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
import discord
|
|
import random
|
|
import sys
|
|
|
|
client = discord.Client()
|
|
|
|
def succ(n):
|
|
if n == 0:
|
|
return 'Failure'
|
|
|
|
result = '{0} success{1}'.format(n, '' if n == 1 else 'es')
|
|
if n >= 5:
|
|
result += ', exceptional success! :tada:'
|
|
return result
|
|
|
|
def chance_die():
|
|
successes = 0
|
|
result = ''
|
|
roll = random.randrange(10)
|
|
if roll == 1:
|
|
return '1\n**Dramatic failure! :fire:**'
|
|
while roll == 0:
|
|
successes += 1
|
|
result += '0->'
|
|
roll = random.randrange(10)
|
|
result += '~~{0}~~'.format(roll)
|
|
return '{0}\n**{1}**'.format(result, succ(successes))
|
|
|
|
def dice(count):
|
|
results = []
|
|
successes = 0
|
|
for i in range(count):
|
|
result = ''
|
|
roll = random.randrange(10)
|
|
while roll == 0:
|
|
successes += 1
|
|
result += '0->'
|
|
roll = random.randrange(10)
|
|
if roll >= 8:
|
|
successes += 1
|
|
result += (('{0}' if roll >= 8 else '~~{0}~~').format(roll))
|
|
results.append(result)
|
|
|
|
return '{0}\n**{1}**'.format(', '.join(results), succ(successes))
|
|
|
|
@client.event
|
|
async def on_ready():
|
|
print('logged in as {0.user}'.format(client), file=sys.stderr)
|
|
await client.change_presence(activity=discord.Game('!n for n dice, !0 for chance die'))
|
|
|
|
@client.event
|
|
async def on_message(message):
|
|
if message.author == client.user:
|
|
return
|
|
|
|
if message.content.startswith('!'):
|
|
try:
|
|
count=int(message.content.split()[0][1:])
|
|
except ValueError:
|
|
print('error parsing message {0}'.format(message.content), file=sys.stderr)
|
|
return
|
|
|
|
if count > 69:
|
|
await message.channel.send(':thinking:')
|
|
elif count <= 0:
|
|
await message.channel.send(chance_die())
|
|
else:
|
|
await message.channel.send(dice(count))
|
|
|
|
with open('token') as f:
|
|
token = f.read().strip()
|
|
client.run(token)
|