forked from kate/uniborg
Merge pull request #2 from Lonami/master
Add more known sedbots to plugins/re
This commit is contained in:
commit
cf36b5bd4a
|
@ -3,6 +3,13 @@ import re
|
||||||
|
|
||||||
import regex
|
import regex
|
||||||
from telethon import events, utils
|
from telethon import events, utils
|
||||||
|
from telethon.tl import types, functions
|
||||||
|
|
||||||
|
|
||||||
|
KNOWN_RE_BOTS = re.compile(
|
||||||
|
r'(regex|moku|BananaButler_)bot',
|
||||||
|
flags=re.IGNORECASE
|
||||||
|
)
|
||||||
|
|
||||||
# Heavily based on
|
# Heavily based on
|
||||||
# https://github.com/SijmenSchoon/regexbot/blob/master/regexbot.py
|
# https://github.com/SijmenSchoon/regexbot/blob/master/regexbot.py
|
||||||
|
@ -54,8 +61,14 @@ def doit(chat_id, match, original):
|
||||||
|
|
||||||
|
|
||||||
async def group_has_regex(group):
|
async def group_has_regex(group):
|
||||||
return any(getattr(x, 'username', None) == 'regexbot'
|
if isinstance(group, types.InputPeerChannel):
|
||||||
for x in await borg.get_participants(group, search='@regexbot'))
|
full = await borg(functions.channels.GetFullChannelRequest(group))
|
||||||
|
elif isinstance(group, types.InputPeerChat):
|
||||||
|
full = await borg(functions.messages.GetFullChatRequest(group.chat_id))
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
return any(KNOWN_RE_BOTS.match(x.username or '') for x in full.users)
|
||||||
|
|
||||||
|
|
||||||
@borg.on(events.NewMessage)
|
@borg.on(events.NewMessage)
|
||||||
|
|
Loading…
Reference in New Issue