ADD commands to set a quote and a get random quote.
parent
8ccfc36d4c
commit
c71b3c5125
@ -1,7 +1,40 @@
|
||||
"""
|
||||
Bot commands.
|
||||
"""
|
||||
import discord
|
||||
import logging
|
||||
from random import choice
|
||||
|
||||
from discord.ext import commands
|
||||
|
||||
from utils.database import get_quote, set_quote
|
||||
|
||||
client = commands.Bot(command_prefix='~')
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@client.command(aliases=['q'])
|
||||
async def quote(bot: object, quote: str) -> str:
|
||||
"""
|
||||
Save a quote into database.
|
||||
"""
|
||||
try:
|
||||
user = bot.author.name
|
||||
set_quote(user, quote)
|
||||
except Exception as ex:
|
||||
return await bot.send(ex.args)
|
||||
else:
|
||||
return await bot.send('Done:\n`%s`' % quote)
|
||||
|
||||
|
||||
@client.command(aliases=['rq'])
|
||||
async def random_quote(bot: object) -> str:
|
||||
"""
|
||||
Get an random quote from database.
|
||||
"""
|
||||
quotes = get_quote()
|
||||
chosen_one = choice(quotes)
|
||||
|
||||
try:
|
||||
return await bot.send(f'{chosen_one[0]}\n`By: {chosen_one[1]}`')
|
||||
except Exception as ex:
|
||||
return await bot.send(ex)
|
||||
|
Loading…
Reference in New Issue