This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
Telegram/modules/handlers/everything.py

117 lines
7.4 KiB
Python
Raw Normal View History

from app import app, isAnAdmin
2022-12-05 19:49:51 +02:00
import asyncio
from pyrogram import filters
2023-01-04 21:13:29 +02:00
from pyrogram.types import Message, ReplyKeyboardMarkup, ReplyKeyboardRemove, ForceReply, InlineKeyboardMarkup, InlineKeyboardButton
2022-12-27 14:36:54 +02:00
from pyrogram.client import Client
2022-12-13 15:24:31 +02:00
from classes.holo_user import HoloUser
2023-01-04 21:13:29 +02:00
from modules.utils import configGet, logWrite, locale, all_locales
from modules.database import col_messages, col_spoilers
2022-12-11 19:50:50 +02:00
async def message_involved(msg: Message) -> bool:
message = col_messages.find_one({"destination.id": msg.reply_to_message.id, "destination.chat": msg.reply_to_message.chat.id})
if message is not None:
return True
return False
2022-12-11 19:50:50 +02:00
async def message_context(msg: Message) -> tuple:
message = col_messages.find_one({"destination.id": msg.reply_to_message.id, "destination.chat": msg.reply_to_message.chat.id})
if message is not None:
return message["origin"]["chat"], message["origin"]["id"]
return 0, 0
2022-12-05 19:49:51 +02:00
# Any other input ==============================================================================================================
2022-12-13 15:24:31 +02:00
@app.on_message(~ filters.scheduled & filters.private)
2022-12-27 14:36:54 +02:00
async def any_stage(app: Client, msg: Message):
2022-12-13 15:24:31 +02:00
if msg.via_bot is None:
2022-12-14 14:58:06 +02:00
holo_user = HoloUser(msg.from_user)
if (msg.reply_to_message is not None) and (await message_involved(msg)):
2022-12-13 15:24:31 +02:00
context = await message_context(msg)
context_message = await app.get_messages(context[0], context[1])
destination_user = HoloUser(context_message.from_user)
await destination_user.message(
origin=context_message,
context=msg,
text=msg.text,
caption=msg.caption,
photo=msg.photo,
video=msg.video,
file=msg.document,
adm_origin=await isAnAdmin(context_message.from_user.id),
adm_context=await isAnAdmin(msg.from_user.id)
)
return
2022-12-12 15:35:11 +02:00
2022-12-21 16:23:36 +02:00
if msg.text is not None:
2023-01-03 21:34:13 +02:00
if configGet("enabled", "features", "applications") is True:
await holo_user.application_next(msg.text, msg=msg)
if configGet("enabled", "features", "sponsorships") is True:
await holo_user.sponsorship_next(msg.text, msg=msg)
2023-01-04 21:13:29 +02:00
if holo_user.application_state()[0] != "fill" and holo_user.sponsorship_state()[0] != "fill":
spoiler = col_spoilers.find_one( {"user": msg.from_user.id, "completed": False} )
if spoiler is None:
return
if msg.photo is not None:
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"photo": msg.photo.file_id}} )
await msg.reply_text(locale("spoiler_send_description", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("spoiler_description", "keyboard"), resize_keyboard=True, one_time_keyboard=True, placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
2023-01-04 22:55:50 +02:00
logWrite(f"Adding photo with id {msg.photo.file_id} to {msg.from_user.id}'s spoiler")
2023-01-04 21:13:29 +02:00
return
2023-01-04 22:55:50 +02:00
if msg.video is not None:
2023-01-04 21:13:29 +02:00
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"video": msg.video.file_id}} )
await msg.reply_text(locale("spoiler_send_description", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("spoiler_description", "keyboard"), resize_keyboard=True, one_time_keyboard=True, placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
2023-01-04 22:55:50 +02:00
logWrite(f"Adding video with id {msg.video.file_id} to {msg.from_user.id}'s spoiler")
2023-01-04 21:13:29 +02:00
return
2023-01-04 22:55:50 +02:00
if msg.animation is not None:
2023-01-04 21:13:29 +02:00
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"animation": msg.animation.file_id}} )
await msg.reply_text(locale("spoiler_send_description", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("spoiler_description", "keyboard"), resize_keyboard=True, one_time_keyboard=True, placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
2023-01-04 22:55:50 +02:00
logWrite(f"Adding animation with id {msg.animation.file_id} to {msg.from_user.id}'s spoiler")
2023-01-04 21:13:29 +02:00
return
if spoiler["photo"] is None and spoiler["video"] is None and spoiler["animation"] is None and spoiler["text"] is None:
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"text": msg.text}} )
await msg.reply_text(locale("spoiler_send_description", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("spoiler_description", "keyboard"), resize_keyboard=True, one_time_keyboard=True, placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
2023-01-04 22:55:50 +02:00
logWrite(f"Adding text '{msg.text}' to {msg.from_user.id}'s spoiler")
2023-01-04 21:13:29 +02:00
else:
for lc in all_locales("spoiler_description", "keyboard"):
if msg.text == lc[-1][0]:
await msg.reply_text(locale("spoiler_description_enter", "message", locale=msg.from_user), reply_markup=ForceReply(placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
return
if msg.text != "-":
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"description": msg.text, "completed": True}} )
else:
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"description": None, "completed": True}} )
2023-01-04 22:55:50 +02:00
logWrite(f"Adding description '{msg.text}' to {msg.from_user.id}'s spoiler")
2023-01-04 21:13:29 +02:00
await msg.reply_text(locale("spoiler_using_description", "message", locale=msg.from_user).format(msg.text), reply_markup=ReplyKeyboardRemove())
await msg.reply_text(locale("spoiler_ready", "message", locale=msg.from_user), quote=False, reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(locale("spoiler_send", "button", locale=msg.from_user), switch_inline_query=f"spoiler:{spoiler['_id'].__str__()}")]]))
2022-12-14 14:58:06 +02:00
2022-12-05 19:49:51 +02:00
@app.on_message(~ filters.scheduled & filters.group)
2022-12-27 14:36:54 +02:00
async def message_in_group(app: Client, msg: Message):
2022-12-05 19:49:51 +02:00
if (msg.chat is not None) and (msg.via_bot is not None):
2023-01-04 20:59:09 +02:00
if (msg.via_bot.id == (await app.get_me()).id) and (msg.chat.id == configGet("users", "groups")):
2023-01-04 22:55:50 +02:00
if msg.text.startswith(locale("spoiler_described", "message").split()[0]) or msg.text.startswith(locale("spoiler_empty", "message").split()[0]):
try:
await msg.forward(configGet("users", "groups"), disable_notification=True)
except:
pass
return
2022-12-05 19:49:51 +02:00
if configGet("remove_application_time") > 0:
logWrite(f"User {msg.from_user.id} requested application in destination group, removing in {configGet('remove_application_time')} minutes")
await asyncio.sleep(configGet("remove_application_time")*60)
await msg.delete()
logWrite(f"Removed application requested by {msg.from_user.id} in destination group")
# ==============================================================================================================================