0afa1ade15
* Ultroid 2025 Co-authored-by: New-dev0 <New-dev0@users.noreply.github.com> Co-authored-by: Amit Sharma <buddhhu@users.noreply.github.com> Co-authored-by: TechiError <techierror@users.noreply.github.com> Co-authored-by: Aditya <me@xditya.me> Co-authored-by: Arnab Paryali <Arnabxd@users.noreply.github.com> Co-authored-by: hellboi_atul <hellboi-atul@users.noreply.github.com> Co-authored-by: sppidy <sppidy@users.noreply.github.com>
111 lines
3.9 KiB
Python
111 lines
3.9 KiB
Python
# Ultroid - UserBot
|
|
# Copyright (C) 2021-2025 TeamUltroid
|
|
#
|
|
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
|
|
# PLease read the GNU Affero General Public License in
|
|
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
|
|
|
|
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)
|