mirror of
https://github.com/MrDetonia/Maki.git
synced 2024-11-10 15:48:06 -05:00
v1.3.0, added .spell command to search for D&D 5e spells
This commit is contained in:
parent
5c1afd1b73
commit
5541a7c128
5
bot.py
5
bot.py
@ -44,11 +44,14 @@ def on_ready():
|
||||
print('User: ' + client.user.name)
|
||||
print('ID: ' + client.user.id)
|
||||
|
||||
# populate D&D spells list
|
||||
for spell in requests.get('http://dnd5eapi.co/api/spells/').json()['results']:
|
||||
spellslist[spell['name']] = spell['url']
|
||||
|
||||
# set "Now Playing" to print version
|
||||
game = discord.Game(name=version)
|
||||
yield from client.change_presence(game=game)
|
||||
|
||||
|
||||
# called when message received
|
||||
@client.event
|
||||
@asyncio.coroutine
|
||||
|
49
commands.py
49
commands.py
@ -13,6 +13,8 @@ import re
|
||||
import requests
|
||||
import random
|
||||
import subprocess
|
||||
import itertools
|
||||
import ftfy
|
||||
|
||||
# LOCAL IMPORTS
|
||||
from common import *
|
||||
@ -197,6 +199,52 @@ def cmd_roll(client, msg):
|
||||
|
||||
|
||||
@asyncio.coroutine
|
||||
def cmd_spell(client, msg):
|
||||
searchterm = msg.content[7:]
|
||||
|
||||
# perform search on user input
|
||||
results = []
|
||||
for result in itertools.islice(search(spellslist, searchterm), 3):
|
||||
results.append(result)
|
||||
|
||||
# default response is an error
|
||||
response = "Couldn't find any matching spells!"
|
||||
|
||||
# otherwise, grab spell data and generate response text
|
||||
if results:
|
||||
result = requests.get(results[0][1]).json()
|
||||
|
||||
response = "**Spell:** " + result['name']
|
||||
if result['concentration'] is "yes":
|
||||
response += " *(C)*"
|
||||
response += " " + str(result['components'])
|
||||
response += "\n\n**Level:** " + str(result['level'])
|
||||
response += "\n\n**Description:**"
|
||||
for s in result['desc']:
|
||||
response += '\n' + s
|
||||
if 'higher_level' in result:
|
||||
response += "\n\n**Higher Level:**"
|
||||
for s in result['higher_level']:
|
||||
response += '\n' + s
|
||||
response += "\n\n**Range:** " + result['range']
|
||||
response += "\n\n**Casting Time:** " + result['casting_time']
|
||||
response += "\n\n**Duration:** " + result['duration']
|
||||
|
||||
# repair encoding errors from API
|
||||
response = ftfy.fix_text(response)
|
||||
|
||||
# append next search matches, if any
|
||||
matches = []
|
||||
for k,_ in results[1:]:
|
||||
matches.append(k)
|
||||
|
||||
if matches:
|
||||
response += "\n\n*Possible Matches: " + str(matches) + "*"
|
||||
|
||||
yield from discord_send(client, msg, response)
|
||||
|
||||
|
||||
@asyncio.coroutine
|
||||
def cmd_qr(client, msg):
|
||||
tmp = msg.content[4:]
|
||||
|
||||
@ -358,6 +406,7 @@ commands = {
|
||||
"sayy": cmd_sayy,
|
||||
"markov": cmd_markov,
|
||||
"roll": cmd_roll,
|
||||
"spell": cmd_spell,
|
||||
"qr": cmd_qr,
|
||||
"np": cmd_np,
|
||||
"steam": cmd_steam,
|
||||
|
@ -10,7 +10,7 @@ import os
|
||||
import json
|
||||
|
||||
# bot version
|
||||
version = "v1.2.3"
|
||||
version = "v1.3.0"
|
||||
|
||||
# TODO: generate this on the fly and make it look acceptable
|
||||
# text shown by .help command
|
||||
@ -27,6 +27,7 @@ My commands are:
|
||||
**.sayy <msg>** | say something a e s t h e t i c a l l y
|
||||
**.markov [<user>]** | generate markov chain over chat history for you or another user
|
||||
**.roll <num>d<val>** | roll x number of y sided dice
|
||||
**.spell <term>** | search for a D&D spell
|
||||
**.qr <msg>** | generate a QR code
|
||||
**.np [<user>]** | fetch now playing from last.fm for you or a specific username
|
||||
**.steam [<user>]** | fetch steam status for you or a specific vanityname
|
||||
@ -43,3 +44,5 @@ if os.path.isfile('hist.json'):
|
||||
|
||||
# quiet modes
|
||||
quiet = {}
|
||||
|
||||
spellslist = {}
|
||||
|
@ -42,6 +42,14 @@ def logger():
|
||||
logger.addHandler(handler)
|
||||
|
||||
|
||||
# fuzzy search over dictionary
|
||||
def search(dict, term):
|
||||
s = term.lower()
|
||||
|
||||
for item in dict.items():
|
||||
if s in item[0].lower():
|
||||
yield item
|
||||
|
||||
# send_message wrapper (deals with Discord's shit API)
|
||||
@asyncio.coroutine
|
||||
def discord_send(client, message, response):
|
||||
|
Loading…
Reference in New Issue
Block a user