From 0624b6b847174165b2364b249e1737911e96b2bb Mon Sep 17 00:00:00 2001 From: Rongrong Date: Sat, 17 Dec 2022 01:43:05 +0800 Subject: [PATCH] feat: multiple slashes -> random sticker --- SlashBot.py | 129 ++++++++++++++++++++++++++++++++++++++-------------- 1 file changed, 96 insertions(+), 33 deletions(-) diff --git a/SlashBot.py b/SlashBot.py index d076407..9706b32 100644 --- a/SlashBot.py +++ b/SlashBot.py @@ -6,32 +6,44 @@ import re import requests import telegram from loguru import logger as _logger -from typing import Optional, Union, Any +from typing import Optional, Union, Any, Callable from telegram.ext import Updater, MessageHandler, filters, Dispatcher from functools import partial from threading import Thread from time import sleep from itertools import product as _product +from itertools import starmap from random import Random, SystemRandom from collections import deque, Counter Filters = filters.Filters -parser = re.compile(r'^(?P[\\/]_?\$?)' - r'(?P([^\s\\]|\\.)*((?<=\S)\\)?)' - r'(\s+(?P.+))?$') -ouenParser = re.compile(r'^(' - r'\\ .* /' - r'|' - r'＼ .* ／' - r'|' - r'(\\.*/\s*){2,}' - r'|' - r'(＼.*／\s*){2,}}' - r'|' - r'[/\\＼／]{2,}' - r')$') -pinParser = re.compile(r'^[\\/]_?pin$') +parser = re.compile( + r'^(?P[\\/]_?\$?)' + r'(?P([^\s\\]|\\.)*((?<=\S)\\)?)' + r'(\s+(?P.+))?$' +) +ouenParser = re.compile( + r'^(' + r'\\ .* /' + r'|' + r'＼ .* ／' + r'|' + r'(\\.*/\s*){2,}' + r'|' + r'(＼.*／\s*){2,}}' + r'|' + r'[/\\＼／]{2,}' + r')$' +) +pinParser = re.compile( + r'^[\\/]_?pin$' +) +randomStickerParser = re.compile( + r'^([\\/])' + r'(_?)' + r'(\1\2?)+$' +) convertEscapes = partial(re.compile(r'\\(\s)').sub, r'\1') htmlEscape = lambda s: s.replace("&", "&").replace("<", "<").replace(">", ">") @@ -43,8 +55,23 @@ PUNCTUATION_TAIL = '.,?!;:~(' \ '。，？！；：～（' -class Vegetable: - _counter = Counter() +class RandomizerMeta(type): + def __init__(cls, name, bases, namespace): + super().__init__(name, bases, namespace) + cls._counter = Counter() + + +class Randomizer(metaclass=RandomizerMeta): + def __class_getitem__(cls, item: str) -> Any: + collection = getattr(cls, item) + if cls._counter[item] % len(collection) == 0: + random.shuffle(collection) + cls._counter[item] += 1 + collection.rotate() + return collection[-1] + + +class Vegetable(Randomizer): permission_denied = deque( product( {'我太菜了', '我好菜', '我好菜啊', '我菜死了'}, @@ -68,13 +95,33 @@ class Vegetable: ) ) - def __class_getitem__(cls, item: str): - collection = getattr(cls, item) - if cls._counter[item] % len(collection) == 0: - random.shuffle(collection) - cls._counter[item] += 1 - collection.rotate() - return collection[-1] + +class Stickers(Randomizer): + _cnstr = partial(telegram.Sticker, + width=512, + height=512, + is_animated=False, + is_video=False, + type=telegram.Sticker.REGULAR) + + stickers = deque( + starmap( + _cnstr, + ( + ('CAACAgUAAxkBAAInuWOcpUaH6eiL7vc9bIw6GedK-DNzAALNAQAC3x9yGXsiKdzwYgnWLAQ', 'AgADzQEAAt8fchk'), + ('CAACAgUAAxkBAAInu2OcpakDQaHnVWVo_80AAVgE7EcujAAC2gEAAt8fchkAAbkxR5lrlfUsBA', 'AgAD2gEAAt8fchk'), + ('CAACAgUAAxkBAAInvWOcpcUcJDWODiIxUoSTs840QJaFAALbAQAC3x9yGcc6smv9nZmELAQ', 'AgAD2wEAAt8fchk'), + ('CAACAgUAAxkBAAKNaWOcn1dZP5Ooe5wX8JrCKkK2qXGzAALnAQAC3x9yGfJozMIlJl_kLAQ', 'AgAD5wEAAt8fchk'), + ('CAACAgUAAxkBAAInt2Ocor-N-gnaJTGR-RtyopIgI0l5AALrAQAC3x9yGWlN_RGM1AESLAQ', 'AgAD6wEAAt8fchk'), + ('CAACAgUAAxkBAAInv2Ocph3PT5XiCArmoehOYzCn1sJRAALyAQAC3x9yGSLk1aHaA09kLAQ', 'AgAD8gEAAt8fchk'), + ('CAACAgUAAxkBAAInwWOcplEELbRuIBkuy3Yyv6YCScxdAAISAgAC3x9yGaW3ftfZrg8eLAQ', 'AgADEgIAAt8fchk'), + ('CAACAgUAAxkBAAInw2Ocpm8VgaPVTUhDPTGHWjnDjzsDAAIbAgAC3x9yGWCDLF4OLhaMLAQ', 'AgADGwIAAt8fchk'), + ) + ) + ) + + def __new__(cls) -> telegram.Sticker: + return cls.__class_getitem__('stickers') _logger.remove() @@ -85,6 +132,16 @@ _logger.add(sys.stderr, "|{message}", level="DEBUG") + +def log(func: Callable): + def wrapper(update: telegram.Update, ctx: telegram.ext.CallbackContext): + logger = ctx.bot_data['logger'] + logger.debug(str(update.to_dict())) + return func(update, ctx, logger) + + return wrapper + + try: random = SystemRandom() except NotImplementedError: @@ -238,10 +295,8 @@ def reply(update: telegram.Update, ctx: telegram.ext.CallbackContext): update.effective_message.reply_text('\u200e' + text, parse_mode='HTML') -def repeat(update: telegram.Update, ctx: telegram.ext.CallbackContext): - logger = ctx.bot_data['logger'] - logger.debug(str(update.to_dict())) - +@log +def repeat(update: telegram.Update, ctx: telegram.ext.CallbackContext, logger: _logger): chat = update.effective_chat msg = update.effective_message if chat.id < 0: @@ -251,10 +306,8 @@ def repeat(update: telegram.Update, ctx: telegram.ext.CallbackContext): msg.copy(chat.id) if chat.has_protected_content else msg.forward(chat.id) -def pin(update: telegram.Update, ctx: telegram.ext.CallbackContext): - logger = ctx.bot_data['logger'] - logger.debug(str(update.to_dict())) - +@log +def pin(update: telegram.Update, _ctx: telegram.ext.CallbackContext, logger: _logger): msg = update.effective_message msg_to_pin = msg.reply_to_message if not msg_to_pin: @@ -273,9 +326,19 @@ def pin(update: telegram.Update, ctx: telegram.ext.CallbackContext): logger.warning(vegetable) +@log +def random_sticker(update: telegram.Update, _ctx: telegram.ext.CallbackContext, logger: _logger): + msg = update.effective_message + sticker = Stickers() + logger.info(sticker) + msg.reply_sticker(sticker) + + def start(token: str): updater = Updater(token=token, use_context=True, request_kwargs={'proxy_url': TELEGRAM_PROXY}) dp: Dispatcher = updater.dispatcher + dp.add_handler(MessageHandler(Filters.regex(randomStickerParser) & ~Filters.update.edited_message, random_sticker, + run_async=True)) dp.add_handler(MessageHandler(Filters.regex(pinParser) & ~Filters.update.edited_message, pin, run_async=True)) dp.add_handler(MessageHandler(Filters.regex(ouenParser) & ~Filters.update.edited_message, repeat, run_async=True)) dp.add_handler(MessageHandler(Filters.regex(parser) & ~Filters.update.edited_message, reply, run_async=True))