Compare commits

..

2 Commits

Author SHA1 Message Date
conatum 0d4f9b9a9a fix(twitch): Fix relative imports 2026-05-31 06:32:31 +10:00
conatum 733f3e3ae6 tweak: Use Generic to support 3.11 2026-05-31 06:31:46 +10:00
4 changed files with 92 additions and 76 deletions
+43 -27
View File
@@ -1,4 +1,14 @@
from typing import Any, Awaitable, Callable, Concatenate, ParamSpec, TypeVar, Union, TYPE_CHECKING
from typing import (
Any,
Generic,
Awaitable,
Callable,
Concatenate,
ParamSpec,
TypeVar,
Union,
TYPE_CHECKING,
)
from functools import wraps, reduce
import operator
@@ -22,9 +32,9 @@ __all__ = (
"ternary_term",
"run_function",
"cascade_term",
'ExecutorTerm',
'LateTerm',
'lazy_term',
"ExecutorTerm",
"LateTerm",
"lazy_term",
)
P = ParamSpec("P")
@@ -37,7 +47,7 @@ T2 = TypeVar("T2")
TermResult = TypeVar("TermResult")
class ExecutorTerm[TermResult]:
class ExecutorTerm(Generic[TermResult]):
# May add more here for initial parsing context
def __init__(self, *args, node_text: str | None = None, **kwargs):
@@ -145,7 +155,7 @@ async def cmdarg_term(ctx, argn: int, greedy: bool) -> str:
if greedy:
return " ".join(args[argn:])
elif argn >= len(args):
return ''
return ""
else:
return args[argn]
@@ -158,33 +168,36 @@ async def cascade_term(ctx, *terms: ExecutorTerm):
return eval
return ""
@lazy_term
async def command_shell(ctx, component_term: ExecutorTerm):
if isinstance(component_term, ExecutorTerm):
result = await component_term(ctx)
else:
logger.warning(
f"command_shell LazyTerm stringifying non-term {component_term}"
)
logger.warning(f"command_shell LazyTerm stringifying non-term {component_term}")
result = component_term
return str(result)
@lazy_term
async def define_equal(ctx, var: str, value: ExecutorTerm):
ctx.this_scope[var] = await value.execute(ctx)
return None
@lazy_term
async def define_walrus(ctx, var: str, value: ExecutorTerm):
result = await value.execute(ctx)
ctx.this_scope[var] = result
return result
@lazy_term
async def define_term(ctx, var: str, value: ExecutorTerm):
ctx.this_scope[var] = value
return None
@lazy_term
async def run_function(ctx, var: ExecutorTerm, *args: ExecutorTerm):
varlue = await var.execute(ctx)
@@ -196,13 +209,14 @@ async def run_function(ctx, var: ExecutorTerm, *args: ExecutorTerm):
else:
return await varlue(ctx, *argues)
@lazy_term
async def get_var(ctx, root: str | ExecutorTerm, *attrs: str):
if isinstance(root, ExecutorTerm):
value = await root.execute(ctx)
name = [root.text or '']
name = [root.text or ""]
elif root not in ctx.scope:
return ' '.join((root, *attrs))
return " ".join((root, *attrs))
else:
value = ctx.scope[root]
name = [root]
@@ -213,39 +227,41 @@ async def get_var(ctx, root: str | ExecutorTerm, *attrs: str):
for attr in attrs:
try:
value = getattr(value, '_exec_attr_'+attr)
value = getattr(value, "_exec_attr_" + attr)
name.append(attr)
except AttributeError:
raise CommandValueError(f"{'.'.join(name)} has no attribute {attr}")
return value
@lazy_term
async def compare(ctx, term1: ExecutorTerm, op: str, term2: ExecutorTerm):
opmap = {
'==': operator.eq,
'!=': operator.ne,
'>': operator.gt,
'>=': operator.ge,
'<': operator.lt,
'<=': operator.le,
"==": operator.eq,
"!=": operator.ne,
">": operator.gt,
">=": operator.ge,
"<": operator.lt,
"<=": operator.le,
}
value1 = await term1.execute(ctx)
value2 = await term2.execute(ctx)
oppy = opmap[op]
return oppy(value1, value2)
@lazy_term
async def binop(ctx, op: str, *terms):
opmap = {
'||': operator.or_,
'&&': operator.and_,
'+': operator.add,
'-': operator.sub,
'*': operator.mul,
'**': operator.pow,
'/': operator.truediv,
'//': operator.floordiv,
'%': operator.mod,
"||": operator.or_,
"&&": operator.and_,
"+": operator.add,
"-": operator.sub,
"*": operator.mul,
"**": operator.pow,
"/": operator.truediv,
"//": operator.floordiv,
"%": operator.mod,
}
# Left associative
oppy = opmap[op]
+9 -10
View File
@@ -1,11 +1,13 @@
import twitchio
from customcmd.client import ExecClient
from customcmd.data import CustomCmd, CustomCmdData
from meta import Bot
from ..customcmd.client import ExecClient
from ..customcmd.data import CustomCmd, CustomCmdData
from .context import CustomContext, CustomCmdUser
class TwitchCmdClient(ExecClient):
def __init__(self, bot: Bot, data: CustomCmdData):
super().__init__()
@@ -53,20 +55,17 @@ class TwitchCmdClient(ExecClient):
used_prefix=prefix,
message=message,
initial_globals={
'author': CustomCmdUser(message.chatter),
'channel': CustomCmdUser(message.broadcaster),
}
"author": CustomCmdUser(message.chatter),
"channel": CustomCmdUser(message.broadcaster),
},
)
result = await self.run_command(ctx, cmd.response)
if result:
await message.broadcaster.send_message(
result,
sender=self.bot.user
)
await message.broadcaster.send_message(result, sender=self.bot.user)
async def get_prefixes(self, message: twitchio.ChatMessage):
# TODO:
return ('!',)
return ("!",)
async def get_commands_in(self, communityid: int) -> dict[str, CustomCmd]:
"""
+7 -7
View File
@@ -4,7 +4,6 @@ import twitchio
from twitchio import PartialUser
from twitchio.ext import commands as cmds
from customcmd.parser.context import ExecContext
from meta import Bot
from twitch.client import TwitchCmdClient
from utils.lib import utc_now
@@ -12,13 +11,12 @@ from utils.lib import utc_now
from . import logger
from ..customcmd.data import CustomCmd, CustomCmdData
from ..customcmd.client import ExecClient
from ..customcmd.parser import ExecContext
# TODO: Make an alphacroc or croccyalpha client for testing
# TODO: Fix the tracker not actually guarding the commands properly
class CustomCmdComponent(cmds.Component):
def __init__(self, bot: Bot):
self.bot = bot
@@ -46,7 +44,7 @@ class CustomCmdComponent(cmds.Component):
# TODO: Probably attempt to list commands or something here.
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
async def cmd_add(self, ctx: cmds.Context, name: str, *, response: str):
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)
try:
await self.cmdclient.run_command(ExecContext(args=[], content=''), response)
await self.cmdclient.run_command(ExecContext(args=[], content=""), response)
except Exception:
# TODO: Nicer errors for pinpointing problem
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
# TODO: Again, permission resolution
@@ -79,7 +79,7 @@ class CustomCmdComponent(cmds.Component):
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()
async def cmd_del(self, ctx: cmds.Context, name: str):
# Lookup the name to make sure a command does exist
+9 -8
View File
@@ -1,11 +1,11 @@
import twitchio
from customcmd.client import ExecClient
from customcmd.data import CustomCmd, CustomCmdData
from meta import Bot
from utils.lib import utc_now, strfdur
from ..customcmd.parser import ExecContext
from ..customcmd.client import ExecClient
from ..customcmd.data import CustomCmd, CustomCmdData
from .ctxvars import stuff
@@ -25,10 +25,7 @@ class CustomCmdUser:
return self.user.display_name or self.user.name or self.user.id
async def _exec_attr_whisper(self, ctx, message: str):
await self.user.send_whisper(
to_user=self.user,
message=message
)
await self.user.send_whisper(to_user=self.user, message=message)
async def _exec_attr_followage(self, ctx, channel: twitchio.PartialUser):
followed_at = None
@@ -38,7 +35,9 @@ class CustomCmdUser:
if 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:
return f"{self.user.mention} is not following {channel.mention}"
@@ -51,13 +50,15 @@ class CustomContext(ExecContext):
"""
CustomCommand ExecContext instantiated from a Twitch message
"""
def __init__(
self,
bot: Bot,
message: twitchio.ChatMessage,
alias: str,
used_prefix: str,
**kwargs):
**kwargs,
):
super().__init__(**kwargs)
self.bot = bot
self.message = message