WhatTheMike/Fox-V3
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

context?

Browse Source
This commit is contained in:
bobloy 2018-05-05 16:21:58 -04:00
parent 2cd594c36a
commit fd0e383b03
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
chatter/chatter.py
View File

@ -4,7 +4,7 @@ from typing import List, Union
import discord
from discord.ext import commands
from redbot.core import Config
from redbot.core import Config, RedContext
from redbot.core.bot import Red
from .source import ChatBot
@ -12,6 +12,7 @@ from .source.trainers import ListTrainer
from datetime import datetime,timedelta
class Chatter:
"""
This cog trains a chatbot that will talk like members of your Guild
@ -74,14 +75,14 @@ class Chatter:
return True
@commands.group()
async def chatter(self, ctx: commands.Context):
async def chatter(self, ctx: RedContext):
"""
Base command for this cog. Check help for the commands list.
"""
if ctx.invoked_subcommand is None:
await ctx.send_help()
@chatter.command()
async def age(self, ctx: commands.Context, days: int):
async def age(self, ctx: RedContext, days: int):
"""
Sets the number of days to look back
Will train on 1 day otherwise
@ -91,7 +92,7 @@ class Chatter:
await ctx.send("Success")
@chatter.command()
async def train(self, ctx: commands.Context, channel: discord.TextChannel = None):
async def train(self, ctx: RedContext, channel: discord.TextChannel = None):
"""
Trains the bot based on language in this guild
"""
@ -118,7 +119,7 @@ class Chatter:
else:
await ctx.send("Error occurred :(")
async def on_message(self, message):
async def on_message(self, message: discord.Message):
"""
Credit to https://github.com/Twentysix26/26-Cogs/blob/master/cleverbot/cleverbot.py
for on_message recognition of @bot
@ -126,6 +127,7 @@ class Chatter:
author = message.author
channel = message.channel
if message.author.id != self.bot.user.id:
to_strip = "@" + author.guild.me.display_name + " "
text = message.clean_content