|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from . import get_help |
|
|
|
|
|
__doc__ = get_help("help_filter") |
|
|
|
|
|
import os |
|
|
import re |
|
|
|
|
|
from telethon.tl.types import User |
|
|
from telethon.utils import pack_bot_file_id |
|
|
|
|
|
from pyUltroid.dB.filter_db import add_filter, get_filter, list_filter, rem_filter |
|
|
from pyUltroid.fns.tools import create_tl_btn, format_btn, get_msg_button |
|
|
|
|
|
from . import events, get_string, mediainfo, udB, ultroid_bot, ultroid_cmd, upload_file |
|
|
from ._inline import something |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="addfilter( (.*)|$)") |
|
|
async def af(e): |
|
|
wrd = (e.pattern_match.group(1).strip()).lower() |
|
|
wt = await e.get_reply_message() |
|
|
chat = e.chat_id |
|
|
if not (wt and wrd): |
|
|
return await e.eor(get_string("flr_1")) |
|
|
btn = format_btn(wt.buttons) if wt.buttons else None |
|
|
if wt and wt.media: |
|
|
wut = mediainfo(wt.media) |
|
|
if wut.startswith(("pic", "gif")): |
|
|
dl = await wt.download_media() |
|
|
m = upload_file(dl) |
|
|
os.remove(dl) |
|
|
elif wut == "video": |
|
|
if wt.media.document.size > 8 * 1000 * 1000: |
|
|
return await e.eor(get_string("com_4"), time=5) |
|
|
dl = await wt.download_media() |
|
|
m = upload_file(dl) |
|
|
os.remove(dl) |
|
|
else: |
|
|
m = pack_bot_file_id(wt.media) |
|
|
if wt.text: |
|
|
txt = wt.text |
|
|
if not btn: |
|
|
txt, btn = get_msg_button(wt.text) |
|
|
add_filter(chat, wrd, txt, m, btn) |
|
|
else: |
|
|
add_filter(chat, wrd, None, m, btn) |
|
|
else: |
|
|
txt = wt.text |
|
|
if not btn: |
|
|
txt, btn = get_msg_button(wt.text) |
|
|
add_filter(chat, wrd, txt, None, btn) |
|
|
await e.eor(get_string("flr_4").format(wrd)) |
|
|
ultroid_bot.add_handler(filter_func, events.NewMessage()) |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="remfilter( (.*)|$)") |
|
|
async def rf(e): |
|
|
wrd = (e.pattern_match.group(1).strip()).lower() |
|
|
chat = e.chat_id |
|
|
if not wrd: |
|
|
return await e.eor(get_string("flr_3")) |
|
|
rem_filter(int(chat), wrd) |
|
|
await e.eor(get_string("flr_5").format(wrd)) |
|
|
|
|
|
|
|
|
@ultroid_cmd(pattern="listfilter$") |
|
|
async def lsnote(e): |
|
|
if x := list_filter(e.chat_id): |
|
|
sd = "Filters Found In This Chats Are\n\n" |
|
|
return await e.eor(sd + x) |
|
|
await e.eor(get_string("flr_6")) |
|
|
|
|
|
|
|
|
async def filter_func(e): |
|
|
if isinstance(e.sender, User) and e.sender.bot: |
|
|
return |
|
|
xx = (e.text).lower() |
|
|
chat = e.chat_id |
|
|
if x := get_filter(chat): |
|
|
for c in x: |
|
|
pat = r"( |^|[^\w])" + re.escape(c) + r"( |$|[^\w])" |
|
|
if re.search(pat, xx): |
|
|
if k := x.get(c): |
|
|
msg = k["msg"] |
|
|
media = k["media"] |
|
|
if k.get("button"): |
|
|
btn = create_tl_btn(k["button"]) |
|
|
return await something(e, msg, media, btn) |
|
|
await e.reply(msg, file=media) |
|
|
|
|
|
|
|
|
if udB.get_key("FILTERS"): |
|
|
ultroid_bot.add_handler(filter_func, events.NewMessage()) |
|
|
|
