fix(twitch): Fix relative imports
This commit is contained in:
+14
-15
@@ -1,11 +1,13 @@
|
|||||||
import twitchio
|
import twitchio
|
||||||
|
|
||||||
from customcmd.client import ExecClient
|
|
||||||
from customcmd.data import CustomCmd, CustomCmdData
|
|
||||||
from meta import Bot
|
from meta import Bot
|
||||||
|
|
||||||
|
from ..customcmd.client import ExecClient
|
||||||
|
from ..customcmd.data import CustomCmd, CustomCmdData
|
||||||
|
|
||||||
from .context import CustomContext, CustomCmdUser
|
from .context import CustomContext, CustomCmdUser
|
||||||
|
|
||||||
|
|
||||||
class TwitchCmdClient(ExecClient):
|
class TwitchCmdClient(ExecClient):
|
||||||
def __init__(self, bot: Bot, data: CustomCmdData):
|
def __init__(self, bot: Bot, data: CustomCmdData):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
@@ -22,12 +24,12 @@ class TwitchCmdClient(ExecClient):
|
|||||||
if message.reply:
|
if message.reply:
|
||||||
# Strip the hidden mention prefix from the start of the message
|
# Strip the hidden mention prefix from the start of the message
|
||||||
prefix = message.reply.parent_user.mention
|
prefix = message.reply.parent_user.mention
|
||||||
content = content[len(prefix) + 1:]
|
content = content[len(prefix) + 1 :]
|
||||||
|
|
||||||
# If we start with a valid prefix, parse out a command name
|
# If we start with a valid prefix, parse out a command name
|
||||||
if prefix := next((p for p in prefixes if content.startswith(p)), None):
|
if prefix := next((p for p in prefixes if content.startswith(p)), None):
|
||||||
# Strip the valid prefix
|
# Strip the valid prefix
|
||||||
content = content[len(prefix):].strip()
|
content = content[len(prefix) :].strip()
|
||||||
if not content:
|
if not content:
|
||||||
# Exit early, no command
|
# Exit early, no command
|
||||||
return
|
return
|
||||||
@@ -45,7 +47,7 @@ class TwitchCmdClient(ExecClient):
|
|||||||
# Read the arguments, build the context, run the command.
|
# Read the arguments, build the context, run the command.
|
||||||
# TODO: Check the permissions and log etc.
|
# TODO: Check the permissions and log etc.
|
||||||
# TODO: Exception handling, as usual
|
# TODO: Exception handling, as usual
|
||||||
# TODO: Nice string reprs
|
# TODO: Nice string reprs
|
||||||
ctx = CustomContext(
|
ctx = CustomContext(
|
||||||
bot=self.bot,
|
bot=self.bot,
|
||||||
args=args,
|
args=args,
|
||||||
@@ -53,20 +55,17 @@ class TwitchCmdClient(ExecClient):
|
|||||||
used_prefix=prefix,
|
used_prefix=prefix,
|
||||||
message=message,
|
message=message,
|
||||||
initial_globals={
|
initial_globals={
|
||||||
'author': CustomCmdUser(message.chatter),
|
"author": CustomCmdUser(message.chatter),
|
||||||
'channel': CustomCmdUser(message.broadcaster),
|
"channel": CustomCmdUser(message.broadcaster),
|
||||||
}
|
},
|
||||||
)
|
)
|
||||||
result = await self.run_command(ctx, cmd.response)
|
result = await self.run_command(ctx, cmd.response)
|
||||||
if result:
|
if result:
|
||||||
await message.broadcaster.send_message(
|
await message.broadcaster.send_message(result, sender=self.bot.user)
|
||||||
result,
|
|
||||||
sender=self.bot.user
|
|
||||||
)
|
|
||||||
|
|
||||||
async def get_prefixes(self, message: twitchio.ChatMessage):
|
async def get_prefixes(self, message: twitchio.ChatMessage):
|
||||||
# TODO:
|
# TODO:
|
||||||
return ('!',)
|
return ("!",)
|
||||||
|
|
||||||
async def get_commands_in(self, communityid: int) -> dict[str, CustomCmd]:
|
async def get_commands_in(self, communityid: int) -> dict[str, CustomCmd]:
|
||||||
"""
|
"""
|
||||||
|
|||||||
+15
-15
@@ -2,23 +2,21 @@ from typing import Optional
|
|||||||
|
|
||||||
import twitchio
|
import twitchio
|
||||||
from twitchio import PartialUser
|
from twitchio import PartialUser
|
||||||
from twitchio.ext import commands as cmds
|
from twitchio.ext import commands as cmds
|
||||||
|
|
||||||
from customcmd.parser.context import ExecContext
|
|
||||||
from meta import Bot
|
from meta import Bot
|
||||||
from twitch.client import TwitchCmdClient
|
from twitch.client import TwitchCmdClient
|
||||||
from utils.lib import utc_now
|
from utils.lib import utc_now
|
||||||
|
|
||||||
from . import logger
|
from . import logger
|
||||||
from ..customcmd.data import CustomCmd, CustomCmdData
|
from ..customcmd.data import CustomCmd, CustomCmdData
|
||||||
from ..customcmd.client import ExecClient
|
from ..customcmd.client import ExecClient
|
||||||
|
from ..customcmd.parser import ExecContext
|
||||||
|
|
||||||
# TODO: Make an alphacroc or croccyalpha client for testing
|
# TODO: Make an alphacroc or croccyalpha client for testing
|
||||||
# TODO: Fix the tracker not actually guarding the commands properly
|
# TODO: Fix the tracker not actually guarding the commands properly
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class CustomCmdComponent(cmds.Component):
|
class CustomCmdComponent(cmds.Component):
|
||||||
def __init__(self, bot: Bot):
|
def __init__(self, bot: Bot):
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
@@ -31,11 +29,11 @@ class CustomCmdComponent(cmds.Component):
|
|||||||
await self.bot.version_check(*self.data.VERSION)
|
await self.bot.version_check(*self.data.VERSION)
|
||||||
|
|
||||||
async def component_teardown(self):
|
async def component_teardown(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
# ----- Methods -----
|
# ----- Methods -----
|
||||||
|
|
||||||
# ----- Event handlers -----
|
# ----- Event handlers -----
|
||||||
@cmds.Component.listener()
|
@cmds.Component.listener()
|
||||||
async def event_message(self, payload: twitchio.ChatMessage):
|
async def event_message(self, payload: twitchio.ChatMessage):
|
||||||
await self.cmdclient.process_message(payload)
|
await self.cmdclient.process_message(payload)
|
||||||
@@ -44,9 +42,9 @@ class CustomCmdComponent(cmds.Component):
|
|||||||
@cmds.group(name="commands")
|
@cmds.group(name="commands")
|
||||||
async def cmd_grp(self, ctx: cmds.Context):
|
async def cmd_grp(self, ctx: cmds.Context):
|
||||||
# TODO: Probably attempt to list commands or something here.
|
# TODO: Probably attempt to list commands or something here.
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@cmd_grp.command(name="add", aliases=('create', 'new'))
|
@cmd_grp.command(name="add", aliases=("create", "new"))
|
||||||
@cmds.is_broadcaster() # TODO: Better command creation perm checks
|
@cmds.is_broadcaster() # TODO: Better command creation perm checks
|
||||||
async def cmd_add(self, ctx: cmds.Context, name: str, *, response: str):
|
async def cmd_add(self, ctx: cmds.Context, name: str, *, response: str):
|
||||||
comm = await self.bot.profiles.fetch_community(ctx.broadcaster)
|
comm = await self.bot.profiles.fetch_community(ctx.broadcaster)
|
||||||
@@ -60,11 +58,13 @@ class CustomCmdComponent(cmds.Component):
|
|||||||
|
|
||||||
# Make sure the command parses (can't guarantee it runs, but at least it can parse)
|
# Make sure the command parses (can't guarantee it runs, but at least it can parse)
|
||||||
try:
|
try:
|
||||||
await self.cmdclient.run_command(ExecContext(args=[], content=''), response)
|
await self.cmdclient.run_command(ExecContext(args=[], content=""), response)
|
||||||
except Exception:
|
except Exception:
|
||||||
# TODO: Nicer errors for pinpointing problem
|
# TODO: Nicer errors for pinpointing problem
|
||||||
await ctx.reply(f"Sorry, could not parse your command!")
|
await ctx.reply(f"Sorry, could not parse your command!")
|
||||||
logger.info(f"Failed to parse attempted customcmd '{response}'", exc_info=True)
|
logger.info(
|
||||||
|
f"Failed to parse attempted customcmd '{response}'", exc_info=True
|
||||||
|
)
|
||||||
|
|
||||||
# If everything is good, add it to the table
|
# If everything is good, add it to the table
|
||||||
# TODO: Again, permission resolution
|
# TODO: Again, permission resolution
|
||||||
@@ -79,10 +79,10 @@ class CustomCmdComponent(cmds.Component):
|
|||||||
|
|
||||||
await ctx.reply(f"Created a new command !{name}")
|
await ctx.reply(f"Created a new command !{name}")
|
||||||
|
|
||||||
@cmd_grp.command(name="del", aliases=('delete', 'remove', 'rm'))
|
@cmd_grp.command(name="del", aliases=("delete", "remove", "rm"))
|
||||||
@cmds.is_broadcaster()
|
@cmds.is_broadcaster()
|
||||||
async def cmd_del(self, ctx: cmds.Context, name: str):
|
async def cmd_del(self, ctx: cmds.Context, name: str):
|
||||||
# Lookup the name to make sure a command does exist
|
# Lookup the name to make sure a command does exist
|
||||||
# Delete the command (again, consider permissions later)
|
# Delete the command (again, consider permissions later)
|
||||||
# Confirm with user.
|
# Confirm with user.
|
||||||
comm = await self.bot.profiles.fetch_community(ctx.broadcaster)
|
comm = await self.bot.profiles.fetch_community(ctx.broadcaster)
|
||||||
|
|||||||
+17
-16
@@ -1,11 +1,11 @@
|
|||||||
import twitchio
|
import twitchio
|
||||||
|
|
||||||
from customcmd.client import ExecClient
|
|
||||||
from customcmd.data import CustomCmd, CustomCmdData
|
|
||||||
from meta import Bot
|
from meta import Bot
|
||||||
from utils.lib import utc_now, strfdur
|
from utils.lib import utc_now, strfdur
|
||||||
|
|
||||||
from ..customcmd.parser import ExecContext
|
from ..customcmd.parser import ExecContext
|
||||||
|
from ..customcmd.client import ExecClient
|
||||||
|
from ..customcmd.data import CustomCmd, CustomCmdData
|
||||||
from .ctxvars import stuff
|
from .ctxvars import stuff
|
||||||
|
|
||||||
|
|
||||||
@@ -25,10 +25,7 @@ class CustomCmdUser:
|
|||||||
return self.user.display_name or self.user.name or self.user.id
|
return self.user.display_name or self.user.name or self.user.id
|
||||||
|
|
||||||
async def _exec_attr_whisper(self, ctx, message: str):
|
async def _exec_attr_whisper(self, ctx, message: str):
|
||||||
await self.user.send_whisper(
|
await self.user.send_whisper(to_user=self.user, message=message)
|
||||||
to_user=self.user,
|
|
||||||
message=message
|
|
||||||
)
|
|
||||||
|
|
||||||
async def _exec_attr_followage(self, ctx, channel: twitchio.PartialUser):
|
async def _exec_attr_followage(self, ctx, channel: twitchio.PartialUser):
|
||||||
followed_at = None
|
followed_at = None
|
||||||
@@ -38,7 +35,9 @@ class CustomCmdUser:
|
|||||||
|
|
||||||
if followed_at:
|
if followed_at:
|
||||||
durstr = strfdur(utc_now() - followed_at)
|
durstr = strfdur(utc_now() - followed_at)
|
||||||
resp = f"{self.user.mention} has been following {channel.mention} for {durstr}"
|
resp = (
|
||||||
|
f"{self.user.mention} has been following {channel.mention} for {durstr}"
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
return f"{self.user.mention} is not following {channel.mention}"
|
return f"{self.user.mention} is not following {channel.mention}"
|
||||||
|
|
||||||
@@ -51,18 +50,20 @@ class CustomContext(ExecContext):
|
|||||||
"""
|
"""
|
||||||
CustomCommand ExecContext instantiated from a Twitch message
|
CustomCommand ExecContext instantiated from a Twitch message
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
bot: Bot,
|
bot: Bot,
|
||||||
message: twitchio.ChatMessage,
|
message: twitchio.ChatMessage,
|
||||||
alias: str,
|
alias: str,
|
||||||
used_prefix: str,
|
used_prefix: str,
|
||||||
**kwargs):
|
**kwargs,
|
||||||
|
):
|
||||||
super().__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
self.message = message
|
self.message = message
|
||||||
self.alias = alias
|
self.alias = alias
|
||||||
self.used_prefix = used_prefix
|
self.used_prefix = used_prefix
|
||||||
|
|
||||||
self.enter_scope(stuff)
|
self.enter_scope(stuff)
|
||||||
# TODO: Need a nice way to define some global functions. Registry pattern?
|
# TODO: Need a nice way to define some global functions. Registry pattern?
|
||||||
|
|||||||
Reference in New Issue
Block a user