| | |
| | |
| | |
| | |
| | |
| | |
| |
|
| | from . import get_help |
| |
|
| | __doc__ = get_help("help_afk") |
| |
|
| |
|
| | import asyncio |
| |
|
| | from telethon import events |
| |
|
| | from pyUltroid.dB.afk_db import add_afk, del_afk, is_afk |
| | from pyUltroid.dB.base import KeyManager |
| |
|
| | from . import ( |
| | LOG_CHANNEL, |
| | NOSPAM_CHAT, |
| | Redis, |
| | asst, |
| | get_string, |
| | mediainfo, |
| | udB, |
| | ultroid_bot, |
| | ultroid_cmd, |
| | upload_file |
| | ) |
| |
|
| | old_afk_msg = [] |
| |
|
| | is_approved = KeyManager("PMPERMIT", cast=list).contains |
| |
|
| |
|
| | @ultroid_cmd(pattern="afk( (.*)|$)", owner_only=True) |
| | async def set_afk(event): |
| | if event.client._bot or is_afk(): |
| | return |
| | text, media, media_type = None, None, None |
| | if event.pattern_match.group(1).strip(): |
| | text = event.text.split(maxsplit=1)[1] |
| | reply = await event.get_reply_message() |
| | if reply: |
| | if reply.text and not text: |
| | text = reply.text |
| | if reply.media: |
| | media_type = mediainfo(reply.media) |
| | if media_type.startswith(("pic", "gif")): |
| | file = await event.client.download_media(reply.media) |
| | media = upload_file(file) |
| | else: |
| | media = reply.file.id |
| | await event.eor("`Done`", time=2) |
| | add_afk(text, media_type, media) |
| | ultroid_bot.add_handler(remove_afk, events.NewMessage(outgoing=True)) |
| | ultroid_bot.add_handler( |
| | on_afk, |
| | events.NewMessage( |
| | incoming=True, func=lambda e: bool(e.mentioned or e.is_private) |
| | ), |
| | ) |
| | msg1, msg2 = None, None |
| | if text and media: |
| | if "sticker" in media_type: |
| | msg1 = await ultroid_bot.send_file(event.chat_id, file=media) |
| | msg2 = await ultroid_bot.send_message( |
| | event.chat_id, get_string("afk_5").format(text) |
| | ) |
| | else: |
| | msg1 = await ultroid_bot.send_message( |
| | event.chat_id, get_string("afk_5").format(text), file=media |
| | ) |
| | elif media: |
| | if "sticker" in media_type: |
| | msg1 = await ultroid_bot.send_file(event.chat_id, file=media) |
| | msg2 = await ultroid_bot.send_message(event.chat_id, get_string("afk_6")) |
| | else: |
| | msg1 = await ultroid_bot.send_message( |
| | event.chat_id, get_string("afk_6"), file=media |
| | ) |
| | elif text: |
| | msg1 = await event.respond(get_string("afk_5").format(text)) |
| | else: |
| | msg1 = await event.respond(get_string("afk_6")) |
| | old_afk_msg.append(msg1) |
| | if msg2: |
| | old_afk_msg.append(msg2) |
| | return await asst.send_message(LOG_CHANNEL, msg2.text) |
| | await asst.send_message(LOG_CHANNEL, msg1.text) |
| |
|
| |
|
| | async def remove_afk(event): |
| | if event.is_private and udB.get_key("PMSETTING") and not is_approved(event.chat_id): |
| | return |
| | elif "afk" in event.text.lower(): |
| | return |
| | elif event.chat_id in NOSPAM_CHAT: |
| | return |
| | if is_afk(): |
| | _, _, _, afk_time = is_afk() |
| | del_afk() |
| | off = await event.reply(get_string("afk_1").format(afk_time)) |
| | await asst.send_message(LOG_CHANNEL, get_string("afk_2").format(afk_time)) |
| | for x in old_afk_msg: |
| | try: |
| | await x.delete() |
| | except BaseException: |
| | pass |
| | await asyncio.sleep(10) |
| | await off.delete() |
| |
|
| |
|
| | async def on_afk(event): |
| | if event.is_private and Redis("PMSETTING") and not is_approved(event.chat_id): |
| | return |
| | elif "afk" in event.text.lower(): |
| | return |
| | elif not is_afk(): |
| | return |
| | if event.chat_id in NOSPAM_CHAT: |
| | return |
| | sender = await event.get_sender() |
| | if sender.bot or sender.verified: |
| | return |
| | text, media_type, media, afk_time = is_afk() |
| | msg1, msg2 = None, None |
| | if text and media: |
| | if "sticker" in media_type: |
| | msg1 = await event.reply(file=media) |
| | msg2 = await event.reply(get_string("afk_3").format(afk_time, text)) |
| | else: |
| | msg1 = await event.reply( |
| | get_string("afk_3").format(afk_time, text), file=media |
| | ) |
| | elif media: |
| | if "sticker" in media_type: |
| | msg1 = await event.reply(file=media) |
| | msg2 = await event.reply(get_string("afk_4").format(afk_time)) |
| | else: |
| | msg1 = await event.reply(get_string("afk_4").format(afk_time), file=media) |
| | elif text: |
| | msg1 = await event.reply(get_string("afk_3").format(afk_time, text)) |
| | else: |
| | msg1 = await event.reply(get_string("afk_4").format(afk_time)) |
| | for x in old_afk_msg: |
| | try: |
| | await x.delete() |
| | except BaseException: |
| | pass |
| | old_afk_msg.append(msg1) |
| | if msg2: |
| | old_afk_msg.append(msg2) |
| |
|
| |
|
| | if udB.get_key("AFK_DB"): |
| | ultroid_bot.add_handler(remove_afk, events.NewMessage(outgoing=True)) |
| | ultroid_bot.add_handler( |
| | on_afk, |
| | events.NewMessage( |
| | incoming=True, func=lambda e: bool(e.mentioned or e.is_private) |
| | ), |
| | ) |
| |
|
