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

Add chat channel

Browse Source
This commit is contained in:
bobloy 2020-08-24 12:51:57 -04:00
parent c2c6d61a35
commit 6f414be6ab
1 changed files with 41 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
chatter/chat.py
View File

@ -3,6 +3,7 @@ import logging
import os import os
import pathlib import pathlib
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Optional
import discord import discord
from chatterbot import ChatBot from chatterbot import ChatBot
@ -17,6 +18,13 @@ from redbot.core.utils.predicates import MessagePredicate
log = logging.getLogger("red.fox_v3.chatter") log = logging.getLogger("red.fox_v3.chatter")
def my_local_get_prefix(prefixes, content):
for p in prefixes:
if content.startswith(p):
return p
return None
class ENG_LG: class ENG_LG:
ISO_639_1 = "en_core_web_lg" ISO_639_1 = "en_core_web_lg"
ISO_639 = "eng" ISO_639 = "eng"
@ -45,7 +53,7 @@ class Chatter(Cog):
self.bot = bot self.bot = bot
self.config = Config.get_conf(self, identifier=6710497116116101114) self.config = Config.get_conf(self, identifier=6710497116116101114)
default_global = {} default_global = {}
default_guild = {"whitelist": None, "days": 1, "convo_delta": 15} default_guild = {"whitelist": None, "days": 1, "convo_delta": 15, "chatchannel": None}
path: pathlib.Path = cog_data_path(self) path: pathlib.Path = cog_data_path(self)
self.data_path = path / "database.sqlite3" self.data_path = path / "database.sqlite3"
@ -183,6 +191,25 @@ class Chatter(Cog):
if ctx.invoked_subcommand is None: if ctx.invoked_subcommand is None:
pass pass
@chatter.command(name="channel")
async def chatter_channel(
self, ctx: commands.Context, channel: Optional[discord.TextChannel] = None
):
"""
Set a channel that the bot will respond in without mentioning it
Pass with no channel object to clear this guild's channel
"""
if channel is None:
await self.config.guild(ctx.guild).chatchannel.set(None)
await ctx.maybe_send_embed("Chat channel for guild is cleared")
else:
if channel.guild != ctx.guild:
await ctx.maybe_send_embed("What are you trying to pull here? :eyes:")
return
await self.config.guild(ctx.guild).chatchannel.set(channel.id)
await ctx.maybe_send_embed(f"Chat channel is now {channel.mention}")
@chatter.command(name="cleardata") @chatter.command(name="cleardata")
async def chatter_cleardata(self, ctx: commands.Context, confirm: bool = False): async def chatter_cleardata(self, ctx: commands.Context, confirm: bool = False):
""" """
@ -434,13 +461,11 @@ class Chatter(Cog):
########### ###########
# Thank you Cog-Creators # Thank you Cog-Creators
channel: discord.TextChannel = message.channel
def my_local_get_prefix(prefixes, content): if channel.id == await self.config.guild(guild).chatchannel():
for p in prefixes: pass # good to go
if content.startswith(p): else:
return p
return None
when_mentionables = commands.when_mentioned(self.bot, message) when_mentionables = commands.when_mentioned(self.bot, message)
prefix = my_local_get_prefix(when_mentionables, message.content) prefix = my_local_get_prefix(when_mentionables, message.content)
@ -449,9 +474,8 @@ class Chatter(Cog):
# print("not mentioned") # print("not mentioned")
return return
channel: discord.TextChannel = message.channel
message.content = message.content.replace(prefix, "", 1) message.content = message.content.replace(prefix, "", 1)
text = message.clean_content text = message.clean_content
async with channel.typing(): async with channel.typing():