|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from datetime import datetime as dt |
|
|
|
|
|
from telethon.events import NewMessage |
|
|
from telethon.tl.types import ( |
|
|
Message, |
|
|
MessageEntityMention, |
|
|
MessageEntityMentionName, |
|
|
User, |
|
|
) |
|
|
from telethon.utils import get_display_name |
|
|
|
|
|
from pyUltroid.fns.helper import inline_mention, time_formatter |
|
|
|
|
|
from . import asst, asst_cmd |
|
|
|
|
|
AFK = {} |
|
|
|
|
|
|
|
|
@asst_cmd(pattern="afk", func=lambda x: not x.is_private) |
|
|
async def go_afk(event): |
|
|
sender = await event.get_sender() |
|
|
if (not isinstance(sender, User)) or sender.bot: |
|
|
return |
|
|
try: |
|
|
reason = event.text.split(" ", maxsplit=1)[1] |
|
|
except IndexError: |
|
|
reason = None |
|
|
if event.is_reply and not reason: |
|
|
replied = await event.get_reply_message() |
|
|
if not reason and replied.text and not replied.media: |
|
|
reason = replied.text |
|
|
else: |
|
|
reason = replied |
|
|
time_ = dt.now() |
|
|
if AFK.get(event.chat_id): |
|
|
AFK[event.chat_id].update({event.sender_id: {"reason": reason, "time": time_}}) |
|
|
else: |
|
|
AFK.update( |
|
|
{event.chat_id: {event.sender_id: {"reason": reason, "time": time_}}} |
|
|
) |
|
|
mention = inline_mention(sender) |
|
|
msg = f"**{mention} went AFK Now!**" |
|
|
if reason and not isinstance(reason, str): |
|
|
await event.reply(reason) |
|
|
else: |
|
|
msg += f"\n\n**Reason : ** `{reason}`" |
|
|
await event.reply(msg) |
|
|
|
|
|
|
|
|
@asst.on(NewMessage(func=lambda x: AFK.get(x.chat_id) and not x.is_private)) |
|
|
async def make_change(event): |
|
|
if event.text.startswith("/afk"): |
|
|
return |
|
|
sender = await event.get_sender() |
|
|
if (not isinstance(sender, User)) or sender.bot: |
|
|
return |
|
|
chat_ = AFK[event.chat_id] |
|
|
if event.sender_id in chat_.keys(): |
|
|
name = get_display_name(event.sender) |
|
|
cha_send = chat_[event.sender_id] |
|
|
time_ = time_formatter((dt.now() - cha_send["time"]).seconds * 1000) |
|
|
msg = f"**{name}** is No Longer AFK!\n**Was AFK for** {time_}" |
|
|
await event.reply(msg) |
|
|
del chat_[event.sender_id] |
|
|
if not chat_: |
|
|
del AFK[event.chat_id] |
|
|
ST_SPAM = [] |
|
|
replied = await event.get_reply_message() |
|
|
if replied: |
|
|
name = get_display_name(replied.sender) |
|
|
if replied.sender_id in chat_.keys(): |
|
|
s_der = chat_[replied.sender_id] |
|
|
res_ = s_der["reason"] |
|
|
time_ = time_formatter((dt.now() - s_der["time"]).seconds * 1000) |
|
|
msg = f"**{name}** is AFK Currently!\n**From :** {time_}" |
|
|
if res_ and isinstance(res_, str): |
|
|
msg += f"\n**Reason :** {res_}" |
|
|
elif res_ and isinstance(res_, Message): |
|
|
await event.reply(res_) |
|
|
await event.reply(msg) |
|
|
ST_SPAM.append(replied.sender_id) |
|
|
for ent, text in event.get_entities_text(): |
|
|
dont_send, entity = None, None |
|
|
if isinstance(ent, MessageEntityMentionName): |
|
|
c_id = ent.user_id |
|
|
elif isinstance(ent, MessageEntityMention): |
|
|
c_id = text |
|
|
else: |
|
|
c_id = None |
|
|
if c_id: |
|
|
entity = await event.client.get_entity(c_id) |
|
|
if entity and entity.id in chat_.keys() and entity.id not in ST_SPAM: |
|
|
ST_SPAM.append(entity.id) |
|
|
s_der = chat_[entity.id] |
|
|
name = get_display_name(entity) |
|
|
res_ = s_der["reason"] |
|
|
time_ = time_formatter((dt.now() - s_der["time"]).seconds * 1000) |
|
|
msg = f"**{name}** is AFK Currently!\n**From :** {time_}" |
|
|
if res_ and isinstance(res_, str): |
|
|
msg += f"\n**Reason :** {res_}" |
|
|
elif res_ and isinstance(res_, Message): |
|
|
await event.reply(res_) |
|
|
await event.reply(msg) |
|
|
|
