overspend1/Moon-UB-overver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

set custom afk & pm permit (#105)

Browse Source 
* set custom afk message

* set custom antipm message
This commit is contained in:
Abhi
2024-11-30 05:50:16 +05:30
committed by GitHub
parent 36fd906bb2
commit 5fdbe84aed
2 changed files with 95 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
modules/afk.py
View File

@@ -23,6 +23,7 @@ from pyrogram.types import Message
from utils.misc import modules_help, prefix
from utils.scripts import ReplyCheck
from utils.db import db
# Variables
AFK = False
@@ -59,13 +60,18 @@ async def collect_afk_messages(bot: Client, message: Message):
CHAT_TYPE = GROUPS if is_group else USERS
if GetChatID(message) not in CHAT_TYPE:
text = (
f"<b>Beep boop. This is an automated message.\n"
f"I am not available right now.\n"
f"Last seen: {last_seen}\n"
f"Reason: <code>{AFK_REASON.upper()}</code>\n"
f"See you after I'm done doing whatever I'm doing.</b>"
)
text = db.get("core.afk", "afk_msg", None)
if text is None:
text = (
f"<b>Beep boop. This is an automated message.\n"
f"I am not available right now.\n"
f"Last seen: {last_seen}\n"
f"Reason: <code>{AFK_REASON.upper()}</code>\n"
f"See you after I'm done doing whatever I'm doing.</b>"
)
else:
last_seen = last_seen.replace("ago", "").strip()
text = text.format(last_seen=last_seen, reason=AFK_REASON)
await bot.send_message(
chat_id=GetChatID(message),
text=text,
@@ -144,6 +150,39 @@ async def afk_unset(_, message: Message):
await message.delete()
@Client.on_message(filters.command("setafkmsg", prefix) & filters.me, group=3)
async def set_afk_msg(_, message: Message):
if not message.reply_to_message:
return await message.edit("Reply to a message to set it as your AFK message.")
msg = message.reply_to_message
afk_msg = msg.text or msg.caption
if not afk_msg:
return await message.edit(
"Reply to a text or caption message to set it as your AFK message."
)
if len(afk_msg) > 200:
return await message.edit(
"AFK message is too long. It should be less than 200 characters."
)
if "{reason}" not in afk_msg:
return await message.edit(
"AFK message should contain <code>{reason}</code> to indicate where the reason will be placed."
)
if "{last_seen}" not in afk_msg:
return await message.edit(
"AFK message should contain <code>{last_seen}</code> to indicate where the last seen time will be placed."
)
old_afk_msg = db.get("core.afk", "afk_msg", None)
if old_afk_msg:
db.remove("core.afk", "afk_msg")
db.set("core.afk", "afk_msg", afk_msg)
await message.edit(f"AFK message set to:\n\n{afk_msg}")
@Client.on_message(filters.me, group=3)
async def auto_afk_unset(_, message: Message):
global AFK, AFK_TIME, AFK_REASON, USERS, GROUPS
@@ -166,4 +205,5 @@ async def auto_afk_unset(_, message: Message):
modules_help["afk"] = {
"afk [reason]": "Go to AFK mode with reason as anything after .afk\nUsage: <code>.afk <reason></code>",
"unafk": "Get out of AFK",
"setafkmsg [reply to message]*": "Set your AFK message. Use <code>{reason}</code> and <code>{last_seen}</code> to indicate where the reason and last seen time will be placed.",
}

50
modules/antipm.py
View File

@@ -52,14 +52,18 @@ async def anti_pm_handler(client: Client, message: Message):
user = await client.get_users(ids)
u_f = user.first_name
user_info = await client.resolve_peer(ids)
default_text = f"""<b>Hello, {u_f}!
default_text = db.get("core.antipm", "antipm_msg", None)
if default_text is None:
default_text = f"""<b>Hello, {u_f}!
This is the Assistant Of {u_n}.</b>
<i>My Boss is away or busy as of now, You can wait for him to respond.
Do not spam further messages else I may have to block you!</i>
<b>This is an automated message by the assistant.</b>
<b><u>Currently You Have <code>{warns}</code> Warnings.</u></b>
"""
"""
else:
default_text = default_text.format(user=u_f, my_name=u_n, warns=warns)
if db.get("core.antipm", "spamrep", False):
await client.invoke(functions.messages.ReportSpam(peer=user_info))
if db.get("core.antipm", "block", False):
@@ -179,10 +183,52 @@ async def del_contact(_, message: Message):
await message.edit("User DisApproved!")
@Client.on_message(filters.command(["setantipmmsg", "sam"], prefix) & filters.me)
async def set_antipm_msg(_, message: Message):
if not message.reply_to_message:
return await message.edit(
"Reply to a message to set it as your antipm message."
)
msg = message.reply_to_message
afk_msg = msg.text or msg.caption
if not afk_msg:
return await message.edit(
"Reply to a text or caption message to set it as your antipm message."
)
if len(afk_msg) > 200:
return await message.edit(
"antipm message is too long. It should be less than 200 characters."
)
if "{user}" not in afk_msg:
return await message.edit(
"antipm message must contain <code>{user}</code> to mention the user."
)
if "{my_name}" not in afk_msg:
return await message.edit(
"antipm message must contain <code>{my_name}</code> to mention your name."
)
if "{warns}" not in afk_msg:
return await message.edit(
"antipm message must contain <code>{warns}</code> to mention the warns count."
)
old_afk_msg = db.get("core.antipm", "antipm_msg", None)
if old_afk_msg:
db.remove("core.antipm", "antipm_msg")
db.set("core.antipm", "antipm_msg", afk_msg)
await message.edit(f"antipm message set to:\n\n{afk_msg}")
modules_help["antipm"] = {
"antipm [enable|disable]*": "Enable Pm permit",
"antipm_report [enable|disable]*": "Enable spam reporting",
"antipm_block [enable|disable]*": "Enable user blocking",
"setantipmmsg [reply to message]*": "Set antipm message. Use {user} to mention the user and {my_name} to mention your name and {warns} to mention the warns count.",
"sam [reply to message]*": "Set antipm message. Use {user} to mention the user and {my_name} to mention your name and {warns} to mention the warns count.",
"a": "Approve User",
"d": "DisApprove User",
}