Changed code style to black
This commit is contained in:
parent
5b8951fe07
commit
3304339244
@ -8,21 +8,33 @@ makedirs(f'{configGet("cache", "locations")}{sep}avatars', exist_ok=True)
|
||||
|
||||
app = FastAPI(title="HoloUA Avatars API", docs_url=None, redoc_url=None, version="1.0")
|
||||
|
||||
|
||||
@app.get("/check", response_class=JSONResponse, include_in_schema=False)
|
||||
@app.head("/check", response_class=JSONResponse, include_in_schema=False)
|
||||
async def check():
|
||||
return JSONResponse({"detail": "I'm alright, thank you"})
|
||||
|
||||
|
||||
@app.get("/", response_class=FileResponse, include_in_schema=False)
|
||||
async def avatar_get(avatar_id: str):
|
||||
if path.exists(f'{configGet("cache", "locations")}{sep}avatars{sep}{avatar_id}'):
|
||||
return FileResponse(f'{configGet("cache", "locations")}{sep}avatars{sep}{avatar_id}', media_type="image/jpg")
|
||||
return FileResponse(
|
||||
f'{configGet("cache", "locations")}{sep}avatars{sep}{avatar_id}',
|
||||
media_type="image/jpg",
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="File not found")
|
||||
|
||||
|
||||
@app.head("/", response_class=Response, include_in_schema=False)
|
||||
async def avatar_head(avatar_id: str):
|
||||
if path.exists(f'{configGet("cache", "locations")}{sep}avatars{sep}{avatar_id}'):
|
||||
return Response(headers={"Content-Length": path.getsize(f'{configGet("cache", "locations")}{sep}avatars{sep}{avatar_id}')})
|
||||
return Response(
|
||||
headers={
|
||||
"Content-Length": path.getsize(
|
||||
f'{configGet("cache", "locations")}{sep}avatars{sep}{avatar_id}'
|
||||
)
|
||||
}
|
||||
)
|
||||
else:
|
||||
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="File not found")
|
||||
raise HTTPException(status_code=HTTP_404_NOT_FOUND, detail="File not found")
|
||||
|
19
app.py
19
app.py
@ -5,10 +5,15 @@ from modules.utils import configGet, jsonLoad
|
||||
from pyrogram.client import Client
|
||||
from pyrogram.errors import bad_request_400
|
||||
|
||||
app = Client("holochecker", bot_token=configGet("bot_token", "bot"), api_id=configGet("api_id", "bot"), api_hash=configGet("api_hash", "bot"))
|
||||
app = Client(
|
||||
"holochecker",
|
||||
bot_token=configGet("bot_token", "bot"),
|
||||
api_id=configGet("api_id", "bot"),
|
||||
api_hash=configGet("api_hash", "bot"),
|
||||
)
|
||||
|
||||
|
||||
async def isAnAdmin(admin_id):
|
||||
|
||||
# Check if user is mentioned in config
|
||||
if (admin_id == configGet("owner")) or (admin_id in configGet("admins")):
|
||||
return True
|
||||
@ -19,14 +24,16 @@ async def isAnAdmin(admin_id):
|
||||
return True if admin_id in jsonLoad(f"cache{sep}admins") else False
|
||||
except (FileNotFoundError, JSONDecodeError):
|
||||
pass
|
||||
|
||||
|
||||
# Check if user is in admin group
|
||||
try:
|
||||
async for member in app.get_chat_members(configGet("admin", "groups")):
|
||||
if member.user.id == admin_id:
|
||||
return True
|
||||
except bad_request_400.ChannelInvalid:
|
||||
logWrite(f"Could not get users in admin group to answer isAnAdmin(). Bot is likely not in the group.")
|
||||
logWrite(
|
||||
f"Could not get users in admin group to answer isAnAdmin(). Bot is likely not in the group."
|
||||
)
|
||||
return False
|
||||
|
||||
return False
|
||||
|
||||
return False
|
||||
|
@ -1,5 +1,8 @@
|
||||
class PlaceNotFoundError(Exception):
|
||||
"""Query provided did not lead to any city or populated area"""
|
||||
"""Query provided did not lead to any city or populated area"""
|
||||
|
||||
def __init__(self, query):
|
||||
self.query = query
|
||||
super().__init__(f"Could not find any place on geonames.org of feature classes A and P by query '{self.query}'")
|
||||
super().__init__(
|
||||
f"Could not find any place on geonames.org of feature classes A and P by query '{self.query}'"
|
||||
)
|
||||
|
@ -1,24 +1,38 @@
|
||||
"""Exceptions that are meant to be used by HoloUser class
|
||||
and other modules that handle those exceptions"""
|
||||
|
||||
|
||||
class UserNotFoundError(Exception):
|
||||
"""HoloUser could not find user with such an ID in database"""
|
||||
|
||||
def __init__(self, user, user_id):
|
||||
self.user = user
|
||||
self.user_id = user_id
|
||||
super().__init__(f"User of type {type(self.user)} with id {self.user_id} was not found")
|
||||
super().__init__(
|
||||
f"User of type {type(self.user)} with id {self.user_id} was not found"
|
||||
)
|
||||
|
||||
|
||||
class UserInvalidError(Exception):
|
||||
"""Provided to HoloUser object is not supported"""
|
||||
"""Provided to HoloUser object is not supported"""
|
||||
|
||||
def __init__(self, user):
|
||||
self.user = user
|
||||
super().__init__(f"Could not find HoloUser by using {type(self.user)} as an input type")
|
||||
super().__init__(
|
||||
f"Could not find HoloUser by using {type(self.user)} as an input type"
|
||||
)
|
||||
|
||||
|
||||
class LabelTooLongError(Exception):
|
||||
def __init__(self, label: str) -> None:
|
||||
self.label = label
|
||||
super().__init__(f"Could not set label to '{label}' because it is {len(label)} characters long (16 is maximum)")
|
||||
super().__init__(
|
||||
f"Could not set label to '{label}' because it is {len(label)} characters long (16 is maximum)"
|
||||
)
|
||||
|
||||
|
||||
class LabelSettingError(Exception):
|
||||
def __init__(self, exp: Exception, trace: str) -> None:
|
||||
super().__init__(f"❌ **Could not set label**\n\nException: `{exp}`\n\n**Traceback:**\n```\n{trace}\n```")
|
||||
super().__init__(
|
||||
f"❌ **Could not set label**\n\nException: `{exp}`\n\n**Traceback:**\n```\n{trace}\n```"
|
||||
)
|
||||
|
@ -4,21 +4,47 @@ from ftfy import fix_text
|
||||
from traceback import format_exc
|
||||
from app import app, isAnAdmin
|
||||
from typing import Any, List, Literal, Union
|
||||
from pyrogram.types import User, ChatMember, ChatPrivileges, Chat, Message, Photo, Video, Document, Animation, Voice, ForceReply, ReplyKeyboardMarkup
|
||||
from pyrogram.types import (
|
||||
User,
|
||||
ChatMember,
|
||||
ChatPrivileges,
|
||||
Chat,
|
||||
Message,
|
||||
Photo,
|
||||
Video,
|
||||
Document,
|
||||
Animation,
|
||||
Voice,
|
||||
ForceReply,
|
||||
ReplyKeyboardMarkup,
|
||||
)
|
||||
from pyrogram.errors import bad_request_400
|
||||
from dateutil.relativedelta import relativedelta
|
||||
from classes.errors.geo import PlaceNotFoundError
|
||||
from classes.errors.holo_user import UserInvalidError, UserNotFoundError, LabelTooLongError, LabelSettingError
|
||||
from classes.errors.holo_user import (
|
||||
UserInvalidError,
|
||||
UserNotFoundError,
|
||||
LabelTooLongError,
|
||||
LabelSettingError,
|
||||
)
|
||||
from classes.templates import DefaultApplicationTemp, DefaultSponsorshipTemp
|
||||
from modules.database import col_tmp, col_users, col_applications, col_sponsorships, col_messages, col_spoilers
|
||||
from modules.database import (
|
||||
col_tmp,
|
||||
col_users,
|
||||
col_applications,
|
||||
col_sponsorships,
|
||||
col_messages,
|
||||
col_spoilers,
|
||||
)
|
||||
from modules.logging import logWrite
|
||||
from modules.utils import configGet, find_location, locale, should_quote
|
||||
|
||||
class HoloUser():
|
||||
|
||||
class HoloUser:
|
||||
"""This object represents a user of HoloChecker bot.
|
||||
It is primarily used to interact with a database in a more python-friendly way,
|
||||
as well as provide better programming experience in case of adding new features, etc.
|
||||
"""
|
||||
"""
|
||||
|
||||
def __init__(self, user: Union[User, List[User], ChatMember, int]) -> None:
|
||||
"""A user of Holo bot. Used to simplify DB interaction.
|
||||
@ -29,7 +55,7 @@ class HoloUser():
|
||||
### Raises:
|
||||
* UserInvalidError: Provided to `HoloUser` object is not supported
|
||||
* UserNotFoundError: `HoloUser` could not find user with such an ID in database
|
||||
"""
|
||||
"""
|
||||
|
||||
# Determine input object class and extract id
|
||||
if isinstance(user, list) and len(user) != 0:
|
||||
@ -67,34 +93,52 @@ class HoloUser():
|
||||
self.username = holo_user["tg_username"]
|
||||
|
||||
if isinstance(user, User):
|
||||
|
||||
if (self.name != user.first_name) and hasattr(user, "first_name") and (user.first_name is not None):
|
||||
if (
|
||||
(self.name != user.first_name)
|
||||
and hasattr(user, "first_name")
|
||||
and (user.first_name is not None)
|
||||
):
|
||||
self.set("name", user.first_name, db_key="tg_name")
|
||||
|
||||
if (self.phone != user.phone_number) and hasattr(user, "phone") and (user.phone_number is not None):
|
||||
if (
|
||||
(self.phone != user.phone_number)
|
||||
and hasattr(user, "phone")
|
||||
and (user.phone_number is not None)
|
||||
):
|
||||
self.set("phone", user.phone_number, db_key="tg_phone")
|
||||
|
||||
if (self.locale != user.language_code) and hasattr(user, "locale") and (user.language_code is not None):
|
||||
if (
|
||||
(self.locale != user.language_code)
|
||||
and hasattr(user, "locale")
|
||||
and (user.language_code is not None)
|
||||
):
|
||||
self.set("locale", user.language_code, db_key="tg_locale")
|
||||
|
||||
if (self.username != user.username) and hasattr(user, "username") and (user.username is not None):
|
||||
if (
|
||||
(self.username != user.username)
|
||||
and hasattr(user, "username")
|
||||
and (user.username is not None)
|
||||
):
|
||||
self.set("username", user.username, db_key="tg_username")
|
||||
|
||||
|
||||
def set(self, key: str, value: Any, db_key: Union[str, None] = None) -> None:
|
||||
"""Set attribute data and save it into database
|
||||
|
||||
### Args:
|
||||
* `key` (`str`): Attribute to be changed
|
||||
* `value` (`Any`): Value to set
|
||||
"""
|
||||
"""
|
||||
if not hasattr(self, key):
|
||||
raise AttributeError()
|
||||
setattr(self, key, value)
|
||||
db_key = key if db_key is None else db_key
|
||||
col_users.update_one(filter={"_id": self.db_id}, update={ "$set": { db_key: value } }, upsert=True)
|
||||
col_users.update_one(
|
||||
filter={"_id": self.db_id}, update={"$set": {db_key: value}}, upsert=True
|
||||
)
|
||||
logWrite(f"Set attribute {key} of user {self.id} to {value}")
|
||||
|
||||
async def message(self,
|
||||
async def message(
|
||||
self,
|
||||
context: Message,
|
||||
origin: Union[Message, None] = None,
|
||||
text: Union[str, None] = None,
|
||||
@ -105,7 +149,7 @@ class HoloUser():
|
||||
animation: Union[str, Animation, None] = None,
|
||||
voice: Union[str, Voice, None] = None,
|
||||
adm_origin: bool = False,
|
||||
adm_context: bool = False
|
||||
adm_context: bool = False,
|
||||
) -> None:
|
||||
"""Send a message to user
|
||||
|
||||
@ -121,7 +165,7 @@ class HoloUser():
|
||||
* voice (`Union[str, Voice, None]`, *optional*): Voice as a voice object or file_id as a string. Defaults to None.
|
||||
* adm_origin (`bool`, *optional*): Whether origin sender is an admin. Defaults to False.
|
||||
* adm_context (`bool`, *optional*): Whether context sender is an admin. Defaults to False.
|
||||
"""
|
||||
"""
|
||||
|
||||
if text is not None:
|
||||
text = fix_text(text)
|
||||
@ -130,18 +174,30 @@ class HoloUser():
|
||||
|
||||
# Check if any text available and log message sending
|
||||
if text is not None:
|
||||
logWrite(f"{context.from_user.id} sent message '{text}' to {self.id} (source message: {context.id})")
|
||||
logWrite(
|
||||
f"{context.from_user.id} sent message '{text}' to {self.id} (source message: {context.id})"
|
||||
)
|
||||
elif caption is not None:
|
||||
logWrite(f"{context.from_user.id} sent message '{caption}' to {self.id} (source message: {context.id})")
|
||||
logWrite(
|
||||
f"{context.from_user.id} sent message '{caption}' to {self.id} (source message: {context.id})"
|
||||
)
|
||||
else:
|
||||
logWrite(f"{context.from_user.id} sent message to {self.id} (source message: {context.id})")
|
||||
logWrite(
|
||||
f"{context.from_user.id} sent message to {self.id} (source message: {context.id})"
|
||||
)
|
||||
|
||||
# Add notices for admin or user
|
||||
if text is not None:
|
||||
if adm_context:
|
||||
text += locale("message_reply_notice", "message")
|
||||
elif adm_origin:
|
||||
text = locale("message_from", "message").format(context.from_user.first_name, context.from_user.id) + text + locale("message_reply_notice", "message")
|
||||
text = (
|
||||
locale("message_from", "message").format(
|
||||
context.from_user.first_name, context.from_user.id
|
||||
)
|
||||
+ text
|
||||
+ locale("message_reply_notice", "message")
|
||||
)
|
||||
else:
|
||||
text = locale("message_reply_notice", "message")
|
||||
|
||||
@ -149,77 +205,118 @@ class HoloUser():
|
||||
if adm_context:
|
||||
caption += locale("message_reply_notice", "message")
|
||||
elif adm_origin:
|
||||
caption = locale("message_from", "message").format(context.from_user.first_name, context.from_user.id) + caption + locale("message_reply_notice", "message")
|
||||
caption = (
|
||||
locale("message_from", "message").format(
|
||||
context.from_user.first_name, context.from_user.id
|
||||
)
|
||||
+ caption
|
||||
+ locale("message_reply_notice", "message")
|
||||
)
|
||||
else:
|
||||
caption = locale("message_reply_notice", "message")
|
||||
|
||||
# Try sending the message
|
||||
try:
|
||||
|
||||
# Check if origin message exists
|
||||
# This check decides whether we send_ a message or reply_ to one
|
||||
if origin is not None:
|
||||
|
||||
if photo is not None:
|
||||
if isinstance(photo, Photo):
|
||||
photo = photo.file_id
|
||||
new_message = await origin.reply_cached_media(photo, caption=caption, quote=True)
|
||||
new_message = await origin.reply_cached_media(
|
||||
photo, caption=caption, quote=True
|
||||
)
|
||||
elif video is not None:
|
||||
if isinstance(video, Video):
|
||||
video = video.file_id
|
||||
new_message = await origin.reply_cached_media(video, caption=caption, quote=True)
|
||||
new_message = await origin.reply_cached_media(
|
||||
video, caption=caption, quote=True
|
||||
)
|
||||
elif file is not None:
|
||||
if isinstance(file, Document):
|
||||
file = file.file_id
|
||||
new_message = await origin.reply_cached_media(file, caption=caption, quote=True)
|
||||
new_message = await origin.reply_cached_media(
|
||||
file, caption=caption, quote=True
|
||||
)
|
||||
elif animation is not None:
|
||||
if isinstance(animation, Animation):
|
||||
animation = animation.file_id
|
||||
new_message = await origin.reply_cached_media(animation, caption=caption, quote=True)
|
||||
new_message = await origin.reply_cached_media(
|
||||
animation, caption=caption, quote=True
|
||||
)
|
||||
elif voice is not None:
|
||||
if isinstance(voice, Voice):
|
||||
voice = voice.file_id
|
||||
new_message = await origin.reply_voice(voice, caption=caption, quote=True)
|
||||
new_message = await origin.reply_voice(
|
||||
voice, caption=caption, quote=True
|
||||
)
|
||||
else:
|
||||
new_message = await origin.reply_text(text, quote=True)
|
||||
|
||||
else:
|
||||
|
||||
if photo is not None:
|
||||
if isinstance(photo, Photo):
|
||||
photo = photo.file_id
|
||||
new_message = await app.send_cached_media(self.id, photo, caption=caption)
|
||||
new_message = await app.send_cached_media(
|
||||
self.id, photo, caption=caption
|
||||
)
|
||||
elif video is not None:
|
||||
if isinstance(video, Video):
|
||||
video = video.file_id
|
||||
new_message = await app.send_cached_media(self.id, video, caption=caption)
|
||||
new_message = await app.send_cached_media(
|
||||
self.id, video, caption=caption
|
||||
)
|
||||
elif file is not None:
|
||||
if isinstance(file, Document):
|
||||
file = file.file_id
|
||||
new_message = await app.send_cached_media(self.id, file, caption=caption)
|
||||
new_message = await app.send_cached_media(
|
||||
self.id, file, caption=caption
|
||||
)
|
||||
elif animation is not None:
|
||||
if isinstance(animation, Animation):
|
||||
animation = animation.file_id
|
||||
new_message = await app.send_cached_media(self.id, animation, caption=caption)
|
||||
new_message = await app.send_cached_media(
|
||||
self.id, animation, caption=caption
|
||||
)
|
||||
elif voice is not None:
|
||||
if isinstance(voice, Voice):
|
||||
voice = voice.file_id
|
||||
new_message = await app.send_cached_media(self.id, voice, caption=caption)
|
||||
new_message = await app.send_cached_media(
|
||||
self.id, voice, caption=caption
|
||||
)
|
||||
else:
|
||||
new_message = await app.send_message(self.id, text)
|
||||
|
||||
# Acknowledge sending a message and save entry into DB
|
||||
await context.reply_text(locale("message_sent", "message"), quote=should_quote(context))
|
||||
col_messages.insert_one({"origin": {"chat": context.chat.id, "id": context.id}, "destination": {"chat": new_message.chat.id, "id": new_message.id}})
|
||||
await context.reply_text(
|
||||
locale("message_sent", "message"), quote=should_quote(context)
|
||||
)
|
||||
col_messages.insert_one(
|
||||
{
|
||||
"origin": {"chat": context.chat.id, "id": context.id},
|
||||
"destination": {"chat": new_message.chat.id, "id": new_message.id},
|
||||
}
|
||||
)
|
||||
|
||||
# Report to admin and to sender about message sending failure
|
||||
except Exception as exp:
|
||||
logWrite(f"Exception {exp} happened as {context.from_user.id} tried to send message to {self.id}. Traceback:\n{format_exc()}")
|
||||
logWrite(
|
||||
f"Exception {exp} happened as {context.from_user.id} tried to send message to {self.id}. Traceback:\n{format_exc()}"
|
||||
)
|
||||
try:
|
||||
await app.send_message(configGet("owner"), locale("message_traceback", "message").format(context.from_user.id, self.id, exp, format_exc()))
|
||||
await app.send_message(
|
||||
configGet("owner"),
|
||||
locale("message_traceback", "message").format(
|
||||
context.from_user.id, self.id, exp, format_exc()
|
||||
),
|
||||
)
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
logWrite(f"Could not notify admin about failure when sending message! Admin has never interacted with bot!")
|
||||
await context.reply_text(locale("message_error", "message"), quote=should_quote(context))
|
||||
logWrite(
|
||||
f"Could not notify admin about failure when sending message! Admin has never interacted with bot!"
|
||||
)
|
||||
await context.reply_text(
|
||||
locale("message_error", "message"), quote=should_quote(context)
|
||||
)
|
||||
|
||||
async def label_set(self, chat: Chat, label: str) -> None:
|
||||
"""Set label in destination group
|
||||
@ -227,15 +324,23 @@ class HoloUser():
|
||||
### Args:
|
||||
* chat (`Chat`): Telegram chat
|
||||
* label (`str`): Label you want to set
|
||||
"""
|
||||
"""
|
||||
if len(label) > 16:
|
||||
raise LabelTooLongError(label)
|
||||
self.label = label
|
||||
try:
|
||||
await app.promote_chat_member(configGet("users", "groups"), self.id, privileges=ChatPrivileges(can_pin_messages=True, can_manage_video_chats=True))
|
||||
await app.promote_chat_member(
|
||||
configGet("users", "groups"),
|
||||
self.id,
|
||||
privileges=ChatPrivileges(
|
||||
can_pin_messages=True, can_manage_video_chats=True
|
||||
),
|
||||
)
|
||||
if not await isAnAdmin(self.id):
|
||||
await sleep(0.5)
|
||||
await app.set_administrator_title(configGet("users", "groups"), self.id, label)
|
||||
await app.set_administrator_title(
|
||||
configGet("users", "groups"), self.id, label
|
||||
)
|
||||
self.set("label", label)
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not set {self.id}'s title to '{self.label}' due to {exp}")
|
||||
@ -246,23 +351,29 @@ class HoloUser():
|
||||
|
||||
### Args:
|
||||
* chat (`Chat`): Telegram chat
|
||||
"""
|
||||
"""
|
||||
self.label = ""
|
||||
self.set("label", "")
|
||||
await app.set_administrator_title(configGet("users", "groups"), self.id, "")
|
||||
if not await isAnAdmin(self.id):
|
||||
await app.promote_chat_member(configGet("users", "groups"), self.id, privileges=ChatPrivileges(
|
||||
can_manage_chat=False,
|
||||
can_pin_messages=False,
|
||||
can_manage_video_chats=False
|
||||
))
|
||||
await app.promote_chat_member(
|
||||
configGet("users", "groups"),
|
||||
self.id,
|
||||
privileges=ChatPrivileges(
|
||||
can_manage_chat=False,
|
||||
can_pin_messages=False,
|
||||
can_manage_video_chats=False,
|
||||
),
|
||||
)
|
||||
|
||||
def application_state(self) -> tuple[Literal["none", "fill", "approved", "rejected"], bool]:
|
||||
def application_state(
|
||||
self,
|
||||
) -> tuple[Literal["none", "fill", "approved", "rejected"], bool]:
|
||||
"""Check the current state of application in tmp collection
|
||||
|
||||
### Returns:
|
||||
* `tuple[Literal["none", "fill", "approved", "rejected"], bool]`: First element is an enum of a state and the second one is whether application is complete.
|
||||
"""
|
||||
"""
|
||||
tmp_application = col_tmp.find_one({"user": self.id, "type": "application"})
|
||||
if tmp_application is None:
|
||||
return "none", False
|
||||
@ -274,16 +385,22 @@ class HoloUser():
|
||||
|
||||
### Returns:
|
||||
* `bool`: `True` if yes and `False` if no
|
||||
"""
|
||||
return True if col_applications.find_one({"user": self.id}) is not None else False
|
||||
"""
|
||||
return (
|
||||
True if col_applications.find_one({"user": self.id}) is not None else False
|
||||
)
|
||||
|
||||
def application_restart(self, reapply: bool = False) -> None:
|
||||
"""Reset application of a user in tmp collection and replace it with an empty one
|
||||
"""
|
||||
"""Reset application of a user in tmp collection and replace it with an empty one"""
|
||||
if col_tmp.find_one({"user": self.id, "type": "application"}) is None:
|
||||
col_tmp.insert_one(document=DefaultApplicationTemp(self.id, reapply=reapply).dict)
|
||||
col_tmp.insert_one(
|
||||
document=DefaultApplicationTemp(self.id, reapply=reapply).dict
|
||||
)
|
||||
else:
|
||||
col_tmp.find_one_and_replace({"user": self.id, "type": "application"}, DefaultApplicationTemp(self.id, reapply=reapply).dict)
|
||||
col_tmp.find_one_and_replace(
|
||||
{"user": self.id, "type": "application"},
|
||||
DefaultApplicationTemp(self.id, reapply=reapply).dict,
|
||||
)
|
||||
|
||||
async def application_next(self, query: str, msg: Message) -> None:
|
||||
"""Move on filling application of user
|
||||
@ -291,7 +408,7 @@ class HoloUser():
|
||||
### Args:
|
||||
* query (`str`): Some kind of input
|
||||
* msg (`Message`): Message that should receive replies
|
||||
"""
|
||||
"""
|
||||
|
||||
# if col_tmp.find_one({"user": self.id, "type": "application"}) is None:
|
||||
|
||||
@ -311,102 +428,294 @@ class HoloUser():
|
||||
return
|
||||
|
||||
stage = progress["stage"]
|
||||
|
||||
|
||||
# if self.sponsorship_state()[0] == "fill":
|
||||
# await msg.reply_text(locale("finish_sponsorship", "message"), quote=should_quote(msg))
|
||||
# return
|
||||
|
||||
if progress["state"] == "fill" and progress["sent"] is False:
|
||||
|
||||
if msg.text is not None:
|
||||
msg.text = fix_text(str(msg.text))
|
||||
|
||||
if stage == 2:
|
||||
|
||||
try:
|
||||
input_dt = datetime.strptime(query, "%d.%m.%Y")
|
||||
except ValueError:
|
||||
logWrite(f"User {msg.from_user.id} failed stage {stage} due to sending invalid date format")
|
||||
await msg.reply_text(locale(f"question2_invalid", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage}", "force_reply", locale=self.locale))))
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} failed stage {stage} due to sending invalid date format"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"question2_invalid", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
if (datetime.now() <= input_dt) or ((datetime.now() - input_dt).days) > ((datetime.now() - datetime.now().replace(year=datetime.now().year - configGet("age_maximum"))).days):
|
||||
logWrite(f"User {msg.from_user.id} failed stage {stage} due to joking")
|
||||
await msg.reply_text(locale("question2_joke", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale("question2", "force_reply", locale=self.locale))))
|
||||
if (datetime.now() <= input_dt) or (
|
||||
(datetime.now() - input_dt).days
|
||||
) > (
|
||||
(
|
||||
datetime.now()
|
||||
- datetime.now().replace(
|
||||
year=datetime.now().year - configGet("age_maximum")
|
||||
)
|
||||
).days
|
||||
):
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} failed stage {stage} due to joking"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("question2_joke", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale("question2", "force_reply", locale=self.locale)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
elif ((datetime.now() - input_dt).days) < ((datetime.now() - datetime.now().replace(year=datetime.now().year - configGet("age_allowed"))).days):
|
||||
logWrite(f"User {msg.from_user.id} failed stage {stage} due to being underage")
|
||||
await msg.reply_text(locale("question2_underage", "message", locale=self.locale).format(str(configGet("age_allowed"))), reply_markup=ForceReply(placeholder=str(locale("question2", "force_reply", locale=self.locale))))
|
||||
elif ((datetime.now() - input_dt).days) < (
|
||||
(
|
||||
datetime.now()
|
||||
- datetime.now().replace(
|
||||
year=datetime.now().year - configGet("age_allowed")
|
||||
)
|
||||
).days
|
||||
):
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} failed stage {stage} due to being underage"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(
|
||||
"question2_underage", "message", locale=self.locale
|
||||
).format(str(configGet("age_allowed"))),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale("question2", "force_reply", locale=self.locale)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
else:
|
||||
progress["application"][str(stage)] = input_dt
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "application"}}, {"$set": {"application": progress["application"], "stage": progress["stage"]+1}})
|
||||
await msg.reply_text(locale(f"question{stage+1}", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage+1}", "force_reply", locale=self.locale))))
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "application"}},
|
||||
{
|
||||
"$set": {
|
||||
"application": progress["application"],
|
||||
"stage": progress["stage"] + 1,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"question{stage+1}", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage+1}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
elif stage == 3:
|
||||
try:
|
||||
progress["application"][str(stage)] = find_location(query)
|
||||
if ("lat" in progress["application"][str(stage)] and "lng" in progress["application"][str(stage)]):
|
||||
progress["application"][str(stage)]["location"] = [float(progress["application"][str(stage)]["lng"]), float(progress["application"][str(stage)]["lat"])]
|
||||
if (
|
||||
"lat" in progress["application"][str(stage)]
|
||||
and "lng" in progress["application"][str(stage)]
|
||||
):
|
||||
progress["application"][str(stage)]["location"] = [
|
||||
float(progress["application"][str(stage)]["lng"]),
|
||||
float(progress["application"][str(stage)]["lat"]),
|
||||
]
|
||||
del progress["application"][str(stage)]["lat"]
|
||||
del progress["application"][str(stage)]["lng"]
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "application"}}, {"$set": {"application": progress["application"], "stage": progress["stage"]+1}})
|
||||
await msg.reply_text(locale("question3_found", "message", locale=self.locale).format(progress["application"][str(stage)]["name"], progress["application"][str(stage)]["adminName1"]))
|
||||
await msg.reply_text(locale(f"question{stage+1}", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage+1}", "force_reply", locale=self.locale))))
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "application"}},
|
||||
{
|
||||
"$set": {
|
||||
"application": progress["application"],
|
||||
"stage": progress["stage"] + 1,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("question3_found", "message", locale=self.locale).format(
|
||||
progress["application"][str(stage)]["name"],
|
||||
progress["application"][str(stage)]["adminName1"],
|
||||
)
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"question{stage+1}", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage+1}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
except PlaceNotFoundError:
|
||||
await msg.reply_text(locale("question3_invalid", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage}", "force_reply", locale=self.locale))))
|
||||
await msg.reply_text(
|
||||
locale("question3_invalid", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
except Exception as exp:
|
||||
await msg.reply_text(locale("question3_error", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage}", "force_reply", locale=self.locale))))
|
||||
await msg.reply_text(
|
||||
locale("question3_error", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
try:
|
||||
await app.send_message(configGet("owner"), locale("question3_traceback", "message", locale=self.locale).format(query, exp, format_exc()))
|
||||
await app.send_message(
|
||||
configGet("owner"),
|
||||
locale(
|
||||
"question3_traceback", "message", locale=self.locale
|
||||
).format(query, exp, format_exc()),
|
||||
)
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
logWrite(f"Could not notify admin about failure when sending message! Admin has never interacted with bot!")
|
||||
logWrite(
|
||||
f"Could not notify admin about failure when sending message! Admin has never interacted with bot!"
|
||||
)
|
||||
return
|
||||
|
||||
elif stage == 10:
|
||||
if len(query) > 1024:
|
||||
await msg.reply_text(locale("question10_too_long", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage}", "force_reply", locale=self.locale))))
|
||||
await msg.reply_text(
|
||||
locale("question10_too_long", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
progress["application"][str(stage)] = query
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "application"}}, {"$set": {"application": progress["application"], "complete": True}})
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "application"}},
|
||||
{
|
||||
"$set": {
|
||||
"application": progress["application"],
|
||||
"complete": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
application_content = []
|
||||
i = 1
|
||||
for question in progress["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), progress['application']['2'])
|
||||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(
|
||||
datetime.now(), progress["application"]["2"]
|
||||
)
|
||||
application_content.append(
|
||||
f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if progress['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application']['3']['name']} ({progress['application']['3']['adminName1']})")
|
||||
if progress["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application']['3']['name']} ({progress['application']['3']['adminName1']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application']['3']['name']} ({progress['application']['3']['adminName1']}, {progress['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application']['3']['name']} ({progress['application']['3']['adminName1']}, {progress['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application'][question]}")
|
||||
application_content.append(
|
||||
f"{locale('question'+str(i), 'message', 'question_titles', locale=self.locale)} {progress['application'][question]}"
|
||||
)
|
||||
i += 1
|
||||
await msg.reply_text(locale("confirm", "message", locale=self.locale).format("\n".join(application_content)), reply_markup=ReplyKeyboardMarkup(locale("confirm", "keyboard", locale=self.locale), resize_keyboard=True))
|
||||
await msg.reply_text(
|
||||
locale("confirm", "message", locale=self.locale).format(
|
||||
"\n".join(application_content)
|
||||
),
|
||||
reply_markup=ReplyKeyboardMarkup(
|
||||
locale("confirm", "keyboard", locale=self.locale),
|
||||
resize_keyboard=True,
|
||||
),
|
||||
)
|
||||
|
||||
else:
|
||||
if len(query) > 256:
|
||||
await msg.reply_text(locale("question_too_long", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage}", "force_reply", locale=self.locale))))
|
||||
await msg.reply_text(
|
||||
locale("question_too_long", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
progress["application"][str(stage)] = query
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "application"}}, {"$set": {"application": progress["application"], "stage": progress["stage"]+1}})
|
||||
await msg.reply_text(locale(f"question{stage+1}", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"question{stage+1}", "force_reply", locale=self.locale))))
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "application"}},
|
||||
{
|
||||
"$set": {
|
||||
"application": progress["application"],
|
||||
"stage": progress["stage"] + 1,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"question{stage+1}", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"question{stage+1}", "force_reply", locale=self.locale
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
logWrite(f"User {self.id} completed stage {stage} of application")
|
||||
|
||||
else:
|
||||
return
|
||||
|
||||
def sponsorship_state(self) -> tuple[Literal["none", "fill", "approved", "rejected"], bool]:
|
||||
def sponsorship_state(
|
||||
self,
|
||||
) -> tuple[Literal["none", "fill", "approved", "rejected"], bool]:
|
||||
"""Check the current state of sponsorship in tmp collection
|
||||
|
||||
### Returns:
|
||||
* `tuple[Literal["none", "fill", "approved", "rejected"], bool]`: First element is an enum of a state and the second one is whether sponsorship application is complete.
|
||||
"""
|
||||
"""
|
||||
tmp_sponsorship = col_tmp.find_one({"user": self.id, "type": "sponsorship"})
|
||||
if tmp_sponsorship is None:
|
||||
return "none", False
|
||||
@ -418,30 +727,37 @@ class HoloUser():
|
||||
|
||||
### Returns:
|
||||
* `bool`: `True` if yes and `False` if no
|
||||
"""
|
||||
return True if col_sponsorships.find_one({"user": self.id, "expires": {"$gt": datetime.now()}}) is not None else False
|
||||
"""
|
||||
return (
|
||||
True
|
||||
if col_sponsorships.find_one(
|
||||
{"user": self.id, "expires": {"$gt": datetime.now()}}
|
||||
)
|
||||
is not None
|
||||
else False
|
||||
)
|
||||
|
||||
def sponsorship_restart(self) -> None:
|
||||
"""Reset sponsorship of a user in tmp collection and replace it with an empty one
|
||||
"""
|
||||
"""Reset sponsorship of a user in tmp collection and replace it with an empty one"""
|
||||
if col_tmp.find_one({"user": self.id, "type": "sponsorship"}) is None:
|
||||
col_tmp.insert_one(document=DefaultSponsorshipTemp(self.id).dict)
|
||||
else:
|
||||
col_tmp.delete_one({"user": self.id, "type": "sponsorship"})
|
||||
col_tmp.insert_one(document=DefaultSponsorshipTemp(self.id).dict)
|
||||
|
||||
async def sponsorship_next(self, query: str, msg: Message, photo: Union[Photo, None] = None) -> None:
|
||||
async def sponsorship_next(
|
||||
self, query: str, msg: Message, photo: Union[Photo, None] = None
|
||||
) -> None:
|
||||
"""Move on filling sponsorship of user
|
||||
|
||||
### Args:
|
||||
* query (`str`): Some kind of input
|
||||
* msg (`Message`): Message that should receive replies
|
||||
"""
|
||||
"""
|
||||
|
||||
progress = col_tmp.find_one({"user": self.id, "type": "sponsorship"})
|
||||
|
||||
if progress is not None:
|
||||
|
||||
stage = progress["stage"]
|
||||
|
||||
if msg.text is not None:
|
||||
@ -450,58 +766,171 @@ class HoloUser():
|
||||
msg.caption = fix_text(msg.caption)
|
||||
|
||||
if progress["state"] == "fill" and progress["sent"] is False:
|
||||
|
||||
if stage == 1:
|
||||
|
||||
if len(query) > 240:
|
||||
logWrite(f"User {msg.from_user.id} failed stage {stage} due to sending invalid date format")
|
||||
await msg.reply_text(locale(f"sponsor1_invalid", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"sponsor{stage}", "force_reply", locale=self.locale))))
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} failed stage {stage} due to sending invalid date format"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"sponsor1_invalid", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"sponsor{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
progress["sponsorship"]["streamer"] = query
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}}, {"$set": {"sponsorship": progress["sponsorship"], "stage": progress["stage"]+1}})
|
||||
await msg.reply_text(locale(f"sponsor{stage+1}", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"sponsor{stage+1}", "force_reply", locale=self.locale))))
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}},
|
||||
{
|
||||
"$set": {
|
||||
"sponsorship": progress["sponsorship"],
|
||||
"stage": progress["stage"] + 1,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"sponsor{stage+1}", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"sponsor{stage+1}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
elif stage == 2:
|
||||
|
||||
try:
|
||||
input_dt = datetime.strptime(query, "%d.%m.%Y")
|
||||
except ValueError:
|
||||
logWrite(f"User {msg.from_user.id} failed stage {stage} due to sending invalid date format")
|
||||
await msg.reply_text(locale(f"sponsor2_invalid", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"sponsor{stage}", "force_reply", locale=self.locale))))
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} failed stage {stage} due to sending invalid date format"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"sponsor2_invalid", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"sponsor{stage}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
if datetime.now() >= input_dt:
|
||||
logWrite(f"User {msg.from_user.id} failed stage {stage} due to sending date in the past")
|
||||
await msg.reply_text(locale("sponsor2_past", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale("sponsor2", "force_reply", locale=self.locale))))
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} failed stage {stage} due to sending date in the past"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("sponsor2_past", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
"sponsor2", "force_reply", locale=self.locale
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
else:
|
||||
progress["sponsorship"]["expires"] = input_dt
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}}, {"$set": {"sponsorship": progress["sponsorship"], "stage": progress["stage"]+1}})
|
||||
await msg.reply_text(locale(f"sponsor{stage+1}", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"sponsor{stage+1}", "force_reply", locale=self.locale))))
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}},
|
||||
{
|
||||
"$set": {
|
||||
"sponsorship": progress["sponsorship"],
|
||||
"stage": progress["stage"] + 1,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"sponsor{stage+1}", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"sponsor{stage+1}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
elif stage == 3:
|
||||
|
||||
if photo is not None:
|
||||
progress["sponsorship"]["proof"] = photo.file_id
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}}, {"$set": {"sponsorship": progress["sponsorship"], "stage": progress["stage"]+1}})
|
||||
await msg.reply_text(locale(f"sponsor{stage+1}", "message", locale=self.locale), reply_markup=ForceReply(placeholder=str(locale(f"sponsor{stage+1}", "force_reply", locale=self.locale))))
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}},
|
||||
{
|
||||
"$set": {
|
||||
"sponsorship": progress["sponsorship"],
|
||||
"stage": progress["stage"] + 1,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(f"sponsor{stage+1}", "message", locale=self.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
f"sponsor{stage+1}",
|
||||
"force_reply",
|
||||
locale=self.locale,
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
elif stage == 4:
|
||||
if len(query) > 16:
|
||||
await msg.reply_text(locale("label_too_long", "message"), reply_markup=ForceReply(placeholder=str(locale("sponsor4", "force_reply", locale=self.locale))))
|
||||
await msg.reply_text(
|
||||
locale("label_too_long", "message"),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(
|
||||
"sponsor4", "force_reply", locale=self.locale
|
||||
)
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
progress["sponsorship"]["label"] = query
|
||||
col_tmp.update_one({"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}}, {"$set": {"sponsorship": progress["sponsorship"], "complete": True}})
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": self.id}, "type": {"$eq": "sponsorship"}},
|
||||
{
|
||||
"$set": {
|
||||
"sponsorship": progress["sponsorship"],
|
||||
"complete": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
await msg.reply_cached_media(
|
||||
progress["sponsorship"]["proof"],
|
||||
caption=locale("sponsor_confirm", "message", locale=self.locale).format(
|
||||
caption=locale(
|
||||
"sponsor_confirm", "message", locale=self.locale
|
||||
).format(
|
||||
progress["sponsorship"]["streamer"],
|
||||
progress["sponsorship"]["expires"].strftime("%d.%m.%Y"),
|
||||
progress["sponsorship"]["label"]
|
||||
progress["sponsorship"]["label"],
|
||||
),
|
||||
reply_markup=ReplyKeyboardMarkup(locale("confirm", "keyboard", locale=self.locale), resize_keyboard=True))
|
||||
reply_markup=ReplyKeyboardMarkup(
|
||||
locale("confirm", "keyboard", locale=self.locale),
|
||||
resize_keyboard=True,
|
||||
),
|
||||
)
|
||||
|
||||
else:
|
||||
return
|
||||
@ -516,5 +945,9 @@ class HoloUser():
|
||||
|
||||
### Returns:
|
||||
* `bool`: `True` if any not finished spoilers available and `False` if none.
|
||||
"""
|
||||
return False if col_spoilers.find_one({"user": self.id, "completed": False}) is None else True
|
||||
"""
|
||||
return (
|
||||
False
|
||||
if col_spoilers.find_one({"user": self.id, "completed": False}) is None
|
||||
else True
|
||||
)
|
||||
|
@ -2,6 +2,7 @@
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
class DefaultApplicationTemp(dict):
|
||||
def __init__(self, user: int, reapply: bool = False):
|
||||
super().__init__({})
|
||||
@ -23,10 +24,11 @@ class DefaultApplicationTemp(dict):
|
||||
"7": None,
|
||||
"8": None,
|
||||
"9": None,
|
||||
"10": None
|
||||
}
|
||||
"10": None,
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
class DefaultSponsorshipTemp(dict):
|
||||
def __init__(self, user: int):
|
||||
super().__init__({})
|
||||
@ -41,6 +43,6 @@ class DefaultSponsorshipTemp(dict):
|
||||
"streamer": None,
|
||||
"expires": datetime.fromtimestamp(0),
|
||||
"proof": None,
|
||||
"label": ""
|
||||
}
|
||||
}
|
||||
"label": "",
|
||||
},
|
||||
}
|
||||
|
@ -48,7 +48,6 @@ from modules.handlers.everything import *
|
||||
from modules.scheduled import *
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
logWrite(f"Starting up with pid {pid}")
|
||||
|
||||
# Yes, it should be in some kind of async main() function but I don't give a shit.
|
||||
@ -57,17 +56,43 @@ if __name__ == "__main__":
|
||||
|
||||
try:
|
||||
if path.exists(path.join(configGet("cache", "locations"), "shutdown_time")):
|
||||
downtime = relativedelta(datetime.now(), datetime.fromtimestamp(jsonLoad(path.join(configGet("cache", "locations"), "shutdown_time"))["timestamp"]))
|
||||
downtime = relativedelta(
|
||||
datetime.now(),
|
||||
datetime.fromtimestamp(
|
||||
jsonLoad(
|
||||
path.join(configGet("cache", "locations"), "shutdown_time")
|
||||
)["timestamp"]
|
||||
),
|
||||
)
|
||||
if downtime.days >= 1:
|
||||
app.send_message(configGet("owner"), locale("startup_downtime_days", "message").format(pid, downtime.days))
|
||||
app.send_message(
|
||||
configGet("owner"),
|
||||
locale("startup_downtime_days", "message").format(
|
||||
pid, downtime.days
|
||||
),
|
||||
)
|
||||
elif downtime.hours >= 1:
|
||||
app.send_message(configGet("owner"), locale("startup_downtime_hours", "message").format(pid, downtime.hours))
|
||||
app.send_message(
|
||||
configGet("owner"),
|
||||
locale("startup_downtime_hours", "message").format(
|
||||
pid, downtime.hours
|
||||
),
|
||||
)
|
||||
else:
|
||||
app.send_message(configGet("owner"), locale("startup_downtime_minutes", "message").format(pid, downtime.minutes))
|
||||
app.send_message(
|
||||
configGet("owner"),
|
||||
locale("startup_downtime_minutes", "message").format(
|
||||
pid, downtime.minutes
|
||||
),
|
||||
)
|
||||
else:
|
||||
app.send_message(configGet("owner"), locale("startup", "message").format(pid))
|
||||
app.send_message(
|
||||
configGet("owner"), locale("startup", "message").format(pid)
|
||||
)
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
logWrite(f"Could not send startup message to bot owner. Perhaps user has not started the bot yet.")
|
||||
logWrite(
|
||||
f"Could not send startup message to bot owner. Perhaps user has not started the bot yet."
|
||||
)
|
||||
|
||||
scheduler.start()
|
||||
|
||||
@ -76,11 +101,16 @@ if __name__ == "__main__":
|
||||
try:
|
||||
app.send_message(configGet("owner"), locale("shutdown", "message").format(pid))
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
logWrite(f"Could not send shutdown message to bot owner. Perhaps user has not started the bot yet.")
|
||||
logWrite(
|
||||
f"Could not send shutdown message to bot owner. Perhaps user has not started the bot yet."
|
||||
)
|
||||
|
||||
app.stop()
|
||||
|
||||
makedirs(configGet("cache", "locations"), exist_ok=True)
|
||||
jsonSave({"timestamp": time()}, path.join(configGet("cache", "locations"), "shutdown_time"))
|
||||
|
||||
killProc(pid)
|
||||
jsonSave(
|
||||
{"timestamp": time()},
|
||||
path.join(configGet("cache", "locations"), "shutdown_time"),
|
||||
)
|
||||
|
||||
killProc(pid)
|
||||
|
@ -6,17 +6,25 @@ from modules.utils import locale
|
||||
from modules.database import col_bans
|
||||
from modules.logging import logWrite
|
||||
|
||||
|
||||
@app.on_callback_query(filters.regex("ban_[\s\S]*"))
|
||||
async def callback_query_reject(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
|
||||
if not await isAnAdmin(int(fullclb[1])):
|
||||
col_bans.insert_one( {"user": int(fullclb[1])} )
|
||||
col_bans.insert_one({"user": int(fullclb[1])})
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("banned", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("banned", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(text=locale("sub_banned", "callback", locale=clb.from_user))
|
||||
|
||||
logWrite(f"User {fullclb[1]} has been banned by {clb.from_user.id}")
|
||||
@ -24,4 +32,4 @@ async def callback_query_reject(app: Client, clb: CallbackQuery):
|
||||
try:
|
||||
await app.send_message(int(fullclb[1]), locale("you_are_banned", "message"))
|
||||
except Exception as exp:
|
||||
logWrite(f"Could send ban message to {fullclb[1]} due to {exp}")
|
||||
logWrite(f"Could send ban message to {fullclb[1]} due to {exp}")
|
||||
|
@ -4,8 +4,11 @@ from pyrogram.types import CallbackQuery
|
||||
from pyrogram.client import Client
|
||||
from modules.utils import locale
|
||||
|
||||
|
||||
# Callback empty ===============================================================================================================
|
||||
@app.on_callback_query(filters.regex("nothing"))
|
||||
async def callback_query_nothing(app: Client, clb: CallbackQuery):
|
||||
await clb.answer(text=locale("nothing", "callback", locale=clb.from_user))
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -1,6 +1,11 @@
|
||||
from datetime import datetime
|
||||
from app import app
|
||||
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, ReplyKeyboardRemove, CallbackQuery
|
||||
from pyrogram.types import (
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
ReplyKeyboardRemove,
|
||||
CallbackQuery,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from pyrogram import filters
|
||||
from classes.holo_user import HoloUser
|
||||
@ -9,91 +14,194 @@ from modules.handlers.confirmation import confirm_yes
|
||||
from modules.handlers.welcome import welcome_pass
|
||||
from modules.database import col_tmp, col_applications
|
||||
|
||||
|
||||
# Callbacks reapply ============================================================================================================
|
||||
@app.on_callback_query(filters.regex("reapply_yes_[\s\S]*"))
|
||||
async def callback_reapply_query_accept(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("approved_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
logWrite(f"User {holo_user.id} got their reapplication approved by {clb.from_user.id}")
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("approved_by", "message").format(clb.from_user.first_name, holo_user.id),
|
||||
disable_notification=True,
|
||||
)
|
||||
logWrite(
|
||||
f"User {holo_user.id} got their reapplication approved by {clb.from_user.id}"
|
||||
)
|
||||
|
||||
await app.send_message(holo_user.id, locale("approved_joined", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
holo_user.id, locale("approved_joined", "message", locale=holo_user)
|
||||
)
|
||||
|
||||
applications = col_applications.find({"user": holo_user.id})
|
||||
|
||||
if len(list(applications)) > 1:
|
||||
col_applications.delete_many({"user": holo_user.id})
|
||||
col_applications.insert_one({"user": holo_user.id, "date": datetime.now(), "admin": clb.from_user.id, "application": col_tmp.find_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}})["application"]})
|
||||
col_applications.insert_one(
|
||||
{
|
||||
"user": holo_user.id,
|
||||
"date": datetime.now(),
|
||||
"admin": clb.from_user.id,
|
||||
"application": col_tmp.find_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}
|
||||
)["application"],
|
||||
}
|
||||
)
|
||||
elif applications == 1:
|
||||
col_applications.find_one_and_replace({"user": holo_user.id}, {"user": holo_user.id, "date": datetime.now(), "admin": clb.from_user.id, "application": col_tmp.find_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}})["application"]})
|
||||
col_applications.find_one_and_replace(
|
||||
{"user": holo_user.id},
|
||||
{
|
||||
"user": holo_user.id,
|
||||
"date": datetime.now(),
|
||||
"admin": clb.from_user.id,
|
||||
"application": col_tmp.find_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}
|
||||
)["application"],
|
||||
},
|
||||
)
|
||||
else:
|
||||
col_applications.insert_one({"user": holo_user.id, "date": datetime.now(), "admin": clb.from_user.id, "application": col_tmp.find_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}})["application"]})
|
||||
col_tmp.update_one({"user": holo_user.id, "type": "application"}, {"$set": {"state": "approved", "sent": False}})
|
||||
col_applications.insert_one(
|
||||
{
|
||||
"user": holo_user.id,
|
||||
"date": datetime.now(),
|
||||
"admin": clb.from_user.id,
|
||||
"application": col_tmp.find_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}
|
||||
)["application"],
|
||||
}
|
||||
)
|
||||
col_tmp.update_one(
|
||||
{"user": holo_user.id, "type": "application"},
|
||||
{"$set": {"state": "approved", "sent": False}},
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("accepted", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sub_accepted", "callback", locale=clb.from_user).format(holo_user.id), show_alert=True)
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sub_accepted", "callback", locale=clb.from_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
need_link = True
|
||||
|
||||
async for member in app.get_chat_members(configGet("users", "groups")):
|
||||
if member.user.id == holo_user.id:
|
||||
need_link = False
|
||||
|
||||
if need_link:
|
||||
link = await app.create_chat_invite_link(configGet("users", "groups"), name=f"Invite for {holo_user.id}", member_limit=1) #, expire_date=datetime.now()+timedelta(days=1))
|
||||
|
||||
await app.send_message(holo_user.id, locale("read_rules", "message", locale=holo_user))
|
||||
if need_link:
|
||||
link = await app.create_chat_invite_link(
|
||||
configGet("users", "groups"),
|
||||
name=f"Invite for {holo_user.id}",
|
||||
member_limit=1,
|
||||
) # , expire_date=datetime.now()+timedelta(days=1))
|
||||
|
||||
await app.send_message(
|
||||
holo_user.id, locale("read_rules", "message", locale=holo_user)
|
||||
)
|
||||
|
||||
for rule_msg in locale("rules", locale=holo_user):
|
||||
await app.send_message(holo_user.id, rule_msg)
|
||||
|
||||
await app.send_message(holo_user.id, locale("approved", "message"), reply_markup=InlineKeyboardMarkup(
|
||||
[[
|
||||
InlineKeyboardButton(str(locale("join", "button", locale=holo_user)), url=link.invite_link)
|
||||
]]
|
||||
))
|
||||
await app.send_message(
|
||||
holo_user.id,
|
||||
locale("approved", "message"),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
str(locale("join", "button", locale=holo_user)),
|
||||
url=link.invite_link,
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
holo_user.set("link", link.invite_link)
|
||||
logWrite(f"User {holo_user.id} got an invite link {link.invite_link}")
|
||||
|
||||
else:
|
||||
await app.send_message(holo_user.id, locale("approved_joined", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
holo_user.id, locale("approved_joined", "message", locale=holo_user)
|
||||
)
|
||||
|
||||
|
||||
@app.on_callback_query(filters.regex("reapply_no_[\s\S]*"))
|
||||
async def callback_query_reapply_reject(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("rejected_by", "message").format(clb.from_user.first_name, fullclb[2]), disable_notification=True)
|
||||
await app.send_message(holo_user.id, locale("rejected", "message", locale=holo_user))
|
||||
logWrite(f"User {fullclb[2]} got their reapplication rejected by {clb.from_user.id}")
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("rejected_by", "message").format(clb.from_user.first_name, fullclb[2]),
|
||||
disable_notification=True,
|
||||
)
|
||||
await app.send_message(
|
||||
holo_user.id, locale("rejected", "message", locale=holo_user)
|
||||
)
|
||||
logWrite(
|
||||
f"User {fullclb[2]} got their reapplication rejected by {clb.from_user.id}"
|
||||
)
|
||||
|
||||
col_tmp.update_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}, {"$set": {"state": "rejected", "sent": False}})
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}},
|
||||
{"$set": {"state": "rejected", "sent": False}},
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("declined", "button")), callback_data="nothing")], [InlineKeyboardButton(text=str(locale("ban", "button")), callback_data=f"ban_{fullclb[2]}")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("declined", "button")), callback_data="nothing"
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("ban", "button")), callback_data=f"ban_{fullclb[2]}"
|
||||
)
|
||||
],
|
||||
]
|
||||
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sub_rejected", "callback", locale=clb.from_user).format(
|
||||
fullclb[2]
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sub_rejected", "callback", locale=clb.from_user).format(fullclb[2]), show_alert=True)
|
||||
|
||||
# Use old application when user reapplies after leaving the chat
|
||||
@app.on_callback_query(filters.regex("reapply_old_[\s\S]*"))
|
||||
async def callback_query_reapply_old(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(clb.from_user)
|
||||
|
||||
if holo_user.sponsorship_state()[0] == "fill":
|
||||
await clb.message.reply_text(locale("finish_sponsorship", "message"), quote=False)
|
||||
await clb.message.reply_text(
|
||||
locale("finish_sponsorship", "message"), quote=False
|
||||
)
|
||||
return
|
||||
|
||||
message = await app.get_messages(clb.from_user.id, int(fullclb[2]))
|
||||
|
||||
if col_tmp.find_one({"user": holo_user.id, "type": "application"}) is None and col_applications.find_one({"user": holo_user.id}) is not None:
|
||||
if (
|
||||
col_tmp.find_one({"user": holo_user.id, "type": "application"}) is None
|
||||
and col_applications.find_one({"user": holo_user.id}) is not None
|
||||
):
|
||||
col_tmp.insert_one(
|
||||
{
|
||||
"user": holo_user.id,
|
||||
@ -103,37 +211,68 @@ async def callback_query_reapply_old(app: Client, clb: CallbackQuery):
|
||||
"state": "fill",
|
||||
"reapply": True,
|
||||
"stage": 10,
|
||||
"application": col_applications.find_one({"user": holo_user.id})["application"]
|
||||
"application": col_applications.find_one({"user": holo_user.id})[
|
||||
"application"
|
||||
],
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
await confirm_yes(app, message, kind="application")
|
||||
await clb.message.edit(clb.message.text, reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(locale("done", "button", locale=clb.from_user), "nothing")]]))
|
||||
await clb.message.edit(
|
||||
clb.message.text,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale("done", "button", locale=clb.from_user), "nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# Start a new application when user reapplies after leaving the chat
|
||||
@app.on_callback_query(filters.regex("reapply_new_[\s\S]*"))
|
||||
async def callback_query_reapply_new(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
|
||||
holo_user = HoloUser(clb.from_user)
|
||||
|
||||
if holo_user.sponsorship_state()[0] == "fill":
|
||||
await clb.message.reply_text(locale("finish_sponsorship", "message"), quote=False)
|
||||
await clb.message.reply_text(
|
||||
locale("finish_sponsorship", "message"), quote=False
|
||||
)
|
||||
return
|
||||
|
||||
await clb.answer(locale("reapply_stopped", "callback", locale=clb.from_user))
|
||||
message = await app.get_messages(clb.from_user.id, int(fullclb[2]))
|
||||
col_tmp.update_one({"user": clb.from_user.id, "type": "application"}, {"$set": {"state": "fill", "completed": False, "stage": 1}})
|
||||
col_tmp.update_one(
|
||||
{"user": clb.from_user.id, "type": "application"},
|
||||
{"$set": {"state": "fill", "completed": False, "stage": 1}},
|
||||
)
|
||||
holo_user.application_restart(reapply=True)
|
||||
await welcome_pass(app, message, once_again=True)
|
||||
logWrite(f"User {clb.from_user.id} restarted the application after leaving the chat earlier")
|
||||
await clb.message.edit(clb.message.text, reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(locale("done", "button", locale=clb.from_user), "nothing")]]))
|
||||
logWrite(
|
||||
f"User {clb.from_user.id} restarted the application after leaving the chat earlier"
|
||||
)
|
||||
await clb.message.edit(
|
||||
clb.message.text,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale("done", "button", locale=clb.from_user), "nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# Abort application fill in progress and restart it
|
||||
@app.on_callback_query(filters.regex("reapply_stop_[\s\S]*"))
|
||||
async def callback_query_reapply_stop(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(clb.from_user)
|
||||
|
||||
@ -142,5 +281,10 @@ async def callback_query_reapply_stop(app: Client, clb: CallbackQuery):
|
||||
message = await app.get_messages(clb.from_user.id, int(fullclb[2]))
|
||||
await welcome_pass(app, message, once_again=True)
|
||||
logWrite(f"User {clb.from_user.id} restarted the application due to typo in it")
|
||||
await clb.message.edit(locale("reapply_restarted", "message", locale=holo_user), reply_markup=ReplyKeyboardRemove())
|
||||
# ==============================================================================================================================
|
||||
await clb.message.edit(
|
||||
locale("reapply_restarted", "message", locale=holo_user),
|
||||
reply_markup=ReplyKeyboardRemove(),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -6,10 +6,10 @@ from pyrogram import filters
|
||||
from modules.utils import locale, logWrite
|
||||
from modules.commands.rules import DefaultRulesMarkup
|
||||
|
||||
|
||||
# Callback rule ================================================================================================================
|
||||
@app.on_callback_query(filters.regex("rule_[\s\S]*"))
|
||||
async def callback_query_rule(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
|
||||
logWrite(f"User {clb.from_user.id} requested to check out rule {fullclb[1]}")
|
||||
@ -18,55 +18,88 @@ async def callback_query_rule(app: Client, clb: CallbackQuery):
|
||||
|
||||
if rule_num == len(locale("rules")):
|
||||
lower_buttons = [
|
||||
InlineKeyboardButton(locale("rules_prev", "button", locale=clb.from_user), callback_data=f"rule_{rule_num-1}")
|
||||
InlineKeyboardButton(
|
||||
locale("rules_prev", "button", locale=clb.from_user),
|
||||
callback_data=f"rule_{rule_num-1}",
|
||||
)
|
||||
]
|
||||
elif rule_num == 1:
|
||||
lower_buttons = [
|
||||
InlineKeyboardButton(locale("rules_next", "button", locale=clb.from_user), callback_data=f"rule_{rule_num+1}")
|
||||
InlineKeyboardButton(
|
||||
locale("rules_next", "button", locale=clb.from_user),
|
||||
callback_data=f"rule_{rule_num+1}",
|
||||
)
|
||||
]
|
||||
else:
|
||||
lower_buttons = [
|
||||
InlineKeyboardButton(locale("rules_prev", "button", locale=clb.from_user), callback_data=f"rule_{rule_num-1}"),
|
||||
InlineKeyboardButton(locale("rules_next", "button", locale=clb.from_user), callback_data=f"rule_{rule_num+1}")
|
||||
InlineKeyboardButton(
|
||||
locale("rules_prev", "button", locale=clb.from_user),
|
||||
callback_data=f"rule_{rule_num-1}",
|
||||
),
|
||||
InlineKeyboardButton(
|
||||
locale("rules_next", "button", locale=clb.from_user),
|
||||
callback_data=f"rule_{rule_num+1}",
|
||||
),
|
||||
]
|
||||
|
||||
try:
|
||||
await clb.message.edit(text=locale("rules", locale=clb.from_user)[rule_num-1], disable_web_page_preview=True, reply_markup=InlineKeyboardMarkup(
|
||||
await clb.message.edit(
|
||||
text=locale("rules", locale=clb.from_user)[rule_num - 1],
|
||||
disable_web_page_preview=True,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(locale("rules_home", "button", locale=clb.from_user), callback_data="rules_home"),
|
||||
InlineKeyboardButton(locale("rules_additional", "button", locale=clb.from_user), callback_data="rules_additional")
|
||||
InlineKeyboardButton(
|
||||
locale("rules_home", "button", locale=clb.from_user),
|
||||
callback_data="rules_home",
|
||||
),
|
||||
InlineKeyboardButton(
|
||||
locale("rules_additional", "button", locale=clb.from_user),
|
||||
callback_data="rules_additional",
|
||||
),
|
||||
],
|
||||
lower_buttons
|
||||
lower_buttons,
|
||||
]
|
||||
)
|
||||
),
|
||||
)
|
||||
except bad_request_400.MessageNotModified:
|
||||
pass
|
||||
|
||||
await clb.answer(text=locale("rules_page", "callback", locale=clb.from_user).format(fullclb[1]))
|
||||
await clb.answer(
|
||||
text=locale("rules_page", "callback", locale=clb.from_user).format(fullclb[1])
|
||||
)
|
||||
|
||||
|
||||
@app.on_callback_query(filters.regex("rules_home"))
|
||||
async def callback_query_rules_home(app: Client, clb: CallbackQuery):
|
||||
|
||||
logWrite(f"User {clb.from_user.id} requested to check out homepage rules")
|
||||
|
||||
try:
|
||||
await clb.message.edit(text=locale("rules_msg", locale=clb.from_user), disable_web_page_preview=True, reply_markup=DefaultRulesMarkup(clb.from_user).keyboard)
|
||||
await clb.message.edit(
|
||||
text=locale("rules_msg", locale=clb.from_user),
|
||||
disable_web_page_preview=True,
|
||||
reply_markup=DefaultRulesMarkup(clb.from_user).keyboard,
|
||||
)
|
||||
except bad_request_400.MessageNotModified:
|
||||
pass
|
||||
|
||||
await clb.answer(text=locale("rules_home", "callback", locale=clb.from_user))
|
||||
|
||||
|
||||
@app.on_callback_query(filters.regex("rules_additional"))
|
||||
async def callback_query_rules_additional(app: Client, clb: CallbackQuery):
|
||||
|
||||
logWrite(f"User {clb.from_user.id} requested to check out additional rules")
|
||||
|
||||
try:
|
||||
await clb.message.edit(text=locale("rules_additional", locale=clb.from_user), disable_web_page_preview=True, reply_markup=DefaultRulesMarkup(clb.from_user).keyboard)
|
||||
await clb.message.edit(
|
||||
text=locale("rules_additional", locale=clb.from_user),
|
||||
disable_web_page_preview=True,
|
||||
reply_markup=DefaultRulesMarkup(clb.from_user).keyboard,
|
||||
)
|
||||
except bad_request_400.MessageNotModified:
|
||||
pass
|
||||
|
||||
await clb.answer(text=locale("rules_additional", "callback", locale=clb.from_user))
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -8,29 +8,78 @@ from bson.objectid import ObjectId
|
||||
|
||||
from modules.utils import configGet, jsonLoad, locale
|
||||
|
||||
|
||||
# Callback sid =================================================================================================================
|
||||
@app.on_callback_query(filters.regex("sid_[\s\S]*"))
|
||||
async def callback_query_sid(app: Client, clb: CallbackQuery):
|
||||
await clb.answer(url=f'https://t.me/{(await app.get_me()).username}?start={clb.data.split("_")[1]}')
|
||||
await clb.answer(
|
||||
url=f'https://t.me/{(await app.get_me()).username}?start={clb.data.split("_")[1]}'
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# Callback shc =================================================================================================================
|
||||
@app.on_callback_query(filters.regex("shc_[\s\S]*"))
|
||||
async def callback_query_shc(app: Client, clb: CallbackQuery):
|
||||
|
||||
if (clb.from_user.id not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))):
|
||||
await clb.answer(locale("spoiler_forbidden", "callback", locale=clb.from_user), show_alert=True)
|
||||
if clb.from_user.id not in jsonLoad(
|
||||
path.join(configGet("cache", "locations"), "group_members")
|
||||
):
|
||||
await clb.answer(
|
||||
locale("spoiler_forbidden", "callback", locale=clb.from_user),
|
||||
show_alert=True,
|
||||
)
|
||||
return
|
||||
|
||||
spoil = col_spoilers.find_one( {"_id": ObjectId(clb.data.split("_")[1])} )
|
||||
spoil = col_spoilers.find_one({"_id": ObjectId(clb.data.split("_")[1])})
|
||||
|
||||
if spoil["description"] == "":
|
||||
desc = locale("spoiler_empty_named", "message", locale=clb.from_user).format(locale(spoil["category"], "message", "spoiler_categories"), clb.from_user.first_name)
|
||||
desc = locale("spoiler_empty_named", "message", locale=clb.from_user).format(
|
||||
locale(spoil["category"], "message", "spoiler_categories"),
|
||||
clb.from_user.first_name,
|
||||
)
|
||||
else:
|
||||
desc = locale("spoiler_described_named", "message", locale=clb.from_user).format(locale(spoil["category"], "message", "spoiler_categories"), clb.from_user.first_name, spoil["description"])
|
||||
desc = locale(
|
||||
"spoiler_described_named", "message", locale=clb.from_user
|
||||
).format(
|
||||
locale(spoil["category"], "message", "spoiler_categories"),
|
||||
clb.from_user.first_name,
|
||||
spoil["description"],
|
||||
)
|
||||
|
||||
await app.send_message(configGet("users", "groups"), desc, reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(locale("spoiler_view", "button", locale=clb.from_user), callback_data=f'sid_{clb.data.split("_")[1]}')]]))
|
||||
await app.send_message(configGet("admin", "groups"), desc, reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(locale("spoiler_view", "button", locale=clb.from_user), callback_data=f'sid_{clb.data.split("_")[1]}')]]))
|
||||
await app.send_message(
|
||||
configGet("users", "groups"),
|
||||
desc,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale("spoiler_view", "button", locale=clb.from_user),
|
||||
callback_data=f'sid_{clb.data.split("_")[1]}',
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
desc,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale("spoiler_view", "button", locale=clb.from_user),
|
||||
callback_data=f'sid_{clb.data.split("_")[1]}',
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
await clb.answer(locale("spoiler_sent", "callback", locale=clb.from_user), show_alert=True)
|
||||
# ==============================================================================================================================
|
||||
await clb.answer(
|
||||
locale("spoiler_sent", "callback", locale=clb.from_user), show_alert=True
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -1,6 +1,11 @@
|
||||
from datetime import datetime
|
||||
from app import app
|
||||
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, ForceReply, CallbackQuery
|
||||
from pyrogram.types import (
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
ForceReply,
|
||||
CallbackQuery,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from pyrogram import filters
|
||||
from classes.errors.holo_user import LabelSettingError
|
||||
@ -8,46 +13,79 @@ from classes.holo_user import HoloUser
|
||||
from modules.utils import configGet, locale, logWrite, should_quote
|
||||
from modules.database import col_tmp, col_sponsorships
|
||||
|
||||
|
||||
# Callbacks sponsorship ========================================================================================================
|
||||
@app.on_callback_query(filters.regex("sponsor_apply_[\s\S]*"))
|
||||
async def callback_query_sponsor_apply(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
if holo_user.application_state()[0] == "fill":
|
||||
await clb.message.reply_text(locale("finish_application", "message"), quote=should_quote(clb.message))
|
||||
await clb.message.reply_text(
|
||||
locale("finish_application", "message"), quote=should_quote(clb.message)
|
||||
)
|
||||
return
|
||||
|
||||
logWrite(f"User {holo_user.id} applied for sponsorship")
|
||||
|
||||
holo_user.sponsorship_restart()
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("sponsor_started", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sponsor_started", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(
|
||||
text=locale("sponsorship_applying", "message", locale=holo_user),
|
||||
reply_markup=InlineKeyboardMarkup(edited_markup),
|
||||
)
|
||||
await app.send_message(
|
||||
holo_user.id,
|
||||
locale(f"sponsor1", "message", locale=holo_user),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(locale(f"sponsor1", "force_reply", locale=holo_user.locale))
|
||||
),
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sponsor_started", "callback", locale=holo_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=False,
|
||||
)
|
||||
|
||||
await clb.message.edit(text=locale("sponsorship_applying", "message", locale=holo_user), reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await app.send_message(holo_user.id, locale(f"sponsor1", "message", locale=holo_user), reply_markup=ForceReply(placeholder=str(locale(f"sponsor1", "force_reply", locale=holo_user.locale))))
|
||||
await clb.answer(text=locale("sponsor_started", "callback", locale=holo_user).format(holo_user.id), show_alert=False)
|
||||
|
||||
@app.on_callback_query(filters.regex("sponsor_yes_[\s\S]*"))
|
||||
async def callback_query_sponsor_yes(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("sponsor_approved_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
await app.send_message(holo_user.id, locale("sponsor_approved", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("sponsor_approved_by", "message").format(
|
||||
clb.from_user.first_name, holo_user.id
|
||||
),
|
||||
disable_notification=True,
|
||||
)
|
||||
await app.send_message(
|
||||
holo_user.id, locale("sponsor_approved", "message", locale=holo_user)
|
||||
)
|
||||
logWrite(f"User {holo_user.id} got sponsorship approved by {clb.from_user.id}")
|
||||
|
||||
if col_sponsorships.find_one({"user": holo_user.id}) is not None:
|
||||
col_sponsorships.update_one({"user": holo_user.id},
|
||||
col_sponsorships.update_one(
|
||||
{"user": holo_user.id},
|
||||
{
|
||||
"$set": {
|
||||
"date": datetime.now(),
|
||||
"admin": clb.from_user.id,
|
||||
"sponsorship": col_tmp.find_one({"user": holo_user.id, "type": "sponsorship"})["sponsorship"]
|
||||
"sponsorship": col_tmp.find_one(
|
||||
{"user": holo_user.id, "type": "sponsorship"}
|
||||
)["sponsorship"],
|
||||
}
|
||||
}
|
||||
},
|
||||
)
|
||||
else:
|
||||
col_sponsorships.insert_one(
|
||||
@ -55,49 +93,84 @@ async def callback_query_sponsor_yes(app: Client, clb: CallbackQuery):
|
||||
"user": holo_user.id,
|
||||
"date": datetime.now(),
|
||||
"admin": clb.from_user.id,
|
||||
"sponsorship": col_tmp.find_one({"user": holo_user.id, "type": "sponsorship"})["sponsorship"]
|
||||
"sponsorship": col_tmp.find_one(
|
||||
{"user": holo_user.id, "type": "sponsorship"}
|
||||
)["sponsorship"],
|
||||
}
|
||||
)
|
||||
|
||||
col_tmp.update_one({"user": holo_user.id, "type":"sponsorship"},
|
||||
{
|
||||
"$set": {
|
||||
"state": "approved",
|
||||
"sent": False
|
||||
}
|
||||
}
|
||||
|
||||
col_tmp.update_one(
|
||||
{"user": holo_user.id, "type": "sponsorship"},
|
||||
{"$set": {"state": "approved", "sent": False}},
|
||||
)
|
||||
|
||||
try:
|
||||
await holo_user.label_set(configGet("users", "groups"), col_tmp.find_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "sponsorship"}})["sponsorship"]["label"])
|
||||
await holo_user.label_set(
|
||||
configGet("users", "groups"),
|
||||
col_tmp.find_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "sponsorship"}}
|
||||
)["sponsorship"]["label"],
|
||||
)
|
||||
except LabelSettingError as exp:
|
||||
await app.send_message(configGet("admin", "groups"), exp.__str__(), disable_notification=True)
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"), exp.__str__(), disable_notification=True
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("accepted", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await app.edit_message_caption(
|
||||
clb.message.chat.id,
|
||||
clb.message.id,
|
||||
caption=clb.message.caption,
|
||||
reply_markup=InlineKeyboardMarkup(edited_markup),
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sponsor_accepted", "callback").format(fullclb[2]), show_alert=False
|
||||
)
|
||||
|
||||
await app.edit_message_caption(clb.message.chat.id, clb.message.id, caption=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sponsor_accepted", "callback").format(fullclb[2]), show_alert=False)
|
||||
|
||||
@app.on_callback_query(filters.regex("sponsor_no_[\s\S]*"))
|
||||
async def callback_query_sponsor_no(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("sponsor_rejected_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
await app.send_message(holo_user.id, locale("sponsor_rejected", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("sponsor_rejected_by", "message").format(
|
||||
clb.from_user.first_name, holo_user.id
|
||||
),
|
||||
disable_notification=True,
|
||||
)
|
||||
await app.send_message(
|
||||
holo_user.id, locale("sponsor_rejected", "message", locale=holo_user)
|
||||
)
|
||||
logWrite(f"User {holo_user.id} got sponsorship rejected by {clb.from_user.id}")
|
||||
|
||||
col_tmp.update_one({"user": holo_user.id, "type": "sponsorship"},
|
||||
{
|
||||
"$set": {
|
||||
"state": "rejected",
|
||||
"sent": False
|
||||
}
|
||||
}
|
||||
col_tmp.update_one(
|
||||
{"user": holo_user.id, "type": "sponsorship"},
|
||||
{"$set": {"state": "rejected", "sent": False}},
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("declined", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("declined", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await app.edit_message_caption(clb.message.chat.id, clb.message.id, caption=clb.message.caption, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sponsor_rejected", "callback").format(fullclb[2]), show_alert=False)
|
||||
await app.edit_message_caption(
|
||||
clb.message.chat.id,
|
||||
clb.message.id,
|
||||
caption=clb.message.caption,
|
||||
reply_markup=InlineKeyboardMarkup(edited_markup),
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sponsor_rejected", "callback").format(fullclb[2]), show_alert=False
|
||||
)
|
||||
|
@ -8,14 +8,18 @@ from modules.utils import configGet, locale, logWrite
|
||||
from modules.database import col_tmp, col_applications
|
||||
from modules.commands.rules import DefaultRulesMarkup
|
||||
|
||||
|
||||
# Callbacks application ========================================================================================================
|
||||
@app.on_callback_query(filters.regex("sub_yes_[\s\S]*"))
|
||||
async def callback_query_accept(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("approved_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("approved_by", "message").format(clb.from_user.first_name, holo_user.id),
|
||||
disable_notification=True,
|
||||
)
|
||||
logWrite(f"User {holo_user.id} got approved by {clb.from_user.id}")
|
||||
|
||||
need_link = True
|
||||
@ -23,65 +27,167 @@ async def callback_query_accept(app: Client, clb: CallbackQuery):
|
||||
async for member in app.get_chat_members(configGet("users", "groups")):
|
||||
if member.user.id == holo_user.id:
|
||||
need_link = False
|
||||
|
||||
|
||||
if need_link:
|
||||
link = await app.create_chat_invite_link(configGet("users", "groups"), name=f"Invite for {holo_user.id}", member_limit=1) #, expire_date=datetime.now()+timedelta(days=1))
|
||||
link = await app.create_chat_invite_link(
|
||||
configGet("users", "groups"),
|
||||
name=f"Invite for {holo_user.id}",
|
||||
member_limit=1,
|
||||
) # , expire_date=datetime.now()+timedelta(days=1))
|
||||
|
||||
await app.send_message(holo_user.id, locale("read_rules", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
holo_user.id, locale("read_rules", "message", locale=holo_user)
|
||||
)
|
||||
|
||||
await app.send_message(holo_user.id, locale("rules_msg", locale=holo_user), disable_web_page_preview=True, reply_markup=DefaultRulesMarkup(holo_user).keyboard)
|
||||
await app.send_message(
|
||||
holo_user.id,
|
||||
locale("rules_msg", locale=holo_user),
|
||||
disable_web_page_preview=True,
|
||||
reply_markup=DefaultRulesMarkup(holo_user).keyboard,
|
||||
)
|
||||
|
||||
await app.send_message(holo_user.id, locale("approved", "message", locale=holo_user), reply_markup=InlineKeyboardMarkup(
|
||||
[[
|
||||
InlineKeyboardButton(str(locale("join", "button", locale=holo_user)), url=link.invite_link)
|
||||
]]
|
||||
))
|
||||
await app.send_message(
|
||||
holo_user.id,
|
||||
locale("approved", "message", locale=holo_user),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
str(locale("join", "button", locale=holo_user)),
|
||||
url=link.invite_link,
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
holo_user.set("link", link.invite_link)
|
||||
logWrite(f"User {holo_user.id} got an invite link {link.invite_link}")
|
||||
|
||||
else:
|
||||
await app.send_message(holo_user.id, locale("approved_joined", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
holo_user.id, locale("approved_joined", "message", locale=holo_user)
|
||||
)
|
||||
|
||||
col_applications.insert_one({"user": holo_user.id, "date": datetime.now(), "admin": clb.from_user.id, "application": col_tmp.find_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}})["application"]})
|
||||
col_tmp.update_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}, {"$set": {"state": "approved", "sent": False}})
|
||||
col_applications.insert_one(
|
||||
{
|
||||
"user": holo_user.id,
|
||||
"date": datetime.now(),
|
||||
"admin": clb.from_user.id,
|
||||
"application": col_tmp.find_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}
|
||||
)["application"],
|
||||
}
|
||||
)
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}},
|
||||
{"$set": {"state": "approved", "sent": False}},
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("accepted", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("accepted", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sub_accepted", "callback", locale=clb.from_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sub_accepted", "callback", locale=clb.from_user).format(holo_user.id), show_alert=True)
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_no_[\s\S]*"))
|
||||
async def callback_query_reject(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("rejected_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
await app.send_message(holo_user.id, locale("rejected", "message", locale=holo_user))
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("rejected_by", "message").format(clb.from_user.first_name, holo_user.id),
|
||||
disable_notification=True,
|
||||
)
|
||||
await app.send_message(
|
||||
holo_user.id, locale("rejected", "message", locale=holo_user)
|
||||
)
|
||||
logWrite(f"User {holo_user.id} got rejected by {clb.from_user.id}")
|
||||
|
||||
col_tmp.update_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}, {"$set": {"state": "rejected", "sent": False}})
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}},
|
||||
{"$set": {"state": "rejected", "sent": False}},
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("declined", "button")), callback_data="nothing")], [InlineKeyboardButton(text=str(locale("ban", "button")), callback_data=f"ban_{fullclb[2]}")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("declined", "button")), callback_data="nothing"
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("ban", "button")), callback_data=f"ban_{fullclb[2]}"
|
||||
)
|
||||
],
|
||||
]
|
||||
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sub_rejected", "callback", locale=clb.from_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sub_rejected", "callback", locale=clb.from_user).format(holo_user.id), show_alert=True)
|
||||
|
||||
@app.on_callback_query(filters.regex("sub_russian_[\s\S]*"))
|
||||
async def callback_query_reject_russian(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("rejected_by_rus", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
await app.send_message(holo_user.id, locale("rejected_russian", "message", locale=holo_user))
|
||||
logWrite(f"User {holo_user.id} got rejected by {clb.from_user.id} due to being russian")
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("rejected_by_rus", "message").format(
|
||||
clb.from_user.first_name, holo_user.id
|
||||
),
|
||||
disable_notification=True,
|
||||
)
|
||||
await app.send_message(
|
||||
holo_user.id, locale("rejected_russian", "message", locale=holo_user)
|
||||
)
|
||||
logWrite(
|
||||
f"User {holo_user.id} got rejected by {clb.from_user.id} due to being russian"
|
||||
)
|
||||
|
||||
col_tmp.update_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}, {"$set": {"state": "rejected", "sent": False}})
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}},
|
||||
{"$set": {"state": "rejected", "sent": False}},
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("declined", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("declined", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sub_russian", "callback", locale=clb.from_user).format(holo_user.id), show_alert=True)
|
||||
# ==============================================================================================================================
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sub_russian", "callback", locale=clb.from_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -1,49 +1,104 @@
|
||||
from app import app
|
||||
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, ChatPermissions, CallbackQuery
|
||||
from pyrogram.types import (
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
ChatPermissions,
|
||||
CallbackQuery,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from pyrogram import filters
|
||||
from classes.holo_user import HoloUser
|
||||
from modules.utils import configGet, locale, logWrite
|
||||
from modules.database import col_tmp
|
||||
|
||||
|
||||
# Callbacks sus users ==========================================================================================================
|
||||
@app.on_callback_query(filters.regex("sus_allow_[\s\S]*"))
|
||||
async def callback_query_sus_allow(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("sus_allowed_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
logWrite(f"User {holo_user.id} was allowed to join with another link by {clb.from_user.id}")
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("sus_allowed_by", "message").format(
|
||||
clb.from_user.first_name, holo_user.id
|
||||
),
|
||||
disable_notification=True,
|
||||
)
|
||||
logWrite(
|
||||
f"User {holo_user.id} was allowed to join with another link by {clb.from_user.id}"
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("sus_allowed", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sus_allowed", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sus_allowed", "callback", locale=clb.from_user).format(holo_user.id), show_alert=True)
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sus_allowed", "callback", locale=clb.from_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
await app.restrict_chat_member(configGet("users", "groups"), holo_user.id, permissions=ChatPermissions(
|
||||
await app.restrict_chat_member(
|
||||
configGet("users", "groups"),
|
||||
holo_user.id,
|
||||
permissions=ChatPermissions(
|
||||
can_send_messages=True,
|
||||
can_send_media_messages=True,
|
||||
can_send_other_messages=True,
|
||||
can_send_polls=True
|
||||
)
|
||||
can_send_polls=True,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
@app.on_callback_query(filters.regex("sus_reject_[\s\S]*"))
|
||||
async def callback_query_sus_reject(app: Client, clb: CallbackQuery):
|
||||
|
||||
fullclb = clb.data.split("_")
|
||||
holo_user = HoloUser(int(fullclb[2]))
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("sus_rejected_by", "message").format(clb.from_user.first_name, holo_user.id), disable_notification=True)
|
||||
logWrite(f"User {holo_user.id} was rejected to join with another link by {clb.from_user.id}")
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("sus_rejected_by", "message").format(
|
||||
clb.from_user.first_name, holo_user.id
|
||||
),
|
||||
disable_notification=True,
|
||||
)
|
||||
logWrite(
|
||||
f"User {holo_user.id} was rejected to join with another link by {clb.from_user.id}"
|
||||
)
|
||||
|
||||
edited_markup = [[InlineKeyboardButton(text=str(locale("sus_rejected", "button")), callback_data="nothing")]]
|
||||
edited_markup = [
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sus_rejected", "button")), callback_data="nothing"
|
||||
)
|
||||
]
|
||||
]
|
||||
|
||||
await clb.message.edit(text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup))
|
||||
await clb.answer(text=locale("sus_rejected", "callback", locale=clb.from_user).format(holo_user.id), show_alert=True)
|
||||
await clb.message.edit(
|
||||
text=clb.message.text, reply_markup=InlineKeyboardMarkup(edited_markup)
|
||||
)
|
||||
await clb.answer(
|
||||
text=locale("sus_rejected", "callback", locale=clb.from_user).format(
|
||||
holo_user.id
|
||||
),
|
||||
show_alert=True,
|
||||
)
|
||||
|
||||
await app.ban_chat_member(configGet("users", "groups"), holo_user.id)
|
||||
|
||||
col_tmp.update_one({"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}}, {"$set": {"state": "rejected", "sent": False}})
|
||||
# ==============================================================================================================================
|
||||
col_tmp.update_one(
|
||||
{"user": {"$eq": holo_user.id}, "type": {"$eq": "application"}},
|
||||
{"$set": {"state": "rejected", "sent": False}},
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -12,51 +12,92 @@ from dateutil.relativedelta import relativedelta
|
||||
from modules.database import col_applications
|
||||
from modules import custom_filters
|
||||
|
||||
|
||||
# Application command ==========================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.command(["application"], prefixes=["/"]) & custom_filters.admin)
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.command(["application"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_application(app: Client, msg: Message):
|
||||
|
||||
try:
|
||||
|
||||
try:
|
||||
holo_user = HoloUser(int(msg.command[1]))
|
||||
except (ValueError, UserNotFoundError):
|
||||
try:
|
||||
holo_user = HoloUser((await app.get_users(msg.command[1])).id)
|
||||
except (bad_request_400.UsernameInvalid, bad_request_400.PeerIdInvalid, bad_request_400.UsernameNotOccupied):
|
||||
await msg.reply_text(locale("no_user_application", "message", locale=msg.from_user).format(msg.command[1]), quote=should_quote(msg))
|
||||
except (
|
||||
bad_request_400.UsernameInvalid,
|
||||
bad_request_400.PeerIdInvalid,
|
||||
bad_request_400.UsernameNotOccupied,
|
||||
):
|
||||
await msg.reply_text(
|
||||
locale(
|
||||
"no_user_application", "message", locale=msg.from_user
|
||||
).format(msg.command[1]),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
return
|
||||
|
||||
application = col_applications.find_one({"user": holo_user.id})
|
||||
|
||||
if application is None:
|
||||
logWrite(f"User {msg.from_user.id} requested application of {holo_user.id} but user does not exists")
|
||||
await msg.reply_text(locale("user_invalid", "message", locale=msg.from_user), quote=should_quote(msg))
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} requested application of {holo_user.id} but user does not exists"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("user_invalid", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
return
|
||||
|
||||
application_content = []
|
||||
i = 1
|
||||
|
||||
for question in application['application']:
|
||||
|
||||
for question in application["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), application['application']['2'])
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(datetime.now(), application["application"]["2"])
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if application['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application']['3']['name']}")
|
||||
if application["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application']['3']['name']}"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application'][question]}")
|
||||
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=msg.from_user)} {application['application'][question]}"
|
||||
)
|
||||
|
||||
i += 1
|
||||
|
||||
application_status = locale("application_status_accepted", "message", locale=msg.from_user).format((await app.get_users(application["admin"])).first_name, application["date"].strftime("%d.%m.%Y, %H:%M"))
|
||||
application_status = locale(
|
||||
"application_status_accepted", "message", locale=msg.from_user
|
||||
).format(
|
||||
(await app.get_users(application["admin"])).first_name,
|
||||
application["date"].strftime("%d.%m.%Y, %H:%M"),
|
||||
)
|
||||
|
||||
logWrite(f"User {msg.from_user.id} requested application of {holo_user.id}")
|
||||
await msg.reply_text(locale("contact", "message", locale=msg.from_user).format(holo_user.id, "\n".join(application_content), application_status), parse_mode=ParseMode.MARKDOWN, quote=should_quote(msg))
|
||||
|
||||
await msg.reply_text(
|
||||
locale("contact", "message", locale=msg.from_user).format(
|
||||
holo_user.id, "\n".join(application_content), application_status
|
||||
),
|
||||
parse_mode=ParseMode.MARKDOWN,
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
|
||||
except IndexError:
|
||||
await msg.reply_text(locale("application_invalid_syntax", "message", locale=msg.from_user), quote=should_quote(msg))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("application_invalid_syntax", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -10,10 +10,15 @@ from modules.utils import should_quote, jsonSave
|
||||
from modules.database import col_applications
|
||||
from modules import custom_filters
|
||||
|
||||
# Applications command =========================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.command(["applications"], prefixes=["/"]) & custom_filters.admin)
|
||||
async def cmd_applications(app: Client, msg: Message):
|
||||
|
||||
# Applications command =========================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.command(["applications"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_applications(app: Client, msg: Message):
|
||||
logWrite(f"Admin {msg.from_user.id} requested export of a database")
|
||||
await app.send_chat_action(msg.chat.id, ChatAction.UPLOAD_DOCUMENT)
|
||||
filename = uuid1()
|
||||
@ -21,10 +26,18 @@ async def cmd_applications(app: Client, msg: Message):
|
||||
for entry in col_applications.find():
|
||||
del entry["_id"]
|
||||
entry["date"] = entry["date"].strftime("%d.%m.%Y, %H:%M")
|
||||
entry["application"]["2"] = entry["application"]["2"].strftime("%d.%m.%Y, %H:%M")
|
||||
entry["application"]["2"] = entry["application"]["2"].strftime(
|
||||
"%d.%m.%Y, %H:%M"
|
||||
)
|
||||
output.append(entry)
|
||||
makedirs("tmp", exist_ok=True)
|
||||
jsonSave(output, f"tmp{sep}{filename}.json")
|
||||
await msg.reply_document(document=f"tmp{sep}{filename}.json", file_name="applications", quote=should_quote(msg))
|
||||
await msg.reply_document(
|
||||
document=f"tmp{sep}{filename}.json",
|
||||
file_name="applications",
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
remove(f"tmp{sep}{filename}.json")
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -6,14 +6,30 @@ from modules.utils import should_quote, logWrite, locale
|
||||
from modules.database import col_tmp, col_spoilers, col_applications
|
||||
from modules import custom_filters
|
||||
|
||||
|
||||
# Cancel command ===============================================================================================================
|
||||
@app.on_message((custom_filters.enabled_applications | custom_filters.enabled_sponsorships) & ~filters.scheduled & filters.command("cancel", prefixes=["/"]) & ~custom_filters.banned)
|
||||
@app.on_message(
|
||||
(custom_filters.enabled_applications | custom_filters.enabled_sponsorships)
|
||||
& ~filters.scheduled
|
||||
& filters.command("cancel", prefixes=["/"])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def command_cancel(app: Client, msg: Message):
|
||||
col_tmp.delete_many( {"user": msg.from_user.id, "sent": False} )
|
||||
col_spoilers.delete_many( {"user": msg.from_user.id, "completed": False} )
|
||||
if col_applications.find_one( {"user": msg.from_user.id} ) is None:
|
||||
await msg.reply_text(locale("cancel_reapply", "message", locale=msg.from_user), quote=should_quote(msg), reply_markup=ReplyKeyboardRemove())
|
||||
col_tmp.delete_many({"user": msg.from_user.id, "sent": False})
|
||||
col_spoilers.delete_many({"user": msg.from_user.id, "completed": False})
|
||||
if col_applications.find_one({"user": msg.from_user.id}) is None:
|
||||
await msg.reply_text(
|
||||
locale("cancel_reapply", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
reply_markup=ReplyKeyboardRemove(),
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(locale("cancel", "message", locale=msg.from_user), quote=should_quote(msg), reply_markup=ReplyKeyboardRemove())
|
||||
await msg.reply_text(
|
||||
locale("cancel", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
reply_markup=ReplyKeyboardRemove(),
|
||||
)
|
||||
logWrite(f"Cancelling all ongoing tmp operations for {msg.from_user.id}")
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,15 +7,30 @@ from pyrogram.errors import bad_request_400
|
||||
from pyrogram.enums.chat_action import ChatAction
|
||||
from classes.errors.holo_user import UserNotFoundError, UserInvalidError
|
||||
from classes.holo_user import HoloUser
|
||||
from modules.utils import jsonLoad, should_quote, logWrite, locale, download_tmp, create_tmp, find_user
|
||||
from modules.utils import (
|
||||
jsonLoad,
|
||||
should_quote,
|
||||
logWrite,
|
||||
locale,
|
||||
download_tmp,
|
||||
create_tmp,
|
||||
find_user,
|
||||
)
|
||||
from modules import custom_filters
|
||||
|
||||
# Identify command =============================================================================================================
|
||||
@app.on_message((custom_filters.enabled_applications | custom_filters.enabled_sponsorships) & ~filters.scheduled & filters.command("identify", prefixes=["/"]) & custom_filters.admin)
|
||||
async def cmd_identify(app: Client, msg: Message):
|
||||
|
||||
# Identify command =============================================================================================================
|
||||
@app.on_message(
|
||||
(custom_filters.enabled_applications | custom_filters.enabled_sponsorships)
|
||||
& ~filters.scheduled
|
||||
& filters.command("identify", prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_identify(app: Client, msg: Message):
|
||||
if len(msg.command) != 2:
|
||||
await msg.reply_text(locale("identify_invalid_syntax", "message", locale=msg.from_user))
|
||||
await msg.reply_text(
|
||||
locale("identify_invalid_syntax", "message", locale=msg.from_user)
|
||||
)
|
||||
return
|
||||
|
||||
try:
|
||||
@ -23,17 +38,44 @@ async def cmd_identify(app: Client, msg: Message):
|
||||
holo_user = HoloUser(int(msg.command[1]))
|
||||
except ValueError:
|
||||
holo_user = HoloUser(await find_user(app, msg.command[1]))
|
||||
except (UserInvalidError, UserNotFoundError, bad_request_400.UsernameInvalid, bad_request_400.PeerIdInvalid, bad_request_400.UsernameNotOccupied, TypeError):
|
||||
await msg.reply_text(locale("identify_not_found", "message", locale=msg.from_user).format(msg.command[1]))
|
||||
except (
|
||||
UserInvalidError,
|
||||
UserNotFoundError,
|
||||
bad_request_400.UsernameInvalid,
|
||||
bad_request_400.PeerIdInvalid,
|
||||
bad_request_400.UsernameNotOccupied,
|
||||
TypeError,
|
||||
):
|
||||
await msg.reply_text(
|
||||
locale("identify_not_found", "message", locale=msg.from_user).format(
|
||||
msg.command[1]
|
||||
)
|
||||
)
|
||||
return
|
||||
|
||||
role = holo_user.label
|
||||
has_application = locale("yes", "message", locale=msg.from_user) if holo_user.application_approved() is True else locale("no", "message", locale=msg.from_user)
|
||||
has_sponsorship = locale("yes", "message", locale=msg.from_user) if holo_user.sponsorship_valid() is True else locale("no", "message", locale=msg.from_user)
|
||||
has_application = (
|
||||
locale("yes", "message", locale=msg.from_user)
|
||||
if holo_user.application_approved() is True
|
||||
else locale("no", "message", locale=msg.from_user)
|
||||
)
|
||||
has_sponsorship = (
|
||||
locale("yes", "message", locale=msg.from_user)
|
||||
if holo_user.sponsorship_valid() is True
|
||||
else locale("no", "message", locale=msg.from_user)
|
||||
)
|
||||
|
||||
username = holo_user.username if holo_user.username is not None else "N/A"
|
||||
in_chat = locale("yes", "message", locale=msg.from_user) if (holo_user.id in jsonLoad(path.join("cache", "group_members"))) else locale("no", "message", locale=msg.from_user)
|
||||
is_admin = locale("yes", "message", locale=msg.from_user) if (await isAnAdmin(holo_user.id)) else locale("no", "message", locale=msg.from_user)
|
||||
in_chat = (
|
||||
locale("yes", "message", locale=msg.from_user)
|
||||
if (holo_user.id in jsonLoad(path.join("cache", "group_members")))
|
||||
else locale("no", "message", locale=msg.from_user)
|
||||
)
|
||||
is_admin = (
|
||||
locale("yes", "message", locale=msg.from_user)
|
||||
if (await isAnAdmin(holo_user.id))
|
||||
else locale("no", "message", locale=msg.from_user)
|
||||
)
|
||||
|
||||
output = locale("identify_success", "message", locale=msg.from_user).format(
|
||||
holo_user.id,
|
||||
@ -43,24 +85,25 @@ async def cmd_identify(app: Client, msg: Message):
|
||||
is_admin,
|
||||
role,
|
||||
has_application,
|
||||
has_sponsorship
|
||||
has_sponsorship,
|
||||
)
|
||||
|
||||
|
||||
user = await app.get_users(holo_user.id)
|
||||
|
||||
if user.photo is not None:
|
||||
await app.send_chat_action(msg.chat.id, action=ChatAction.UPLOAD_PHOTO)
|
||||
await msg.reply_photo(
|
||||
create_tmp((await download_tmp(app, user.photo.big_file_id))[1], kind="image"),
|
||||
create_tmp(
|
||||
(await download_tmp(app, user.photo.big_file_id))[1], kind="image"
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
caption=output
|
||||
caption=output,
|
||||
)
|
||||
else:
|
||||
await app.send_chat_action(msg.chat.id, action=ChatAction.TYPING)
|
||||
await msg.reply_text(
|
||||
output,
|
||||
quote=should_quote(msg)
|
||||
)
|
||||
await msg.reply_text(output, quote=should_quote(msg))
|
||||
|
||||
logWrite(f"User {msg.from_user.id} identified user {holo_user.id}")
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -9,13 +9,28 @@ from classes.holo_user import HoloUser
|
||||
|
||||
|
||||
# Issue command ================================================================================================================
|
||||
@app.on_message(custom_filters.enabled_general & ~filters.scheduled & filters.private & filters.command(["issue"], prefixes=["/"]) & ~custom_filters.banned)
|
||||
@app.on_message(
|
||||
custom_filters.enabled_general
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(["issue"], prefixes=["/"])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def cmd_issue(app: Client, msg: Message):
|
||||
await msg.reply_text(locale("issue", "message", locale=msg.from_user), disable_web_page_preview=True, reply_markup=InlineKeyboardMarkup(
|
||||
await msg.reply_text(
|
||||
locale("issue", "message", locale=msg.from_user),
|
||||
disable_web_page_preview=True,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(locale("issue", "button", locale=msg.from_user), url=configGet("issues"))
|
||||
InlineKeyboardButton(
|
||||
locale("issue", "button", locale=msg.from_user),
|
||||
url=configGet("issues"),
|
||||
)
|
||||
]
|
||||
]
|
||||
))
|
||||
# ==============================================================================================================================
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,10 +7,15 @@ from classes.errors.holo_user import LabelTooLongError, LabelSettingError
|
||||
from classes.holo_user import HoloUser
|
||||
from modules import custom_filters
|
||||
|
||||
# Label command ================================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.command(["label"], prefixes=["/"]) & custom_filters.admin)
|
||||
async def cmd_label(app: Client, msg: Message):
|
||||
|
||||
# Label command ================================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.command(["label"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_label(app: Client, msg: Message):
|
||||
if len(msg.command) < 3:
|
||||
await msg.reply_text("Invalid syntax:\n`/label USER LABEL`")
|
||||
return
|
||||
@ -18,15 +23,16 @@ async def cmd_label(app: Client, msg: Message):
|
||||
target = await find_user(app, msg.command[1])
|
||||
|
||||
if target is not None:
|
||||
|
||||
target = HoloUser(target)
|
||||
|
||||
label = " ".join(msg.command[2:])
|
||||
|
||||
if label.lower() == "reset":
|
||||
await target.label_reset(msg.chat)
|
||||
await msg.reply_text(f"Resetting **{target.id}**'s label...", quote=should_quote(msg))
|
||||
|
||||
await msg.reply_text(
|
||||
f"Resetting **{target.id}**'s label...", quote=should_quote(msg)
|
||||
)
|
||||
|
||||
else:
|
||||
try:
|
||||
await target.label_set(msg.chat, label)
|
||||
@ -34,10 +40,19 @@ async def cmd_label(app: Client, msg: Message):
|
||||
await msg.reply_text(locale("label_too_long", "message"))
|
||||
return
|
||||
except LabelSettingError as exp:
|
||||
await app.send_message(configGet("admin", "groups"), exp.__str__(), disable_notification=True)
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
exp.__str__(),
|
||||
disable_notification=True,
|
||||
)
|
||||
return
|
||||
await msg.reply_text(f"Setting **{target.id}**'s label to **{label}**...", quote=should_quote(msg))
|
||||
|
||||
await msg.reply_text(
|
||||
f"Setting **{target.id}**'s label to **{label}**...",
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
|
||||
else:
|
||||
await msg.reply_text(f"User not found")
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,28 +7,70 @@ from classes.holo_user import HoloUser
|
||||
from modules.utils import logWrite, locale, should_quote, find_user
|
||||
from modules import custom_filters
|
||||
|
||||
|
||||
# Message command ==============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_general & ~filters.scheduled & filters.command(["message"], prefixes=["/"]) & custom_filters.admin)
|
||||
@app.on_message(
|
||||
custom_filters.enabled_general
|
||||
& ~filters.scheduled
|
||||
& filters.command(["message"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_message(app: Client, msg: Message):
|
||||
|
||||
try:
|
||||
|
||||
try:
|
||||
destination = HoloUser(int(msg.command[1]))
|
||||
except (ValueError, UserInvalidError):
|
||||
destination = HoloUser(await find_user(app, query=msg.command[1]))
|
||||
|
||||
if ((msg.text is not None) and (len(str(msg.text).split()) > 2)):
|
||||
await destination.message(context=msg, text=" ".join(str(msg.text).split()[2:]), caption=msg.caption, photo=msg.photo, video=msg.video, file=msg.document, voice=msg.voice, animation=msg.animation, adm_context=True)
|
||||
elif ((msg.caption is not None) and (len(msg.caption.split()) > 2)):
|
||||
await destination.message(context=msg, text=str(msg.text), caption=" ".join(msg.caption.split()[2:]), photo=msg.photo, video=msg.video, file=msg.document, voice=msg.voice, animation=msg.animation, adm_context=True)
|
||||
if (msg.text is not None) and (len(str(msg.text).split()) > 2):
|
||||
await destination.message(
|
||||
context=msg,
|
||||
text=" ".join(str(msg.text).split()[2:]),
|
||||
caption=msg.caption,
|
||||
photo=msg.photo,
|
||||
video=msg.video,
|
||||
file=msg.document,
|
||||
voice=msg.voice,
|
||||
animation=msg.animation,
|
||||
adm_context=True,
|
||||
)
|
||||
elif (msg.caption is not None) and (len(msg.caption.split()) > 2):
|
||||
await destination.message(
|
||||
context=msg,
|
||||
text=str(msg.text),
|
||||
caption=" ".join(msg.caption.split()[2:]),
|
||||
photo=msg.photo,
|
||||
video=msg.video,
|
||||
file=msg.document,
|
||||
voice=msg.voice,
|
||||
animation=msg.animation,
|
||||
adm_context=True,
|
||||
)
|
||||
else:
|
||||
await destination.message(context=msg, text=None, caption=None, photo=msg.photo, video=msg.video, file=msg.document, voice=msg.voice, animation=msg.animation, adm_context=True)
|
||||
|
||||
await destination.message(
|
||||
context=msg,
|
||||
text=None,
|
||||
caption=None,
|
||||
photo=msg.photo,
|
||||
video=msg.video,
|
||||
file=msg.document,
|
||||
voice=msg.voice,
|
||||
animation=msg.animation,
|
||||
adm_context=True,
|
||||
)
|
||||
|
||||
except IndexError:
|
||||
await msg.reply_text(locale("message_invalid_syntax", "message", locale=msg.from_user), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("message_invalid_syntax", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
logWrite(f"Admin {msg.from_user.id} tried to send message but 'IndexError'")
|
||||
except ValueError:
|
||||
await msg.reply_text(locale("message_invalid_syntax", "message", locale=msg.from_user), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("message_invalid_syntax", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
logWrite(f"Admin {msg.from_user.id} tried to send message but 'ValueError'")
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -11,50 +11,111 @@ from modules.utils import configGet, jsonLoad, locale, should_quote, find_locati
|
||||
from modules.database import col_applications, col_users
|
||||
from classes.errors.geo import PlaceNotFoundError
|
||||
|
||||
# Nearby command ===============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & (filters.private | (filters.chat(configGet("admin", "groups")) | filters.chat(configGet("users", "groups")))) & filters.command(["nearby"], prefixes=["/"]) & (custom_filters.allowed | custom_filters.admin) & ~custom_filters.banned)
|
||||
async def cmd_nearby(app: Client, msg: Message):
|
||||
|
||||
# Nearby command ===============================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& (
|
||||
filters.private
|
||||
| (
|
||||
filters.chat(configGet("admin", "groups"))
|
||||
| filters.chat(configGet("users", "groups"))
|
||||
)
|
||||
)
|
||||
& filters.command(["nearby"], prefixes=["/"])
|
||||
& (custom_filters.allowed | custom_filters.admin)
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def cmd_nearby(app: Client, msg: Message):
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
|
||||
# Check if any place provided
|
||||
if len(msg.command) == 1: # Action if no place provided
|
||||
if len(msg.command) == 1: # Action if no place provided
|
||||
application = col_applications.find_one({"user": msg.from_user.id})
|
||||
if application is None:
|
||||
await msg.reply_text(locale("nearby_user_empty", "message", locale=holo_user))
|
||||
await msg.reply_text(
|
||||
locale("nearby_user_empty", "message", locale=holo_user)
|
||||
)
|
||||
return
|
||||
location = application["application"]["3"]["location"][0], application["application"]["3"]["location"][1]
|
||||
else: # Find a place from input query
|
||||
location = (
|
||||
application["application"]["3"]["location"][0],
|
||||
application["application"]["3"]["location"][1],
|
||||
)
|
||||
else: # Find a place from input query
|
||||
logWrite(f"Looking for the location by query '{' '.join(msg.command[1:])}'")
|
||||
try:
|
||||
location_coordinates = find_location(" ".join(msg.command[1:]))
|
||||
location = float(location_coordinates["lng"]), float(location_coordinates["lat"])
|
||||
except PlaceNotFoundError: # Place is not found
|
||||
await msg.reply_text(locale("nearby_invalid", "message", locale=holo_user), quote=should_quote(msg))
|
||||
location = float(location_coordinates["lng"]), float(
|
||||
location_coordinates["lat"]
|
||||
)
|
||||
except PlaceNotFoundError: # Place is not found
|
||||
await msg.reply_text(
|
||||
locale("nearby_invalid", "message", locale=holo_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
return
|
||||
except Exception as exp: # Error occurred while finding the place
|
||||
await msg.reply_text(locale("nearby_error", "message", locale=holo_user).format(exp, print_exc()), quote=should_quote(msg))
|
||||
except Exception as exp: # Error occurred while finding the place
|
||||
await msg.reply_text(
|
||||
locale("nearby_error", "message", locale=holo_user).format(
|
||||
exp, print_exc()
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
return
|
||||
|
||||
# Find all users registered in the area provided
|
||||
output = []
|
||||
applications_nearby = col_applications.find( {"application.3.location": { "$nearSphere": {"$geometry": {"type": "Point", "coordinates": [location[0], location[1]]}, "$maxDistance": configGet("search_radius")*1000} } } )
|
||||
applications_nearby = col_applications.find(
|
||||
{
|
||||
"application.3.location": {
|
||||
"$nearSphere": {
|
||||
"$geometry": {
|
||||
"type": "Point",
|
||||
"coordinates": [location[0], location[1]],
|
||||
},
|
||||
"$maxDistance": configGet("search_radius") * 1000,
|
||||
}
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
for entry in applications_nearby:
|
||||
if not entry["user"] == msg.from_user.id:
|
||||
user = col_users.find_one( {"user": entry["user"]} )
|
||||
user = col_users.find_one({"user": entry["user"]})
|
||||
if user is not None:
|
||||
if entry["user"] in jsonLoad(path.join(configGet("cache", "locations"), "group_members")):
|
||||
if user["tg_username"] not in [None, "None", ""]: # Check if user has any name
|
||||
output.append(f'• **{user["tg_name"]}** (@{user["tg_username"]}):\n - {entry["application"]["3"]["name"]}, {entry["application"]["3"]["adminName1"]}')
|
||||
if entry["user"] in jsonLoad(
|
||||
path.join(configGet("cache", "locations"), "group_members")
|
||||
):
|
||||
if user["tg_username"] not in [
|
||||
None,
|
||||
"None",
|
||||
"",
|
||||
]: # Check if user has any name
|
||||
output.append(
|
||||
f'• **{user["tg_name"]}** (@{user["tg_username"]}):\n - {entry["application"]["3"]["name"]}, {entry["application"]["3"]["adminName1"]}'
|
||||
)
|
||||
else:
|
||||
output.append(f'• **{user["tg_name"]}**:\n - {entry["application"]["3"]["name"]}, {entry["application"]["3"]["adminName1"]}')
|
||||
output.append(
|
||||
f'• **{user["tg_name"]}**:\n - {entry["application"]["3"]["name"]}, {entry["application"]["3"]["adminName1"]}'
|
||||
)
|
||||
|
||||
logWrite(f"{holo_user.id} tried to find someone nearby {location[1]} {location[0]} in the radius of {configGet('search_radius')} kilometers")
|
||||
logWrite(
|
||||
f"{holo_user.id} tried to find someone nearby {location[1]} {location[0]} in the radius of {configGet('search_radius')} kilometers"
|
||||
)
|
||||
|
||||
# Check if any users found
|
||||
if len(output) > 0:
|
||||
await msg.reply_text(locale("nearby_result", "message", locale=holo_user).format("\n".join(output)), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("nearby_result", "message", locale=holo_user).format(
|
||||
"\n".join(output)
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(locale("nearby_empty", "message", locale=holo_user), quote=should_quote(msg))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("nearby_empty", "message", locale=holo_user), quote=should_quote(msg)
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -9,18 +9,32 @@ from modules.handlers.welcome import welcome_pass
|
||||
from modules.database import col_tmp, col_applications
|
||||
from modules import custom_filters
|
||||
|
||||
# Reapply command ==============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.private & filters.command(["reapply"], prefixes=["/"]) & ~custom_filters.banned)
|
||||
async def cmd_reapply(app: Client, msg: Message):
|
||||
|
||||
# Reapply command ==============================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(["reapply"], prefixes=["/"])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def cmd_reapply(app: Client, msg: Message):
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
|
||||
# Check if user has approved/rejected tmp application
|
||||
if ((holo_user.application_state()[0] in ["approved", "rejected"]) or (holo_user.application_state()[0] == "none")) and holo_user.spoiler_state() is False:
|
||||
|
||||
if (
|
||||
(holo_user.application_state()[0] in ["approved", "rejected"])
|
||||
or (holo_user.application_state()[0] == "none")
|
||||
) and holo_user.spoiler_state() is False:
|
||||
# Check if user's tmp application is already completed or even sent
|
||||
if ((holo_user.application_state()[1] is True) and (not col_tmp.find_one({"user": holo_user.id, "type": "application"})["sent"])) or (holo_user.application_state()[0] == "none"):
|
||||
|
||||
if (
|
||||
(holo_user.application_state()[1] is True)
|
||||
and (
|
||||
not col_tmp.find_one({"user": holo_user.id, "type": "application"})[
|
||||
"sent"
|
||||
]
|
||||
)
|
||||
) or (holo_user.application_state()[0] == "none"):
|
||||
left_chat = True
|
||||
|
||||
async for member in app.get_chat_members(configGet("users", "groups")):
|
||||
@ -28,64 +42,132 @@ async def cmd_reapply(app: Client, msg: Message):
|
||||
left_chat = False
|
||||
|
||||
if left_chat is True:
|
||||
|
||||
if (holo_user.application_state()[1] is True and holo_user.application_state()[0] not in ["fill", "rejected"]):
|
||||
if holo_user.application_state()[
|
||||
1
|
||||
] is True and holo_user.application_state()[0] not in [
|
||||
"fill",
|
||||
"rejected",
|
||||
]:
|
||||
await msg.reply_text(
|
||||
locale("reapply_left_chat", "message", locale=holo_user),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"reapply_old_one",
|
||||
"button",
|
||||
locale=holo_user,
|
||||
),
|
||||
f"reapply_old_{msg.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"reapply_new_one",
|
||||
"button",
|
||||
locale=holo_user,
|
||||
),
|
||||
f"reapply_new_{msg.id}",
|
||||
)
|
||||
],
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
await msg.reply_text(locale("reapply_left_chat", "message", locale=holo_user), reply_markup=InlineKeyboardMarkup([
|
||||
[
|
||||
InlineKeyboardButton(locale("reapply_old_one", "button", locale=holo_user), f"reapply_old_{msg.id}")
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(locale("reapply_new_one", "button", locale=holo_user), f"reapply_new_{msg.id}")
|
||||
]
|
||||
]))
|
||||
|
||||
elif col_tmp.find_one({"user": holo_user.id, "type": "application"}) is None and col_applications.find_one({"user": holo_user.id}) is not None:
|
||||
|
||||
await msg.reply_text(locale("reapply_left_chat", "message", locale=holo_user), reply_markup=InlineKeyboardMarkup([
|
||||
[
|
||||
InlineKeyboardButton(locale("reapply_old_one", "button", locale=holo_user), f"reapply_old_{msg.id}")
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(locale("reapply_new_one", "button", locale=holo_user), f"reapply_new_{msg.id}")
|
||||
]
|
||||
]))
|
||||
elif (
|
||||
col_tmp.find_one({"user": holo_user.id, "type": "application"})
|
||||
is None
|
||||
and col_applications.find_one({"user": holo_user.id}) is not None
|
||||
):
|
||||
await msg.reply_text(
|
||||
locale("reapply_left_chat", "message", locale=holo_user),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"reapply_old_one",
|
||||
"button",
|
||||
locale=holo_user,
|
||||
),
|
||||
f"reapply_old_{msg.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"reapply_new_one",
|
||||
"button",
|
||||
locale=holo_user,
|
||||
),
|
||||
f"reapply_new_{msg.id}",
|
||||
)
|
||||
],
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
else:
|
||||
|
||||
holo_user.application_restart(reapply=True)
|
||||
await welcome_pass(app, msg, once_again=True)
|
||||
|
||||
else:
|
||||
|
||||
if holo_user.sponsorship_state()[0] == "fill":
|
||||
await msg.reply_text(locale("finish_sponsorship", "message"), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("finish_sponsorship", "message"), quote=should_quote(msg)
|
||||
)
|
||||
return
|
||||
|
||||
|
||||
holo_user.application_restart(reapply=True)
|
||||
await welcome_pass(app, msg, once_again=True)
|
||||
|
||||
else:
|
||||
|
||||
await msg.reply_text(locale("reapply_in_progress", "message", locale=holo_user).format(locale("confirm", "keyboard", locale=holo_user)[1][0]), reply_markup=InlineKeyboardMarkup([
|
||||
[
|
||||
InlineKeyboardButton(locale("applying_stop", "button", locale=holo_user), f"reapply_stop_{msg.id}")
|
||||
]
|
||||
]))
|
||||
await msg.reply_text(
|
||||
locale("reapply_in_progress", "message", locale=holo_user).format(
|
||||
locale("confirm", "keyboard", locale=holo_user)[1][0]
|
||||
),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale("applying_stop", "button", locale=holo_user),
|
||||
f"reapply_stop_{msg.id}",
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
elif holo_user.spoiler_state() is True:
|
||||
|
||||
await msg.reply_text(locale("spoiler_in_progress", "message", locale=holo_user))
|
||||
|
||||
else:
|
||||
if (holo_user.application_state()[0] == "fill") and (
|
||||
col_tmp.find_one({"user": holo_user.id, "type": "application"})["sent"]
|
||||
is True
|
||||
):
|
||||
await msg.reply_text(
|
||||
locale("reapply_forbidden", "message", locale=holo_user)
|
||||
)
|
||||
|
||||
if (holo_user.application_state()[0] == "fill") and (col_tmp.find_one({"user": holo_user.id, "type": "application"})["sent"] is True):
|
||||
await msg.reply_text(locale("reapply_forbidden", "message", locale=holo_user))
|
||||
|
||||
else:
|
||||
await msg.reply_text(locale("reapply_in_progress", "message", locale=holo_user).format(locale("confirm", "keyboard", locale=holo_user)[1][0]), reply_markup=InlineKeyboardMarkup([
|
||||
[
|
||||
InlineKeyboardButton(locale("applying_stop", "button", locale=holo_user), f"reapply_stop_{msg.id}")
|
||||
]
|
||||
]))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("reapply_in_progress", "message", locale=holo_user).format(
|
||||
locale("confirm", "keyboard", locale=holo_user)[1][0]
|
||||
),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale("applying_stop", "button", locale=holo_user),
|
||||
f"reapply_stop_{msg.id}",
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -11,14 +11,28 @@ from modules import custom_filters
|
||||
|
||||
pid = getpid()
|
||||
|
||||
# Reboot command ===============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_general & ~filters.scheduled & filters.private & filters.command(["kill", "die", "reboot"], prefixes=["/"]) & custom_filters.admin)
|
||||
async def cmd_kill(app: Client, msg: Message):
|
||||
|
||||
# Reboot command ===============================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_general
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(["kill", "die", "reboot"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_kill(app: Client, msg: Message):
|
||||
logWrite(f"Shutting down bot with pid {pid}")
|
||||
await msg.reply_text(locale("shutdown", "message", locale=msg.from_user).format(pid), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("shutdown", "message", locale=msg.from_user).format(pid),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
scheduler.shutdown()
|
||||
makedirs(configGet("cache", "locations"), exist_ok=True)
|
||||
jsonSave({"timestamp": time()}, path.join(configGet("cache", "locations"), "shutdown_time"))
|
||||
jsonSave(
|
||||
{"timestamp": time()},
|
||||
path.join(configGet("cache", "locations"), "shutdown_time"),
|
||||
)
|
||||
exit()
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -9,40 +9,65 @@ from modules import custom_filters
|
||||
|
||||
pid = getpid()
|
||||
|
||||
|
||||
# Reset commands command =======================================================================================================
|
||||
@app.on_message(custom_filters.enabled_general & ~filters.scheduled & filters.private & filters.command(["resetcommands"], prefixes=["/"]) & custom_filters.admin)
|
||||
@app.on_message(
|
||||
custom_filters.enabled_general
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(["resetcommands"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_resetcommands(app: Client, msg: Message):
|
||||
|
||||
if msg.from_user.id == configGet("owner"):
|
||||
|
||||
logWrite(f"Resetting all commands on owner's request")
|
||||
|
||||
valid_locales = []
|
||||
files_locales = listdir(f'{configGet("locale", "locations")}')
|
||||
|
||||
|
||||
for entry in files_locales:
|
||||
if entry.endswith(".json"):
|
||||
valid_locales.append(".".join(entry.split(".")[:-1]))
|
||||
|
||||
logWrite(f'Resetting commands in groups {configGet("admin", "groups")} and {configGet("users", "groups")}', debug=True)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("admin", "groups")))
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("users", "groups")))
|
||||
logWrite(
|
||||
f'Resetting commands in groups {configGet("admin", "groups")} and {configGet("users", "groups")}',
|
||||
debug=True,
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("admin", "groups"))
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("users", "groups"))
|
||||
)
|
||||
for lc in valid_locales:
|
||||
try:
|
||||
logWrite(f'Resetting commands in groups {configGet("admin", "groups")} and {configGet("users", "groups")} [{lc}]', debug=True)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("admin", "groups")), language_code=lc)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("users", "groups")), language_code=lc)
|
||||
logWrite(
|
||||
f'Resetting commands in groups {configGet("admin", "groups")} and {configGet("users", "groups")} [{lc}]',
|
||||
debug=True,
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("admin", "groups")),
|
||||
language_code=lc,
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("users", "groups")),
|
||||
language_code=lc,
|
||||
)
|
||||
except:
|
||||
pass
|
||||
|
||||
for admin in configGet("admins"):
|
||||
try:
|
||||
logWrite(f'Resetting commands for admin {admin}', debug=True)
|
||||
logWrite(f"Resetting commands for admin {admin}", debug=True)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=admin))
|
||||
for lc in valid_locales:
|
||||
try:
|
||||
logWrite(f'Resetting commands for admin {admin} [{lc}]', debug=True)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=admin), language_code=lc)
|
||||
logWrite(
|
||||
f"Resetting commands for admin {admin} [{lc}]", debug=True
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=admin), language_code=lc
|
||||
)
|
||||
except:
|
||||
pass
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
@ -51,21 +76,40 @@ async def cmd_resetcommands(app: Client, msg: Message):
|
||||
try:
|
||||
logWrite(f'Resetting commands for owner {configGet("owner")}', debug=True)
|
||||
for lc in valid_locales:
|
||||
logWrite(f'Resetting commands for owner {configGet("owner")} [{lc}]', debug=True)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("owner")), language_code=lc)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("owner")))
|
||||
logWrite(
|
||||
f'Resetting commands for owner {configGet("owner")} [{lc}]',
|
||||
debug=True,
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("owner")),
|
||||
language_code=lc,
|
||||
)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("owner"))
|
||||
)
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
pass
|
||||
|
||||
for lc in valid_locales:
|
||||
logWrite(f'Resetting commands for locale {lc}', debug=True)
|
||||
await app.delete_bot_commands(scope=BotCommandScopeDefault(), language_code=lc)
|
||||
logWrite(f"Resetting commands for locale {lc}", debug=True)
|
||||
await app.delete_bot_commands(
|
||||
scope=BotCommandScopeDefault(), language_code=lc
|
||||
)
|
||||
|
||||
logWrite(f'Resetting default commands', debug=True)
|
||||
logWrite(f"Resetting default commands", debug=True)
|
||||
await app.delete_bot_commands()
|
||||
|
||||
await msg.reply_text("OK", quote=should_quote(msg))
|
||||
|
||||
logWrite(str(await app.get_bot_commands()), debug=True)
|
||||
logWrite(str(await app.get_bot_commands(scope=BotCommandScopeChat(chat_id=configGet("owner")))), debug=True)
|
||||
# ==============================================================================================================================
|
||||
logWrite(
|
||||
str(
|
||||
await app.get_bot_commands(
|
||||
scope=BotCommandScopeChat(chat_id=configGet("owner"))
|
||||
)
|
||||
),
|
||||
debug=True,
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,36 +7,55 @@ from modules.utils import locale
|
||||
from modules import custom_filters
|
||||
from classes.holo_user import HoloUser
|
||||
|
||||
|
||||
class DefaultRulesMarkup(list):
|
||||
def __init__(self, language_code: Union[str, HoloUser, User, None]):
|
||||
super().__init__([])
|
||||
self.keyboard = InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(locale("rules_home", "button", locale=language_code), callback_data="rules_home"),
|
||||
InlineKeyboardButton(locale("rules_additional", "button", locale=language_code), callback_data="rules_additional")
|
||||
InlineKeyboardButton(
|
||||
locale("rules_home", "button", locale=language_code),
|
||||
callback_data="rules_home",
|
||||
),
|
||||
InlineKeyboardButton(
|
||||
locale("rules_additional", "button", locale=language_code),
|
||||
callback_data="rules_additional",
|
||||
),
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton("1", callback_data="rule_1"),
|
||||
InlineKeyboardButton("2", callback_data="rule_2"),
|
||||
InlineKeyboardButton("3", callback_data="rule_3")
|
||||
InlineKeyboardButton("3", callback_data="rule_3"),
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton("4", callback_data="rule_4"),
|
||||
InlineKeyboardButton("5", callback_data="rule_5"),
|
||||
InlineKeyboardButton("6", callback_data="rule_6")
|
||||
InlineKeyboardButton("6", callback_data="rule_6"),
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton("7", callback_data="rule_7"),
|
||||
InlineKeyboardButton("8", callback_data="rule_8"),
|
||||
InlineKeyboardButton("9", callback_data="rule_9")
|
||||
]
|
||||
InlineKeyboardButton("9", callback_data="rule_9"),
|
||||
],
|
||||
]
|
||||
)
|
||||
|
||||
|
||||
# Rules command =============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_general & ~filters.scheduled & filters.private & ~custom_filters.banned & filters.command(["rules"], prefixes=["/"]))
|
||||
@app.on_message(
|
||||
custom_filters.enabled_general
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& ~custom_filters.banned
|
||||
& filters.command(["rules"], prefixes=["/"])
|
||||
)
|
||||
async def cmd_rules(app: Client, msg: Message):
|
||||
await msg.reply_text(locale("rules_msg", locale=msg.from_user), disable_web_page_preview=True, reply_markup=DefaultRulesMarkup(msg.from_user).keyboard)
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("rules_msg", locale=msg.from_user),
|
||||
disable_web_page_preview=True,
|
||||
reply_markup=DefaultRulesMarkup(msg.from_user).keyboard,
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -9,24 +9,30 @@ from modules.utils import locale
|
||||
from modules.database import col_spoilers, col_applications
|
||||
from modules import custom_filters
|
||||
|
||||
# Spoiler command ==============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_spoilers & ~filters.scheduled & filters.private & ~custom_filters.banned & filters.command(["spoiler"], prefixes=["/"]))
|
||||
async def cmd_spoiler(app: Client, msg: Message):
|
||||
|
||||
# Spoiler command ==============================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_spoilers
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& ~custom_filters.banned
|
||||
& filters.command(["spoiler"], prefixes=["/"])
|
||||
)
|
||||
async def cmd_spoiler(app: Client, msg: Message):
|
||||
try:
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
except (UserInvalidError, UserNotFoundError):
|
||||
return
|
||||
|
||||
if col_applications.find_one( {"user": holo_user.id} ) is None:
|
||||
|
||||
if col_applications.find_one({"user": holo_user.id}) is None:
|
||||
await msg.reply_text(locale("not_member", "message", locale=msg.from_user))
|
||||
return
|
||||
|
||||
if holo_user.application_state()[0] != "fill" and holo_user.sponsorship_state()[0] != "fill":
|
||||
|
||||
if col_spoilers.find_one( {"user": holo_user.id, "completed": False} ) is None:
|
||||
|
||||
if (
|
||||
holo_user.application_state()[0] != "fill"
|
||||
and holo_user.sponsorship_state()[0] != "fill"
|
||||
):
|
||||
if col_spoilers.find_one({"user": holo_user.id, "completed": False}) is None:
|
||||
col_spoilers.insert_one(
|
||||
{
|
||||
"user": holo_user.id,
|
||||
@ -39,13 +45,24 @@ async def cmd_spoiler(app: Client, msg: Message):
|
||||
"animation": None,
|
||||
"document": None,
|
||||
"caption": None,
|
||||
"text": None
|
||||
"text": None,
|
||||
}
|
||||
)
|
||||
|
||||
await msg.reply_text(locale("spoiler_started", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("spoiler_categories", "keyboard"), resize_keyboard=True, one_time_keyboard=True))
|
||||
await msg.reply_text(
|
||||
locale("spoiler_started", "message", locale=msg.from_user),
|
||||
reply_markup=ReplyKeyboardMarkup(
|
||||
locale("spoiler_categories", "keyboard"),
|
||||
resize_keyboard=True,
|
||||
one_time_keyboard=True,
|
||||
),
|
||||
)
|
||||
logWrite(f"User {msg.from_user.id} started creating new spoiler")
|
||||
|
||||
else:
|
||||
await msg.reply_text(locale("spoiler_unfinished", "message", locale=msg.from_user))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("spoiler_unfinished", "message", locale=msg.from_user)
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,17 +7,44 @@ from modules import custom_filters
|
||||
from modules.utils import locale, should_quote
|
||||
from modules.database import col_applications
|
||||
|
||||
|
||||
# Sponsorship command ==========================================================================================================
|
||||
@app.on_message(custom_filters.enabled_sponsorships & ~filters.scheduled & filters.command(["sponsorship"], prefixes=["/"]) & ~custom_filters.banned & (custom_filters.allowed | custom_filters.admin))
|
||||
@app.on_message(
|
||||
custom_filters.enabled_sponsorships
|
||||
& ~filters.scheduled
|
||||
& filters.command(["sponsorship"], prefixes=["/"])
|
||||
& ~custom_filters.banned
|
||||
& (custom_filters.allowed | custom_filters.admin)
|
||||
)
|
||||
async def cmd_sponsorship(app: Client, msg: Message):
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
if holo_user.application_state()[0] == "fill":
|
||||
await msg.reply_text(locale("finish_application", "message", locale=msg.from_user), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("finish_application", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
return
|
||||
if holo_user.spoiler_state() is True:
|
||||
await msg.reply_text(locale("spoiler_in_progress", "message", locale=holo_user))
|
||||
return
|
||||
await msg.reply_text(locale("sponsorship_apply", "message", locale=msg.from_user), reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(text=str(locale("sponsor_apply", "button", locale=msg.from_user)), callback_data=f"sponsor_apply_{msg.from_user.id}")]]), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("sponsorship_apply", "message", locale=msg.from_user),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(
|
||||
locale("sponsor_apply", "button", locale=msg.from_user)
|
||||
),
|
||||
callback_data=f"sponsor_apply_{msg.from_user.id}",
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
# else:
|
||||
# await msg.reply_text(locale("sponsorship_application_empty", "message"))
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -8,42 +8,67 @@ from modules import custom_filters
|
||||
from bson.objectid import ObjectId
|
||||
from bson.errors import InvalidId
|
||||
|
||||
# Start command ================================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"]) & ~custom_filters.banned)
|
||||
async def cmd_start(app: Client, msg: Message):
|
||||
|
||||
# Start command ================================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(["start"], prefixes=["/"])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def cmd_start(app: Client, msg: Message):
|
||||
user = col_users.find_one({"user": msg.from_user.id})
|
||||
|
||||
if user is None:
|
||||
|
||||
col_users.insert_one({
|
||||
"user": msg.from_user.id,
|
||||
"link": None,
|
||||
"label": "",
|
||||
"tg_name": msg.from_user.first_name,
|
||||
"tg_phone": msg.from_user.phone_number,
|
||||
"tg_locale": msg.from_user.language_code,
|
||||
"tg_username": msg.from_user.username
|
||||
})
|
||||
col_users.insert_one(
|
||||
{
|
||||
"user": msg.from_user.id,
|
||||
"link": None,
|
||||
"label": "",
|
||||
"tg_name": msg.from_user.first_name,
|
||||
"tg_phone": msg.from_user.phone_number,
|
||||
"tg_locale": msg.from_user.language_code,
|
||||
"tg_username": msg.from_user.username,
|
||||
}
|
||||
)
|
||||
|
||||
logWrite(f"User {msg.from_user.id} started bot interaction")
|
||||
await msg.reply_text(locale("start", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("welcome", "keyboard", locale=msg.from_user), resize_keyboard=True))
|
||||
await msg.reply_text(
|
||||
locale("start", "message", locale=msg.from_user),
|
||||
reply_markup=ReplyKeyboardMarkup(
|
||||
locale("welcome", "keyboard", locale=msg.from_user),
|
||||
resize_keyboard=True,
|
||||
),
|
||||
)
|
||||
|
||||
if len(msg.command) > 1:
|
||||
try:
|
||||
spoiler = col_spoilers.find_one( {"_id": ObjectId(msg.command[1])} )
|
||||
spoiler = col_spoilers.find_one({"_id": ObjectId(msg.command[1])})
|
||||
if spoiler["photo"] is not None:
|
||||
await msg.reply_cached_media(spoiler["photo"], caption=spoiler["caption"])
|
||||
await msg.reply_cached_media(
|
||||
spoiler["photo"], caption=spoiler["caption"]
|
||||
)
|
||||
if spoiler["video"] is not None:
|
||||
await msg.reply_cached_media(spoiler["video"], caption=spoiler["caption"])
|
||||
await msg.reply_cached_media(
|
||||
spoiler["video"], caption=spoiler["caption"]
|
||||
)
|
||||
if spoiler["audio"] is not None:
|
||||
await msg.reply_cached_media(spoiler["audio"], caption=spoiler["caption"])
|
||||
await msg.reply_cached_media(
|
||||
spoiler["audio"], caption=spoiler["caption"]
|
||||
)
|
||||
if spoiler["animation"] is not None:
|
||||
await msg.reply_cached_media(spoiler["animation"], caption=spoiler["caption"])
|
||||
await msg.reply_cached_media(
|
||||
spoiler["animation"], caption=spoiler["caption"]
|
||||
)
|
||||
if spoiler["document"] is not None:
|
||||
await msg.reply_cached_media(spoiler["document"], caption=spoiler["caption"])
|
||||
await msg.reply_cached_media(
|
||||
spoiler["document"], caption=spoiler["caption"]
|
||||
)
|
||||
if spoiler["text"] is not None:
|
||||
await msg.reply_text(spoiler["text"])
|
||||
except InvalidId:
|
||||
await msg.reply_text(f"Got an invalid ID {msg.command[1]}")
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,16 +7,41 @@ from modules.utils import configGet, locale
|
||||
from modules.database import col_warnings
|
||||
from modules import custom_filters
|
||||
|
||||
# Warn command =================================================================================================================
|
||||
@app.on_message(custom_filters.enabled_warnings & ~filters.scheduled & filters.command(["warn"], prefixes=["/"]) & custom_filters.admin)
|
||||
async def cmd_warn(app: Client, msg: Message):
|
||||
|
||||
# Warn command =================================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_warnings
|
||||
& ~filters.scheduled
|
||||
& filters.command(["warn"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_warn(app: Client, msg: Message):
|
||||
if msg.chat.id == configGet("users", "groups"):
|
||||
if msg.reply_to_message_id != None:
|
||||
message = " ".join(msg.command[1:]) if len(msg.command) > 1 else ""
|
||||
col_warnings.insert_one({"user": msg.reply_to_message.from_user.id, "admin": msg.from_user.id, "date": datetime.now(), "reason": message})
|
||||
col_warnings.insert_one(
|
||||
{
|
||||
"user": msg.reply_to_message.from_user.id,
|
||||
"admin": msg.from_user.id,
|
||||
"date": datetime.now(),
|
||||
"reason": message,
|
||||
}
|
||||
)
|
||||
if message == "":
|
||||
await msg.reply_text(locale("warned", "message").format(msg.reply_to_message.from_user.first_name, msg.reply_to_message.from_user.id))
|
||||
await msg.reply_text(
|
||||
locale("warned", "message").format(
|
||||
msg.reply_to_message.from_user.first_name,
|
||||
msg.reply_to_message.from_user.id,
|
||||
)
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(locale("warned_reason", "message").format(msg.reply_to_message.from_user.first_name, msg.reply_to_message.from_user.id, message))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("warned_reason", "message").format(
|
||||
msg.reply_to_message.from_user.first_name,
|
||||
msg.reply_to_message.from_user.id,
|
||||
message,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,12 +7,20 @@ from modules.utils import configGet, locale, should_quote
|
||||
from modules.database import col_users, col_warnings
|
||||
from modules import custom_filters
|
||||
|
||||
# Warnings command =============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_warnings & ~filters.scheduled & filters.command(["warnings"], prefixes=["/"]) & custom_filters.admin)
|
||||
async def cmd_warnings(app: Client, msg: Message):
|
||||
|
||||
# Warnings command =============================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_warnings
|
||||
& ~filters.scheduled
|
||||
& filters.command(["warnings"], prefixes=["/"])
|
||||
& custom_filters.admin
|
||||
)
|
||||
async def cmd_warnings(app: Client, msg: Message):
|
||||
if len(msg.command) <= 1:
|
||||
await msg.reply_text(locale("syntax_warnings", "message", locale=msg.from_user), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("syntax_warnings", "message", locale=msg.from_user),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
return
|
||||
|
||||
try:
|
||||
@ -21,7 +29,11 @@ async def cmd_warnings(app: Client, msg: Message):
|
||||
target_name = user_db["tg_name"]
|
||||
except:
|
||||
list_of_users = []
|
||||
async for m in app.get_chat_members(configGet("users", "groups"), filter=ChatMembersFilter.SEARCH, query=msg.command[1]):
|
||||
async for m in app.get_chat_members(
|
||||
configGet("users", "groups"),
|
||||
filter=ChatMembersFilter.SEARCH,
|
||||
query=msg.command[1],
|
||||
):
|
||||
list_of_users.append(m)
|
||||
|
||||
if len(list_of_users) != 0:
|
||||
@ -29,16 +41,37 @@ async def cmd_warnings(app: Client, msg: Message):
|
||||
target_name = target.first_name
|
||||
target_id = str(target.id)
|
||||
else:
|
||||
await msg.reply_text(locale("no_user_warnings", "message", locale=msg.from_user).format(msg.command[1]))
|
||||
await msg.reply_text(
|
||||
locale("no_user_warnings", "message", locale=msg.from_user).format(
|
||||
msg.command[1]
|
||||
)
|
||||
)
|
||||
return
|
||||
|
||||
warns = len(list(col_warnings.find({"user": target_id})))
|
||||
|
||||
if warns == 0:
|
||||
await msg.reply_text(locale("no_warnings", "message", locale=msg.from_user).format(target_name, target_id), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("no_warnings", "message", locale=msg.from_user).format(
|
||||
target_name, target_id
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
else:
|
||||
if warns <= 5:
|
||||
await msg.reply_text(locale("warnings_1", "message", locale=msg.from_user).format(target_name, target_id, warns), quote=should_quote(msg))
|
||||
await msg.reply_text(
|
||||
locale("warnings_1", "message", locale=msg.from_user).format(
|
||||
target_name, target_id, warns
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(locale("warnings_2", "message", locale=msg.from_user).format(target_name, target_id, warns), quote=should_quote(msg))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("warnings_2", "message", locale=msg.from_user).format(
|
||||
target_name, target_id, warns
|
||||
),
|
||||
quote=should_quote(msg),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -12,44 +12,73 @@ from pyrogram.types import Message
|
||||
async def admin_func(_, __, msg: Message):
|
||||
return await isAnAdmin(msg.from_user.id)
|
||||
|
||||
|
||||
async def member_func(_, __, msg: Message):
|
||||
return True if (msg.from_user.id in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))) else False
|
||||
return (
|
||||
True
|
||||
if (
|
||||
msg.from_user.id
|
||||
in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))
|
||||
)
|
||||
else False
|
||||
)
|
||||
|
||||
|
||||
async def allowed_func(_, __, msg: Message):
|
||||
output = False
|
||||
output = True if (col_applications.find_one({"user": msg.from_user.id}) is not None) else False
|
||||
if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (msg.from_user.id not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))):
|
||||
output = (
|
||||
True
|
||||
if (col_applications.find_one({"user": msg.from_user.id}) is not None)
|
||||
else False
|
||||
)
|
||||
if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (
|
||||
msg.from_user.id
|
||||
not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))
|
||||
):
|
||||
output = False
|
||||
return output
|
||||
|
||||
|
||||
async def banned_func(_, __, msg: Message):
|
||||
return True if col_bans.find_one( {"user": msg.from_user.id} ) is not None else False
|
||||
return True if col_bans.find_one({"user": msg.from_user.id}) is not None else False
|
||||
|
||||
|
||||
async def enabled_general_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "general")
|
||||
|
||||
|
||||
async def enabled_applications_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "applications")
|
||||
|
||||
|
||||
async def enabled_sponsorships_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "sponsorships")
|
||||
|
||||
|
||||
async def enabled_warnings_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "warnings")
|
||||
|
||||
|
||||
async def enabled_invites_check_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "invites_check")
|
||||
|
||||
|
||||
async def enabled_dinovoice_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "dinovoice")
|
||||
|
||||
|
||||
async def enabled_spoilers_func(_, __, msg: Message):
|
||||
return configGet("enabled", "features", "spoilers")
|
||||
|
||||
|
||||
async def filling_sponsorship_func(_, __, msg: Message):
|
||||
return True if col_tmp.find_one({"user": msg.from_user.id, "type": "sponsorship"}) is not None else False
|
||||
return (
|
||||
True
|
||||
if col_tmp.find_one({"user": msg.from_user.id, "type": "sponsorship"})
|
||||
is not None
|
||||
else False
|
||||
)
|
||||
|
||||
|
||||
admin = filters.create(admin_func)
|
||||
member = filters.create(member_func)
|
||||
@ -65,4 +94,4 @@ enabled_invites_check = filters.create(enabled_invites_check_func)
|
||||
enabled_dinovoice = filters.create(enabled_dinovoice_func)
|
||||
enabled_spoilers = filters.create(enabled_spoilers_func)
|
||||
|
||||
filling_sponsorship = filters.create(filling_sponsorship_func)
|
||||
filling_sponsorship = filters.create(filling_sponsorship_func)
|
||||
|
@ -9,18 +9,16 @@ with open("config.json", "r", encoding="utf-8") as f:
|
||||
f.close()
|
||||
|
||||
if db_config["user"] is not None and db_config["password"] is not None:
|
||||
con_string = 'mongodb://{0}:{1}@{2}:{3}/{4}'.format(
|
||||
con_string = "mongodb://{0}:{1}@{2}:{3}/{4}".format(
|
||||
db_config["user"],
|
||||
db_config["password"],
|
||||
db_config["host"],
|
||||
db_config["port"],
|
||||
db_config["name"]
|
||||
db_config["name"],
|
||||
)
|
||||
else:
|
||||
con_string = 'mongodb://{0}:{1}/{2}'.format(
|
||||
db_config["host"],
|
||||
db_config["port"],
|
||||
db_config["name"]
|
||||
con_string = "mongodb://{0}:{1}/{2}".format(
|
||||
db_config["host"], db_config["port"], db_config["name"]
|
||||
)
|
||||
|
||||
db_client = MongoClient(con_string)
|
||||
@ -28,7 +26,18 @@ db = db_client.get_database(name=db_config["name"])
|
||||
|
||||
collections = db.list_collection_names()
|
||||
|
||||
for collection in ["tmp", "bans", "users", "context", "youtube", "spoilers", "messages", "warnings", "applications", "sponsorships"]:
|
||||
for collection in [
|
||||
"tmp",
|
||||
"bans",
|
||||
"users",
|
||||
"context",
|
||||
"youtube",
|
||||
"spoilers",
|
||||
"messages",
|
||||
"warnings",
|
||||
"applications",
|
||||
"sponsorships",
|
||||
]:
|
||||
if not collection in collections:
|
||||
db.create_collection(collection)
|
||||
|
||||
@ -43,4 +52,4 @@ col_warnings = db.get_collection("warnings")
|
||||
col_applications = db.get_collection("applications")
|
||||
col_sponsorships = db.get_collection("sponsorships")
|
||||
|
||||
col_applications.create_index([("application.3.location", GEOSPHERE)])
|
||||
col_applications.create_index([("application.3.location", GEOSPHERE)])
|
||||
|
@ -3,7 +3,13 @@ from dateutil.relativedelta import relativedelta
|
||||
from datetime import datetime
|
||||
from app import app
|
||||
from pyrogram import filters
|
||||
from pyrogram.types import ReplyKeyboardRemove, InlineKeyboardMarkup, InlineKeyboardButton, ForceReply, Message
|
||||
from pyrogram.types import (
|
||||
ReplyKeyboardRemove,
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
ForceReply,
|
||||
Message,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from pyrogram.enums.parse_mode import ParseMode
|
||||
from classes.holo_user import HoloUser
|
||||
@ -16,16 +22,30 @@ from modules import custom_filters
|
||||
confirmation_1 = []
|
||||
for pattern in all_locales("confirm", "keyboard"):
|
||||
confirmation_1.append(pattern[0][0])
|
||||
@app.on_message((custom_filters.enabled_applications | custom_filters.enabled_sponsorships) & ~filters.scheduled & filters.private & filters.command(confirmation_1, prefixes=[""]) & ~custom_filters.banned)
|
||||
async def confirm_yes(app: Client, msg: Message, kind: Literal["application", "sponsorship", "unknown"] = "unknown"):
|
||||
|
||||
|
||||
@app.on_message(
|
||||
(custom_filters.enabled_applications | custom_filters.enabled_sponsorships)
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(confirmation_1, prefixes=[""])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def confirm_yes(
|
||||
app: Client,
|
||||
msg: Message,
|
||||
kind: Literal["application", "sponsorship", "unknown"] = "unknown",
|
||||
):
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
|
||||
if configGet("enabled", "features", "applications") is True:
|
||||
|
||||
if (kind == "application") or ((holo_user.application_state()[0] == "fill") and (holo_user.application_state()[1] is True)):
|
||||
|
||||
tmp_application = col_tmp.find_one({"user": holo_user.id, "type": "application"})
|
||||
if (kind == "application") or (
|
||||
(holo_user.application_state()[0] == "fill")
|
||||
and (holo_user.application_state()[1] is True)
|
||||
):
|
||||
tmp_application = col_tmp.find_one(
|
||||
{"user": holo_user.id, "type": "application"}
|
||||
)
|
||||
|
||||
if tmp_application is None:
|
||||
logWrite(f"Application of {holo_user.id} is nowhere to be found.")
|
||||
@ -34,73 +54,120 @@ async def confirm_yes(app: Client, msg: Message, kind: Literal["application", "s
|
||||
if tmp_application["sent"] is True:
|
||||
return
|
||||
|
||||
await msg.reply_text(locale("application_sent", "message"), reply_markup=ReplyKeyboardRemove())
|
||||
await msg.reply_text(
|
||||
locale("application_sent", "message"),
|
||||
reply_markup=ReplyKeyboardRemove(),
|
||||
)
|
||||
|
||||
application_content = []
|
||||
i = 1
|
||||
|
||||
for question in tmp_application['application']:
|
||||
|
||||
for question in tmp_application["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), tmp_application['application']['2'])
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(
|
||||
datetime.now(), tmp_application["application"]["2"]
|
||||
)
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if tmp_application['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application']['3']['name']}")
|
||||
if tmp_application["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application']['3']['name']}"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application']['3']['name']} ({tmp_application['application']['3']['adminName1']}, {tmp_application['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application']['3']['name']} ({tmp_application['application']['3']['adminName1']}, {tmp_application['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application'][question]}")
|
||||
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {tmp_application['application'][question]}"
|
||||
)
|
||||
|
||||
i += 1
|
||||
|
||||
if tmp_application["reapply"] is True and col_applications.find_one({"user": holo_user.id}) is not None:
|
||||
if (
|
||||
tmp_application["reapply"] is True
|
||||
and col_applications.find_one({"user": holo_user.id}) is not None
|
||||
):
|
||||
await app.send_message(
|
||||
chat_id=configGet("admin", "groups"),
|
||||
text=(locale("reapply_got", "message")).format(str(holo_user.id),msg.from_user.first_name, msg.from_user.username, "\n".join(application_content)),
|
||||
text=(locale("reapply_got", "message")).format(
|
||||
str(holo_user.id),
|
||||
msg.from_user.first_name,
|
||||
msg.from_user.username,
|
||||
"\n".join(application_content),
|
||||
),
|
||||
parse_mode=ParseMode.MARKDOWN,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("reapply_yes", "button")), callback_data=f"reapply_yes_{holo_user.id}")
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("reapply_yes", "button")),
|
||||
callback_data=f"reapply_yes_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("reapply_no", "button")), callback_data=f"reapply_no_{holo_user.id}")
|
||||
]
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("reapply_no", "button")),
|
||||
callback_data=f"reapply_no_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
]
|
||||
)
|
||||
),
|
||||
)
|
||||
else:
|
||||
await app.send_message(
|
||||
chat_id=configGet("admin", "groups"),
|
||||
text=(locale("application_got", "message")).format(str(holo_user.id), msg.from_user.first_name, msg.from_user.username, "\n".join(application_content)),
|
||||
text=(locale("application_got", "message")).format(
|
||||
str(holo_user.id),
|
||||
msg.from_user.first_name,
|
||||
msg.from_user.username,
|
||||
"\n".join(application_content),
|
||||
),
|
||||
parse_mode=ParseMode.MARKDOWN,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sub_yes", "button")), callback_data=f"sub_yes_{holo_user.id}")
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sub_yes", "button")),
|
||||
callback_data=f"sub_yes_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sub_no", "button")), callback_data=f"sub_no_{holo_user.id}")
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sub_no", "button")),
|
||||
callback_data=f"sub_no_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sub_russian", "button")), callback_data=f"sub_russian_{holo_user.id}")
|
||||
]
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sub_russian", "button")),
|
||||
callback_data=f"sub_russian_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
]
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
logWrite(f"User {holo_user.id} sent his application and it will now be reviewed")
|
||||
logWrite(
|
||||
f"User {holo_user.id} sent his application and it will now be reviewed"
|
||||
)
|
||||
|
||||
col_tmp.update_one({"user": holo_user.id, "type": "application"}, {"$set": {"sent": True}})
|
||||
col_tmp.update_one(
|
||||
{"user": holo_user.id, "type": "application"}, {"$set": {"sent": True}}
|
||||
)
|
||||
|
||||
return
|
||||
|
||||
if configGet("enabled", "features", "sponsorships") is True:
|
||||
|
||||
if (kind == "sponsorship") or ((holo_user.sponsorship_state()[0] == "fill") and (holo_user.sponsorship_state()[1] is True)):
|
||||
|
||||
tmp_sponsorship = col_tmp.find_one({"user": holo_user.id, "type": "sponsorship"})
|
||||
if (kind == "sponsorship") or (
|
||||
(holo_user.sponsorship_state()[0] == "fill")
|
||||
and (holo_user.sponsorship_state()[1] is True)
|
||||
):
|
||||
tmp_sponsorship = col_tmp.find_one(
|
||||
{"user": holo_user.id, "type": "sponsorship"}
|
||||
)
|
||||
|
||||
if tmp_sponsorship is None:
|
||||
logWrite(f"Sponsorship of {holo_user.id} is nowhere to be found.")
|
||||
@ -109,58 +176,116 @@ async def confirm_yes(app: Client, msg: Message, kind: Literal["application", "s
|
||||
if tmp_sponsorship["sent"] is True:
|
||||
return
|
||||
|
||||
await msg.reply_text(locale("sponsorship_sent", "message"), reply_markup=ReplyKeyboardRemove())
|
||||
await msg.reply_text(
|
||||
locale("sponsorship_sent", "message"),
|
||||
reply_markup=ReplyKeyboardRemove(),
|
||||
)
|
||||
|
||||
sponsorship_content = []
|
||||
|
||||
for question in tmp_sponsorship['sponsorship']:
|
||||
|
||||
for question in tmp_sponsorship["sponsorship"]:
|
||||
if question == "expires":
|
||||
sponsorship_content.append(f"{locale(f'question_{question}', 'message', 'sponsor_titles')} {tmp_sponsorship['sponsorship'][question].strftime('%d.%m.%Y')}")
|
||||
sponsorship_content.append(
|
||||
f"{locale(f'question_{question}', 'message', 'sponsor_titles')} {tmp_sponsorship['sponsorship'][question].strftime('%d.%m.%Y')}"
|
||||
)
|
||||
elif question == "proof":
|
||||
continue
|
||||
else:
|
||||
sponsorship_content.append(f"{locale(f'question_{question}', 'message', 'sponsor_titles')} {tmp_sponsorship['sponsorship'][question]}")
|
||||
|
||||
await app.send_cached_media(configGet("admin", "groups"), tmp_sponsorship["sponsorship"]["proof"], caption=(locale("sponsor_got", "message")).format(str(holo_user.id), msg.from_user.first_name, msg.from_user.username, "\n".join(sponsorship_content)), parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(
|
||||
sponsorship_content.append(
|
||||
f"{locale(f'question_{question}', 'message', 'sponsor_titles')} {tmp_sponsorship['sponsorship'][question]}"
|
||||
)
|
||||
|
||||
await app.send_cached_media(
|
||||
configGet("admin", "groups"),
|
||||
tmp_sponsorship["sponsorship"]["proof"],
|
||||
caption=(locale("sponsor_got", "message")).format(
|
||||
str(holo_user.id),
|
||||
msg.from_user.first_name,
|
||||
msg.from_user.username,
|
||||
"\n".join(sponsorship_content),
|
||||
),
|
||||
parse_mode=ParseMode.MARKDOWN,
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sponsor_yes", "button")), callback_data=f"sponsor_yes_{holo_user.id}")
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sponsor_yes", "button")),
|
||||
callback_data=f"sponsor_yes_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sponsor_no", "button")), callback_data=f"sponsor_no_{holo_user.id}")
|
||||
]
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sponsor_no", "button")),
|
||||
callback_data=f"sponsor_no_{holo_user.id}",
|
||||
)
|
||||
],
|
||||
]
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
# remove(f"tmp{sep}{filename}.jpg")
|
||||
|
||||
logWrite(f"User {holo_user.id} sent his sponsorship application and it will now be reviewed")
|
||||
logWrite(
|
||||
f"User {holo_user.id} sent his sponsorship application and it will now be reviewed"
|
||||
)
|
||||
|
||||
col_tmp.update_one({"user": holo_user.id, "type": "sponsorship"}, {"$set": {"sent": True}})
|
||||
col_tmp.update_one(
|
||||
{"user": holo_user.id, "type": "sponsorship"}, {"$set": {"sent": True}}
|
||||
)
|
||||
|
||||
return
|
||||
|
||||
|
||||
confirmation_2 = []
|
||||
for pattern in all_locales("confirm", "keyboard"):
|
||||
confirmation_2.append(pattern[1][0])
|
||||
@app.on_message((custom_filters.enabled_applications | custom_filters.enabled_sponsorships) & ~filters.scheduled & filters.private & filters.command(confirmation_2, prefixes=[""]) & ~custom_filters.banned)
|
||||
async def confirm_no(app: Client, msg: Message, kind: Literal["application", "sponsorship", "unknown"] = "unknown"):
|
||||
|
||||
|
||||
@app.on_message(
|
||||
(custom_filters.enabled_applications | custom_filters.enabled_sponsorships)
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(confirmation_2, prefixes=[""])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def confirm_no(
|
||||
app: Client,
|
||||
msg: Message,
|
||||
kind: Literal["application", "sponsorship", "unknown"] = "unknown",
|
||||
):
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
|
||||
if configGet("enabled", "features", "applications") is True:
|
||||
if (kind == "application") or ((holo_user.application_state()[0] == "fill") and (holo_user.application_state()[1] is True)):
|
||||
if (kind == "application") or (
|
||||
(holo_user.application_state()[0] == "fill")
|
||||
and (holo_user.application_state()[1] is True)
|
||||
):
|
||||
holo_user.application_restart()
|
||||
await welcome_pass(app, msg, once_again=True)
|
||||
logWrite(f"User {msg.from_user.id} restarted the application due to typo in it")
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} restarted the application due to typo in it"
|
||||
)
|
||||
return
|
||||
|
||||
if configGet("enabled", "features", "sponsorships") is True:
|
||||
if (kind == "sponsorship") or ((holo_user.sponsorship_state()[0] == "fill") and (holo_user.sponsorship_state()[1] is True)):
|
||||
if (kind == "sponsorship") or (
|
||||
(holo_user.sponsorship_state()[0] == "fill")
|
||||
and (holo_user.sponsorship_state()[1] is True)
|
||||
):
|
||||
holo_user.sponsorship_restart()
|
||||
await app.send_message(holo_user.id, locale(f"sponsor1", "message", locale=holo_user.locale), reply_markup=ForceReply(placeholder=str(locale(f"sponsor1", "force_reply", locale=holo_user.locale))))
|
||||
logWrite(f"User {msg.from_user.id} restarted the sponsorship application due to typo in it")
|
||||
await app.send_message(
|
||||
holo_user.id,
|
||||
locale(f"sponsor1", "message", locale=holo_user.locale),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=str(
|
||||
locale(f"sponsor1", "force_reply", locale=holo_user.locale)
|
||||
)
|
||||
),
|
||||
)
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} restarted the sponsorship application due to typo in it"
|
||||
)
|
||||
return
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -9,46 +9,79 @@ from modules.database import col_applications
|
||||
from classes.holo_user import HoloUser
|
||||
from modules import custom_filters
|
||||
|
||||
# Contact getting ==============================================================================================================
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.contact & filters.private & (custom_filters.allowed | custom_filters.admin) & ~custom_filters.banned)
|
||||
async def get_contact(app: Client, msg: Message):
|
||||
|
||||
# Contact getting ==============================================================================================================
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.contact
|
||||
& filters.private
|
||||
& (custom_filters.allowed | custom_filters.admin)
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def get_contact(app: Client, msg: Message):
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
|
||||
if msg.contact.user_id != None:
|
||||
|
||||
application = col_applications.find_one({"user": msg.contact.user_id})
|
||||
|
||||
if application is None:
|
||||
logWrite(f"User {holo_user.id} requested application of {msg.contact.user_id} but user does not exists")
|
||||
await msg.reply_text(locale("contact_invalid", "message", locale=holo_user.locale))
|
||||
logWrite(
|
||||
f"User {holo_user.id} requested application of {msg.contact.user_id} but user does not exists"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("contact_invalid", "message", locale=holo_user.locale)
|
||||
)
|
||||
return
|
||||
|
||||
application_content = []
|
||||
i = 1
|
||||
|
||||
for question in application['application']:
|
||||
|
||||
for question in application["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), application['application']['2'])
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(datetime.now(), application["application"]["2"])
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if application['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application']['3']['name']}")
|
||||
if application["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application']['3']['name']}"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application'][question]}")
|
||||
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=holo_user.locale)} {application['application'][question]}"
|
||||
)
|
||||
|
||||
i += 1
|
||||
|
||||
application_status = locale("application_status_accepted", "message", locale=holo_user.locale).format((await app.get_users(application["admin"])).first_name, application["date"].strftime("%d.%m.%Y, %H:%M"))
|
||||
application_status = locale(
|
||||
"application_status_accepted", "message", locale=holo_user.locale
|
||||
).format(
|
||||
(await app.get_users(application["admin"])).first_name,
|
||||
application["date"].strftime("%d.%m.%Y, %H:%M"),
|
||||
)
|
||||
|
||||
logWrite(f"User {holo_user.id} requested application of {msg.contact.user_id}")
|
||||
await msg.reply_text(locale("contact", "message", locale=holo_user.locale).format(str(msg.contact.user_id), "\n".join(application_content), application_status))
|
||||
|
||||
await msg.reply_text(
|
||||
locale("contact", "message", locale=holo_user.locale).format(
|
||||
str(msg.contact.user_id),
|
||||
"\n".join(application_content),
|
||||
application_status,
|
||||
)
|
||||
)
|
||||
|
||||
else:
|
||||
logWrite(f"User {holo_user.id} requested application of someone but user is not telegram user")
|
||||
await msg.reply_text(locale("contact_not_member", "message", locale=holo_user.locale))
|
||||
# ==============================================================================================================================
|
||||
logWrite(
|
||||
f"User {holo_user.id} requested application of someone but user is not telegram user"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("contact_not_member", "message", locale=holo_user.locale)
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -3,35 +3,55 @@ from app import app, isAnAdmin
|
||||
import asyncio
|
||||
from ftfy import fix_text
|
||||
from pyrogram import filters
|
||||
from pyrogram.types import Message, ForceReply, InlineKeyboardMarkup, InlineKeyboardButton, ReplyKeyboardRemove
|
||||
from pyrogram.types import (
|
||||
Message,
|
||||
ForceReply,
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
ReplyKeyboardRemove,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from classes.holo_user import HoloUser
|
||||
from modules.utils import configGet, logWrite, locale, all_locales
|
||||
from modules.database import col_messages, col_spoilers
|
||||
from modules import custom_filters
|
||||
|
||||
|
||||
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})
|
||||
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
|
||||
|
||||
|
||||
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})
|
||||
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
|
||||
|
||||
|
||||
# Any other input ==============================================================================================================
|
||||
@app.on_message(~ filters.scheduled & (filters.private | filters.chat(configGet("admin", "groups"))) & ~custom_filters.banned)
|
||||
@app.on_message(
|
||||
~filters.scheduled
|
||||
& (filters.private | filters.chat(configGet("admin", "groups")))
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def any_stage(app: Client, msg: Message):
|
||||
|
||||
if msg.via_bot is None:
|
||||
|
||||
holo_user = HoloUser(msg.from_user)
|
||||
|
||||
if (msg.reply_to_message is not None) and (await message_involved(msg)):
|
||||
|
||||
context = await message_context(msg)
|
||||
context_message = await app.get_messages(context[0], context[1])
|
||||
|
||||
@ -51,7 +71,7 @@ async def any_stage(app: Client, msg: Message):
|
||||
animation=msg.animation,
|
||||
voice=msg.voice,
|
||||
adm_origin=await isAnAdmin(context_message.from_user.id),
|
||||
adm_context=await isAnAdmin(msg.from_user.id)
|
||||
adm_context=await isAnAdmin(msg.from_user.id),
|
||||
)
|
||||
|
||||
return
|
||||
@ -60,29 +80,30 @@ async def any_stage(app: Client, msg: Message):
|
||||
return
|
||||
|
||||
if msg.text is not None:
|
||||
|
||||
if configGet("enabled", "features", "applications") is True:
|
||||
await holo_user.application_next(str(msg.text), msg=msg)
|
||||
|
||||
|
||||
if configGet("enabled", "features", "sponsorships") is True:
|
||||
|
||||
await holo_user.sponsorship_next(str(msg.text), msg)
|
||||
|
||||
if msg.photo is not None:
|
||||
await holo_user.sponsorship_next(str(msg.text), msg=msg, photo=msg.photo)
|
||||
|
||||
if holo_user.application_state()[0] != "fill" and holo_user.sponsorship_state()[0] != "fill":
|
||||
|
||||
if (
|
||||
holo_user.application_state()[0] != "fill"
|
||||
and holo_user.sponsorship_state()[0] != "fill"
|
||||
):
|
||||
if configGet("enabled", "features", "spoilers") is False:
|
||||
return
|
||||
|
||||
spoiler = col_spoilers.find_one( {"user": msg.from_user.id, "completed": False} )
|
||||
spoiler = col_spoilers.find_one(
|
||||
{"user": msg.from_user.id, "completed": False}
|
||||
)
|
||||
|
||||
if spoiler is None:
|
||||
return
|
||||
|
||||
if spoiler["category"] is None:
|
||||
|
||||
found = False
|
||||
|
||||
# Find category in all locales
|
||||
@ -93,86 +114,226 @@ async def any_stage(app: Client, msg: Message):
|
||||
category = key
|
||||
|
||||
if found is False:
|
||||
await msg.reply_text(locale("spoiler_incorrect_category", "message", locale=msg.from_user))
|
||||
await msg.reply_text(
|
||||
locale(
|
||||
"spoiler_incorrect_category",
|
||||
"message",
|
||||
locale=msg.from_user,
|
||||
)
|
||||
)
|
||||
return
|
||||
|
||||
col_spoilers.find_one_and_update( {"_id": spoiler["_id"]}, {"$set": {"category": category}} )
|
||||
await msg.reply_text(locale("spoiler_send_description", "message", locale=msg.from_user), reply_markup=ForceReply(placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
|
||||
col_spoilers.find_one_and_update(
|
||||
{"_id": spoiler["_id"]}, {"$set": {"category": category}}
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("spoiler_send_description", "message", locale=msg.from_user),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=locale(
|
||||
"spoiler_description", "force_reply", locale=msg.from_user
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
|
||||
if spoiler["description"] is None and (spoiler["photo"] is None and spoiler["video"] is None and spoiler["audio"] is None and spoiler["animation"] is None and spoiler["text"] is None):
|
||||
|
||||
if spoiler["description"] is None and (
|
||||
spoiler["photo"] is None
|
||||
and spoiler["video"] is None
|
||||
and spoiler["audio"] is None
|
||||
and spoiler["animation"] is None
|
||||
and spoiler["text"] is None
|
||||
):
|
||||
# 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 str(msg.text) != "-":
|
||||
msg.text = fix_text(str(msg.text))
|
||||
if len(str(msg.text)) > 1024:
|
||||
await msg.reply_text(locale("spoiler_description_too_long", "message", locale=msg.from_user), reply_markup=ForceReply(placeholder=locale("spoiler_description", "force_reply", locale=msg.from_user)))
|
||||
await msg.reply_text(
|
||||
locale(
|
||||
"spoiler_description_too_long",
|
||||
"message",
|
||||
locale=msg.from_user,
|
||||
),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=locale(
|
||||
"spoiler_description",
|
||||
"force_reply",
|
||||
locale=msg.from_user,
|
||||
)
|
||||
),
|
||||
)
|
||||
return
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"description": msg.text}} )
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{"$set": {"description": msg.text}},
|
||||
)
|
||||
else:
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"description": ""}} )
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{"$set": {"description": ""}},
|
||||
)
|
||||
|
||||
logWrite(f"Adding description '{str(msg.text)}' to {msg.from_user.id}'s spoiler")
|
||||
await msg.reply_text(locale("spoiler_using_description", "message", locale=msg.from_user).format(msg.text), reply_markup=ForceReply(placeholder=locale("spoiler_content", "force_reply", locale=msg.from_user)))
|
||||
logWrite(
|
||||
f"Adding description '{str(msg.text)}' to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale(
|
||||
"spoiler_using_description", "message", locale=msg.from_user
|
||||
).format(msg.text),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=locale(
|
||||
"spoiler_content", "force_reply", locale=msg.from_user
|
||||
)
|
||||
),
|
||||
)
|
||||
|
||||
return
|
||||
|
||||
ready = False
|
||||
|
||||
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, "caption": msg.caption, "completed": True}} )
|
||||
logWrite(f"Adding photo with id {msg.photo.file_id} to {msg.from_user.id}'s spoiler")
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{
|
||||
"$set": {
|
||||
"photo": msg.photo.file_id,
|
||||
"caption": msg.caption,
|
||||
"completed": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
logWrite(
|
||||
f"Adding photo with id {msg.photo.file_id} to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
ready = True
|
||||
|
||||
if msg.video is not None:
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"video": msg.video.file_id, "caption": msg.caption, "completed": True}} )
|
||||
logWrite(f"Adding audio with id {msg.video.file_id} to {msg.from_user.id}'s spoiler")
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{
|
||||
"$set": {
|
||||
"video": msg.video.file_id,
|
||||
"caption": msg.caption,
|
||||
"completed": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
logWrite(
|
||||
f"Adding audio with id {msg.video.file_id} to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
ready = True
|
||||
|
||||
if msg.audio is not None:
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"audio": msg.audio.file_id, "caption": msg.caption, "completed": True}} )
|
||||
logWrite(f"Adding video with id {msg.audio.file_id} to {msg.from_user.id}'s spoiler")
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{
|
||||
"$set": {
|
||||
"audio": msg.audio.file_id,
|
||||
"caption": msg.caption,
|
||||
"completed": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
logWrite(
|
||||
f"Adding video with id {msg.audio.file_id} to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
ready = True
|
||||
|
||||
if msg.animation is not None:
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"animation": msg.animation.file_id, "caption": msg.caption, "completed": True}} )
|
||||
logWrite(f"Adding animation with id {msg.animation.file_id} to {msg.from_user.id}'s spoiler")
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{
|
||||
"$set": {
|
||||
"animation": msg.animation.file_id,
|
||||
"caption": msg.caption,
|
||||
"completed": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
logWrite(
|
||||
f"Adding animation with id {msg.animation.file_id} to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
ready = True
|
||||
|
||||
if msg.document is not None:
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"document": msg.document.file_id, "caption": msg.caption, "completed": True}} )
|
||||
logWrite(f"Adding document with id {msg.document.file_id} to {msg.from_user.id}'s spoiler")
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{
|
||||
"$set": {
|
||||
"document": msg.document.file_id,
|
||||
"caption": msg.caption,
|
||||
"completed": True,
|
||||
}
|
||||
},
|
||||
)
|
||||
logWrite(
|
||||
f"Adding document with id {msg.document.file_id} to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
ready = True
|
||||
|
||||
if spoiler["photo"] is None and spoiler["video"] is None and spoiler["audio"] is None and spoiler["animation"] is None and spoiler["document"] is None and spoiler["text"] is None:
|
||||
if (
|
||||
spoiler["photo"] is None
|
||||
and spoiler["video"] is None
|
||||
and spoiler["audio"] is None
|
||||
and spoiler["animation"] is None
|
||||
and spoiler["document"] is None
|
||||
and spoiler["text"] is None
|
||||
):
|
||||
if msg.text is not None:
|
||||
col_spoilers.find_one_and_update( {"user": msg.from_user.id, "completed": False}, {"$set": {"text": str(msg.text), "completed": True}} )
|
||||
logWrite(f"Adding text '{str(msg.text)}' to {msg.from_user.id}'s spoiler")
|
||||
col_spoilers.find_one_and_update(
|
||||
{"user": msg.from_user.id, "completed": False},
|
||||
{"$set": {"text": str(msg.text), "completed": True}},
|
||||
)
|
||||
logWrite(
|
||||
f"Adding text '{str(msg.text)}' to {msg.from_user.id}'s spoiler"
|
||||
)
|
||||
ready = True
|
||||
|
||||
if ready is True:
|
||||
await msg.reply_text(locale("spoiler_ready", "message", locale=msg.from_user), reply_markup=ReplyKeyboardRemove())
|
||||
await msg.reply_text(
|
||||
locale("spoiler_ready", "message", locale=msg.from_user),
|
||||
reply_markup=ReplyKeyboardRemove(),
|
||||
)
|
||||
if configGet("allow_external", "features", "spoilers") is True:
|
||||
await msg.reply_text(
|
||||
locale("spoiler_send", "message", locale=msg.from_user),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(locale("spoiler_preview", "button", locale=msg.from_user), callback_data=f"sid_{spoiler['_id'].__str__()}")
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"spoiler_preview",
|
||||
"button",
|
||||
locale=msg.from_user,
|
||||
),
|
||||
callback_data=f"sid_{spoiler['_id'].__str__()}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(locale("spoiler_send_chat", "button", locale=msg.from_user), callback_data=f"shc_{spoiler['_id'].__str__()}")
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"spoiler_send_chat",
|
||||
"button",
|
||||
locale=msg.from_user,
|
||||
),
|
||||
callback_data=f"shc_{spoiler['_id'].__str__()}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(locale("spoiler_send_other", "button", locale=msg.from_user), switch_inline_query=f"spoiler:{spoiler['_id'].__str__()}")
|
||||
]
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"spoiler_send_other",
|
||||
"button",
|
||||
locale=msg.from_user,
|
||||
),
|
||||
switch_inline_query=f"spoiler:{spoiler['_id'].__str__()}",
|
||||
)
|
||||
],
|
||||
]
|
||||
)
|
||||
),
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(
|
||||
@ -180,33 +341,65 @@ async def any_stage(app: Client, msg: Message):
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(locale("spoiler_preview", "button", locale=msg.from_user), callback_data=f"sid_{spoiler['_id'].__str__()}")
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"spoiler_preview",
|
||||
"button",
|
||||
locale=msg.from_user,
|
||||
),
|
||||
callback_data=f"sid_{spoiler['_id'].__str__()}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(locale("spoiler_send_chat", "button", locale=msg.from_user), callback_data=f"shc_{spoiler['_id'].__str__()}")
|
||||
]
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"spoiler_send_chat",
|
||||
"button",
|
||||
locale=msg.from_user,
|
||||
),
|
||||
callback_data=f"shc_{spoiler['_id'].__str__()}",
|
||||
)
|
||||
],
|
||||
]
|
||||
)
|
||||
),
|
||||
)
|
||||
else:
|
||||
await msg.reply_text(locale("spoiler_incorrect_content", "message", locale=msg.from_user))
|
||||
await msg.reply_text(
|
||||
locale("spoiler_incorrect_content", "message", locale=msg.from_user)
|
||||
)
|
||||
|
||||
|
||||
@app.on_message(~ filters.scheduled & filters.group)
|
||||
@app.on_message(~filters.scheduled & filters.group)
|
||||
async def message_in_group(app: Client, msg: Message):
|
||||
if (msg.chat is not None) and (msg.via_bot is not None):
|
||||
if (msg.via_bot.id == (await app.get_me()).id) and (msg.chat.id == configGet("users", "groups")):
|
||||
if str(msg.text).startswith(locale("spoiler_described", "message").split()[0]) or str(msg.text).startswith(locale("spoiler_empty", "message").split()[0]):
|
||||
if (msg.via_bot.id == (await app.get_me()).id) and (
|
||||
msg.chat.id == configGet("users", "groups")
|
||||
):
|
||||
if str(msg.text).startswith(
|
||||
locale("spoiler_described", "message").split()[0]
|
||||
) or str(msg.text).startswith(
|
||||
locale("spoiler_empty", "message").split()[0]
|
||||
):
|
||||
logWrite(f"User {msg.from_user.id} sent spoiler to user's group")
|
||||
try:
|
||||
logWrite("Forwarding spoiler to admin's group")
|
||||
await msg.copy(configGet("admin", "groups"), disable_notification=True)
|
||||
await msg.copy(
|
||||
configGet("admin", "groups"), disable_notification=True
|
||||
)
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not forward spoiler to admin's group due to '{exp}': {print_exc()}")
|
||||
logWrite(
|
||||
f"Could not forward spoiler to admin's group due to '{exp}': {print_exc()}"
|
||||
)
|
||||
return
|
||||
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)
|
||||
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")
|
||||
# ==============================================================================================================================
|
||||
logWrite(
|
||||
f"Removed application requested by {msg.from_user.id} in destination group"
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -1,6 +1,11 @@
|
||||
from datetime import datetime
|
||||
from app import app, isAnAdmin
|
||||
from pyrogram.types import ChatPermissions, InlineKeyboardMarkup, InlineKeyboardButton, ChatMemberUpdated
|
||||
from pyrogram.types import (
|
||||
ChatPermissions,
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
ChatMemberUpdated,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from modules.utils import configGet, locale
|
||||
from modules import custom_filters
|
||||
@ -9,45 +14,64 @@ from modules.database import col_applications
|
||||
from classes.holo_user import HoloUser
|
||||
from dateutil.relativedelta import relativedelta
|
||||
|
||||
|
||||
# Filter users on join =========================================================================================================
|
||||
@app.on_chat_member_updated(custom_filters.enabled_invites_check, group=configGet("users", "groups"))
|
||||
#@app.on_message(filters.new_chat_members, group=configGet("users", "groups"))
|
||||
@app.on_chat_member_updated(
|
||||
custom_filters.enabled_invites_check, group=configGet("users", "groups")
|
||||
)
|
||||
# @app.on_message(filters.new_chat_members, group=configGet("users", "groups"))
|
||||
async def filter_join(app: Client, member: ChatMemberUpdated):
|
||||
|
||||
if member.invite_link != None:
|
||||
|
||||
holo_user = HoloUser(member.from_user)
|
||||
|
||||
if (holo_user.link is not None) and (holo_user.link == member.invite_link.invite_link):
|
||||
|
||||
logWrite(f"User {holo_user.id} joined destination group with correct link {holo_user.link}")
|
||||
if (holo_user.link is not None) and (
|
||||
holo_user.link == member.invite_link.invite_link
|
||||
):
|
||||
logWrite(
|
||||
f"User {holo_user.id} joined destination group with correct link {holo_user.link}"
|
||||
)
|
||||
|
||||
application = col_applications.find_one({"user": holo_user.id})
|
||||
application_content = []
|
||||
i = 1
|
||||
|
||||
for question in application['application']:
|
||||
|
||||
for question in application["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), application['application']['2'])
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(datetime.now(), application["application"]["2"])
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if application['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']}")
|
||||
if application["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']}"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application'][question]}")
|
||||
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application'][question]}"
|
||||
)
|
||||
|
||||
i += 1
|
||||
|
||||
await app.send_message(configGet("users", "groups"), locale("joined_application", "message").format(member.from_user.first_name, member.from_user.username, "\n".join(application_content)))
|
||||
await app.send_message(
|
||||
configGet("users", "groups"),
|
||||
locale("joined_application", "message").format(
|
||||
member.from_user.first_name,
|
||||
member.from_user.username,
|
||||
"\n".join(application_content),
|
||||
),
|
||||
)
|
||||
|
||||
return
|
||||
|
||||
|
||||
if await isAnAdmin(member.invite_link.creator.id):
|
||||
|
||||
logWrite(f"User {holo_user.id} joined destination group with link {holo_user.link} of an admin {member.invite_link.creator.id}")
|
||||
logWrite(
|
||||
f"User {holo_user.id} joined destination group with link {holo_user.link} of an admin {member.invite_link.creator.id}"
|
||||
)
|
||||
|
||||
application = col_applications.find_one({"user": holo_user.id})
|
||||
|
||||
@ -57,42 +81,75 @@ async def filter_join(app: Client, member: ChatMemberUpdated):
|
||||
application_content = []
|
||||
i = 1
|
||||
|
||||
for question in application['application']:
|
||||
|
||||
for question in application["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), application['application']['2'])
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(datetime.now(), application["application"]["2"])
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if application['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']}")
|
||||
if application["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']}"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles')} {application['application'][question]}")
|
||||
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles')} {application['application'][question]}"
|
||||
)
|
||||
|
||||
i += 1
|
||||
|
||||
await app.send_message(configGet("users", "groups"), locale("joined_application", "message").format(member.from_user.first_name, member.from_user.username, "\n".join(application_content)))
|
||||
await app.send_message(
|
||||
configGet("users", "groups"),
|
||||
locale("joined_application", "message").format(
|
||||
member.from_user.first_name,
|
||||
member.from_user.username,
|
||||
"\n".join(application_content),
|
||||
),
|
||||
)
|
||||
|
||||
return
|
||||
|
||||
logWrite(f"User {holo_user.id} joined destination group with stolen/unapproved link {holo_user.link}")
|
||||
logWrite(
|
||||
f"User {holo_user.id} joined destination group with stolen/unapproved link {holo_user.link}"
|
||||
)
|
||||
|
||||
await app.send_message(configGet("admin", "groups"), locale("joined_false_link", "message").format(member.from_user.first_name, member.from_user.id), reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
await app.send_message(
|
||||
configGet("admin", "groups"),
|
||||
locale("joined_false_link", "message").format(
|
||||
member.from_user.first_name, member.from_user.id
|
||||
),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sus_allow", "button")), callback_data=f"sus_allow_{member.from_user.id}")
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(text=str(locale("sus_reject", "button")), callback_data=f"sus_reject_{member.from_user.id}")
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sus_allow", "button")),
|
||||
callback_data=f"sus_allow_{member.from_user.id}",
|
||||
)
|
||||
],
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
text=str(locale("sus_reject", "button")),
|
||||
callback_data=f"sus_reject_{member.from_user.id}",
|
||||
)
|
||||
],
|
||||
]
|
||||
]
|
||||
))
|
||||
await app.restrict_chat_member(member.chat.id, member.from_user.id, permissions=ChatPermissions(
|
||||
),
|
||||
)
|
||||
await app.restrict_chat_member(
|
||||
member.chat.id,
|
||||
member.from_user.id,
|
||||
permissions=ChatPermissions(
|
||||
can_send_messages=False,
|
||||
can_send_media_messages=False,
|
||||
can_send_other_messages=False,
|
||||
can_send_polls=False
|
||||
)
|
||||
can_send_polls=False,
|
||||
),
|
||||
)
|
||||
# ==============================================================================================================================
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -7,7 +7,13 @@ from modules.logging import logWrite
|
||||
from modules.utils import configGet, locale
|
||||
from modules import custom_filters
|
||||
|
||||
@app.on_message(custom_filters.enabled_dinovoice & ~filters.scheduled & filters.voice & filters.chat(configGet("users", "groups")))
|
||||
|
||||
@app.on_message(
|
||||
custom_filters.enabled_dinovoice
|
||||
& ~filters.scheduled
|
||||
& filters.voice
|
||||
& filters.chat(configGet("users", "groups"))
|
||||
)
|
||||
async def voice_message(app: Client, msg: Message):
|
||||
logWrite(f"User {msg.from_user.id} sent voice message in destination group")
|
||||
await msg.reply_text(choice(locale("voice_message", "message")))
|
||||
await msg.reply_text(choice(locale("voice_message", "message")))
|
||||
|
@ -12,15 +12,23 @@ for pattern in all_locales("welcome", "keyboard"):
|
||||
welcome_1.append(pattern[0][0])
|
||||
for pattern in all_locales("return", "keyboard"):
|
||||
welcome_1.append(pattern[0][0])
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.private & filters.command(welcome_1, prefixes=[""]) & ~custom_filters.banned)
|
||||
|
||||
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(welcome_1, prefixes=[""])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def welcome_pass(app: Client, msg: Message, once_again: bool = False) -> None:
|
||||
"""Set user's stage to 1 and start a fresh application
|
||||
|
||||
### Args:
|
||||
* app (app): Pyrogram Client to use
|
||||
* app (app): Pyrogram Client to use
|
||||
* msg (Message): Message with .from_user.id attribute equal to the end-user ID whose application will be started
|
||||
* once_again (bool, optional): Set to False if it's the first time as user applies. Defaults to True.
|
||||
"""
|
||||
"""
|
||||
|
||||
if not once_again:
|
||||
await msg.reply_text(locale("privacy_notice", "message"))
|
||||
@ -29,19 +37,41 @@ async def welcome_pass(app: Client, msg: Message, once_again: bool = False) -> N
|
||||
|
||||
if once_again is False:
|
||||
holo_user.application_restart()
|
||||
|
||||
|
||||
if once_again is True:
|
||||
logWrite(f"User {msg.from_user.id} confirmed starting the application")
|
||||
else:
|
||||
logWrite(f"User {msg.from_user.id} confirmed starting the application once again")
|
||||
await msg.reply_text(locale("question1", "message", locale=msg.from_user), reply_markup=ForceReply(placeholder=locale("question1", "force_reply", locale=msg.from_user)))
|
||||
logWrite(
|
||||
f"User {msg.from_user.id} confirmed starting the application once again"
|
||||
)
|
||||
await msg.reply_text(
|
||||
locale("question1", "message", locale=msg.from_user),
|
||||
reply_markup=ForceReply(
|
||||
placeholder=locale("question1", "force_reply", locale=msg.from_user)
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
welcome_2 = []
|
||||
for pattern in all_locales("welcome", "keyboard"):
|
||||
welcome_2.append(pattern[1][0])
|
||||
@app.on_message(custom_filters.enabled_applications & ~filters.scheduled & filters.private & filters.command(welcome_2, prefixes=[""]) & ~custom_filters.banned)
|
||||
async def welcome_reject(app: Client, msg: Message):
|
||||
|
||||
|
||||
@app.on_message(
|
||||
custom_filters.enabled_applications
|
||||
& ~filters.scheduled
|
||||
& filters.private
|
||||
& filters.command(welcome_2, prefixes=[""])
|
||||
& ~custom_filters.banned
|
||||
)
|
||||
async def welcome_reject(app: Client, msg: Message):
|
||||
logWrite(f"User {msg.from_user.id} rejected to start the application")
|
||||
await msg.reply_text(locale("goodbye", "message", locale=msg.from_user), reply_markup=ReplyKeyboardMarkup(locale("return", "keyboard", locale=msg.from_user), resize_keyboard=True))
|
||||
# ==============================================================================================================================
|
||||
await msg.reply_text(
|
||||
locale("goodbye", "message", locale=msg.from_user),
|
||||
reply_markup=ReplyKeyboardMarkup(
|
||||
locale("return", "keyboard", locale=msg.from_user), resize_keyboard=True
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# ==============================================================================================================================
|
||||
|
@ -4,7 +4,13 @@ all inline queries that bot receives"""
|
||||
from datetime import datetime
|
||||
from os import path, sep
|
||||
from app import app, isAnAdmin
|
||||
from pyrogram.types import InlineQueryResultArticle, InputTextMessageContent, InlineQuery, InlineKeyboardMarkup, InlineKeyboardButton
|
||||
from pyrogram.types import (
|
||||
InlineQueryResultArticle,
|
||||
InputTextMessageContent,
|
||||
InlineQuery,
|
||||
InlineKeyboardMarkup,
|
||||
InlineKeyboardButton,
|
||||
)
|
||||
from pyrogram.client import Client
|
||||
from pyrogram.enums.chat_type import ChatType
|
||||
from pyrogram.enums.chat_members_filter import ChatMembersFilter
|
||||
@ -17,51 +23,95 @@ from modules.database import col_applications, col_spoilers
|
||||
from bson.objectid import ObjectId
|
||||
from bson.errors import InvalidId
|
||||
|
||||
|
||||
@app.on_inline_query()
|
||||
async def inline_answer(client: Client, inline_query: InlineQuery):
|
||||
|
||||
results = []
|
||||
|
||||
if configGet("allow_external", "features", "spoilers") is True:
|
||||
|
||||
if inline_query.query.startswith("spoiler:"):
|
||||
|
||||
try:
|
||||
|
||||
spoil = col_spoilers.find_one( {"_id": ObjectId(inline_query.query.removeprefix("spoiler:"))} )
|
||||
spoil = col_spoilers.find_one(
|
||||
{"_id": ObjectId(inline_query.query.removeprefix("spoiler:"))}
|
||||
)
|
||||
|
||||
if spoil is not None:
|
||||
|
||||
desc = locale("spoiler_empty", "message", locale=inline_query.from_user).format(locale(spoil["category"], "message", "spoiler_categories")) if spoil["description"] == "" else locale("spoiler_described", "message", locale=inline_query.from_user).format(locale(spoil["category"], "message", "spoiler_categories"), spoil["description"])
|
||||
desc = (
|
||||
locale(
|
||||
"spoiler_empty", "message", locale=inline_query.from_user
|
||||
).format(
|
||||
locale(spoil["category"], "message", "spoiler_categories")
|
||||
)
|
||||
if spoil["description"] == ""
|
||||
else locale(
|
||||
"spoiler_described",
|
||||
"message",
|
||||
locale=inline_query.from_user,
|
||||
).format(
|
||||
locale(spoil["category"], "message", "spoiler_categories"),
|
||||
spoil["description"],
|
||||
)
|
||||
)
|
||||
|
||||
results = [
|
||||
InlineQueryResultArticle(
|
||||
title=locale("title", "inline", "spoiler", locale=inline_query.from_user),
|
||||
description=locale("description", "inline", "spoiler", locale=inline_query.from_user),
|
||||
input_message_content=InputTextMessageContent(desc, disable_web_page_preview=True),
|
||||
reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(locale("spoiler_view", "button", locale=inline_query.from_user), callback_data=f'sid_{inline_query.query.removeprefix("spoiler:")}')]])
|
||||
title=locale(
|
||||
"title",
|
||||
"inline",
|
||||
"spoiler",
|
||||
locale=inline_query.from_user,
|
||||
),
|
||||
description=locale(
|
||||
"description",
|
||||
"inline",
|
||||
"spoiler",
|
||||
locale=inline_query.from_user,
|
||||
),
|
||||
input_message_content=InputTextMessageContent(
|
||||
desc, disable_web_page_preview=True
|
||||
),
|
||||
reply_markup=InlineKeyboardMarkup(
|
||||
[
|
||||
[
|
||||
InlineKeyboardButton(
|
||||
locale(
|
||||
"spoiler_view",
|
||||
"button",
|
||||
locale=inline_query.from_user,
|
||||
),
|
||||
callback_data=f'sid_{inline_query.query.removeprefix("spoiler:")}',
|
||||
)
|
||||
]
|
||||
]
|
||||
),
|
||||
)
|
||||
]
|
||||
|
||||
except InvalidId:
|
||||
results = []
|
||||
|
||||
|
||||
await inline_query.answer(
|
||||
results=results
|
||||
)
|
||||
await inline_query.answer(results=results)
|
||||
|
||||
return
|
||||
|
||||
if inline_query.chat_type in [ChatType.CHANNEL]:
|
||||
await inline_query.answer(
|
||||
results=[
|
||||
InlineQueryResultArticle(
|
||||
title=locale("title", "inline", "not_pm", locale=inline_query.from_user),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale("message_content", "inline", "not_pm", locale=inline_query.from_user)
|
||||
),
|
||||
description=locale("description", "inline", "not_pm", locale=inline_query.from_user)
|
||||
results=[
|
||||
InlineQueryResultArticle(
|
||||
title=locale(
|
||||
"title", "inline", "not_pm", locale=inline_query.from_user
|
||||
),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale(
|
||||
"message_content",
|
||||
"inline",
|
||||
"not_pm",
|
||||
locale=inline_query.from_user,
|
||||
)
|
||||
),
|
||||
description=locale(
|
||||
"description", "inline", "not_pm", locale=inline_query.from_user
|
||||
),
|
||||
)
|
||||
]
|
||||
)
|
||||
@ -71,39 +121,59 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
|
||||
InlineQueryResultArticle(
|
||||
title=locale("title", "inline", "forbidden", locale=inline_query.from_user),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale("message_content", "inline", "forbidden", locale=inline_query.from_user)
|
||||
locale(
|
||||
"message_content",
|
||||
"inline",
|
||||
"forbidden",
|
||||
locale=inline_query.from_user,
|
||||
)
|
||||
),
|
||||
description=locale(
|
||||
"description", "inline", "forbidden", locale=inline_query.from_user
|
||||
),
|
||||
description=locale("description", "inline", "forbidden", locale=inline_query.from_user)
|
||||
)
|
||||
]
|
||||
|
||||
try:
|
||||
holo_user = HoloUser(inline_query.from_user)
|
||||
except (UserNotFoundError, UserInvalidError):
|
||||
logWrite(f"Could not find application of {inline_query.from_user.id}, ignoring inline query", debug=True)
|
||||
await inline_query.answer(
|
||||
results=results_forbidden
|
||||
logWrite(
|
||||
f"Could not find application of {inline_query.from_user.id}, ignoring inline query",
|
||||
debug=True,
|
||||
)
|
||||
await inline_query.answer(results=results_forbidden)
|
||||
return
|
||||
|
||||
if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (inline_query.from_user.id not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))):
|
||||
if path.exists(path.join(configGet("cache", "locations"), "admins")) and (inline_query.from_user.id not in jsonLoad(path.join(configGet("cache", "locations"), "admins"))):
|
||||
logWrite(f"{inline_query.from_user.id} is not an admin and not in members group, ignoring inline query", debug=True)
|
||||
await inline_query.answer(
|
||||
results=results_forbidden
|
||||
if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (
|
||||
inline_query.from_user.id
|
||||
not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))
|
||||
):
|
||||
if path.exists(path.join(configGet("cache", "locations"), "admins")) and (
|
||||
inline_query.from_user.id
|
||||
not in jsonLoad(path.join(configGet("cache", "locations"), "admins"))
|
||||
):
|
||||
logWrite(
|
||||
f"{inline_query.from_user.id} is not an admin and not in members group, ignoring inline query",
|
||||
debug=True,
|
||||
)
|
||||
await inline_query.answer(results=results_forbidden)
|
||||
return
|
||||
|
||||
if holo_user.application_approved() or (await isAnAdmin(holo_user.id) is True):
|
||||
|
||||
max_results = configGet("inline_preview_count") if inline_query.query != "" else 200
|
||||
max_results = (
|
||||
configGet("inline_preview_count") if inline_query.query != "" else 200
|
||||
)
|
||||
|
||||
list_of_users = []
|
||||
async for m in app.get_chat_members(configGet("users", "groups"), limit=max_results, filter=ChatMembersFilter.SEARCH, query=inline_query.query):
|
||||
async for m in app.get_chat_members(
|
||||
configGet("users", "groups"),
|
||||
limit=max_results,
|
||||
filter=ChatMembersFilter.SEARCH,
|
||||
query=inline_query.query,
|
||||
):
|
||||
list_of_users.append(m)
|
||||
|
||||
for match in list_of_users:
|
||||
|
||||
application = col_applications.find_one({"user": match.user.id})
|
||||
|
||||
if application is None:
|
||||
@ -112,34 +182,63 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
|
||||
application_content = []
|
||||
i = 1
|
||||
|
||||
for question in application['application']:
|
||||
|
||||
for question in application["application"]:
|
||||
if i == 2:
|
||||
age = relativedelta(datetime.now(), application['application']['2'])
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)")
|
||||
age = relativedelta(datetime.now(), application["application"]["2"])
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)"
|
||||
)
|
||||
elif i == 3:
|
||||
if application['application']['3']['countryCode'] == "UA":
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']}")
|
||||
if application["application"]["3"]["countryCode"] == "UA":
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']}"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})")
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})"
|
||||
)
|
||||
else:
|
||||
application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application'][question]}")
|
||||
|
||||
application_content.append(
|
||||
f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application'][question]}"
|
||||
)
|
||||
|
||||
i += 1
|
||||
|
||||
if match.user.photo != None:
|
||||
try:
|
||||
if not path.exists(f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}'):
|
||||
print(f'Downloaded avatar {match.user.photo.big_file_id} of {match.user.id} and uploaded to {configGet("api")}/?avatar_id={match.user.photo.big_file_id}', flush=True)
|
||||
await app.download_media(match.user.photo.big_file_id, file_name=f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}')
|
||||
if not path.exists(
|
||||
f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}'
|
||||
):
|
||||
print(
|
||||
f'Downloaded avatar {match.user.photo.big_file_id} of {match.user.id} and uploaded to {configGet("api")}/?avatar_id={match.user.photo.big_file_id}',
|
||||
flush=True,
|
||||
)
|
||||
await app.download_media(
|
||||
match.user.photo.big_file_id,
|
||||
file_name=f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}',
|
||||
)
|
||||
results.append(
|
||||
InlineQueryResultArticle(
|
||||
title=str(match.user.first_name),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale("message_content", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||||
locale(
|
||||
"message_content",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(
|
||||
match.user.first_name,
|
||||
match.user.username,
|
||||
"\n".join(application_content),
|
||||
)
|
||||
),
|
||||
description=locale("description", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username),
|
||||
thumb_url=f'{configGet("api")}/?avatar_id={match.user.photo.big_file_id}'
|
||||
description=locale(
|
||||
"description",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(match.user.first_name, match.user.username),
|
||||
thumb_url=f'{configGet("api")}/?avatar_id={match.user.photo.big_file_id}',
|
||||
)
|
||||
)
|
||||
except ValueError:
|
||||
@ -147,9 +246,23 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
|
||||
InlineQueryResultArticle(
|
||||
title=str(match.user.first_name),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale("message_content", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||||
locale(
|
||||
"message_content",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(
|
||||
match.user.first_name,
|
||||
match.user.username,
|
||||
"\n".join(application_content),
|
||||
)
|
||||
),
|
||||
description=locale("description", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username)
|
||||
description=locale(
|
||||
"description",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(match.user.first_name, match.user.username),
|
||||
)
|
||||
)
|
||||
except FileNotFoundError:
|
||||
@ -157,9 +270,23 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
|
||||
InlineQueryResultArticle(
|
||||
title=str(match.user.first_name),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale("message_content", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||||
locale(
|
||||
"message_content",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(
|
||||
match.user.first_name,
|
||||
match.user.username,
|
||||
"\n".join(application_content),
|
||||
)
|
||||
),
|
||||
description=locale("description", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username)
|
||||
description=locale(
|
||||
"description",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(match.user.first_name, match.user.username),
|
||||
)
|
||||
)
|
||||
else:
|
||||
@ -167,14 +294,24 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
|
||||
InlineQueryResultArticle(
|
||||
title=str(match.user.first_name),
|
||||
input_message_content=InputTextMessageContent(
|
||||
locale("message_content", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||||
locale(
|
||||
"message_content",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(
|
||||
match.user.first_name,
|
||||
match.user.username,
|
||||
"\n".join(application_content),
|
||||
)
|
||||
),
|
||||
description=locale("description", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username)
|
||||
description=locale(
|
||||
"description",
|
||||
"inline",
|
||||
"user",
|
||||
locale=inline_query.from_user,
|
||||
).format(match.user.first_name, match.user.username),
|
||||
)
|
||||
)
|
||||
|
||||
await inline_query.answer(
|
||||
results=results,
|
||||
cache_time=10,
|
||||
is_personal=True
|
||||
)
|
||||
await inline_query.answer(results=results, cache_time=10, is_personal=True)
|
||||
|
@ -5,19 +5,20 @@ from shutil import copyfileobj
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
with open(getcwd()+path.sep+"config.json", "r", encoding='utf8') as file:
|
||||
with open(getcwd() + path.sep + "config.json", "r", encoding="utf8") as file:
|
||||
json_contents = loads(file.read())
|
||||
log_size = json_contents["logging"]["size"]
|
||||
log_folder = json_contents["logging"]["location"]
|
||||
file.close()
|
||||
|
||||
|
||||
# Check latest log size
|
||||
def checkSize(debug=False):
|
||||
"""Check size of latest.log file and rotate it if needed
|
||||
|
||||
### Args:
|
||||
* debug (`bool`, *optional*): Whether this is a debug log. Defaults to `False`.
|
||||
"""
|
||||
"""
|
||||
|
||||
global log_folder
|
||||
|
||||
@ -30,15 +31,24 @@ def checkSize(debug=False):
|
||||
makedirs(log_folder, exist_ok=True)
|
||||
log = stat(path.join(log_folder, log_file))
|
||||
if (log.st_size / 1024) > log_size:
|
||||
with open(path.join(log_folder, log_file), 'rb') as f_in:
|
||||
with gzipopen(path.join(log_folder, f'{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.log.gz'), 'wb') as f_out:
|
||||
with open(path.join(log_folder, log_file), "rb") as f_in:
|
||||
with gzipopen(
|
||||
path.join(
|
||||
log_folder,
|
||||
f'{datetime.now().strftime("%d.%m.%Y_%H:%M:%S")}.log.gz',
|
||||
),
|
||||
"wb",
|
||||
) as f_out:
|
||||
copyfileobj(f_in, f_out)
|
||||
print(f'Copied {path.join(log_folder, datetime.now().strftime("%d.%m.%Y_%H:%M:%S"))}.log.gz')
|
||||
open(path.join(log_folder, log_file), 'w').close()
|
||||
print(
|
||||
f'Copied {path.join(log_folder, datetime.now().strftime("%d.%m.%Y_%H:%M:%S"))}.log.gz'
|
||||
)
|
||||
open(path.join(log_folder, log_file), "w").close()
|
||||
except FileNotFoundError:
|
||||
print(f'Log file {path.join(log_folder, log_file)} does not exist')
|
||||
print(f"Log file {path.join(log_folder, log_file)} does not exist")
|
||||
pass
|
||||
|
||||
|
||||
# Append string to log
|
||||
def logAppend(message: str, debug=False):
|
||||
"""Write message to log file
|
||||
@ -46,7 +56,7 @@ def logAppend(message: str, debug=False):
|
||||
### Args:
|
||||
* message (`str`): Message to write
|
||||
* debug (`bool`, *optional*): Whether this is a debug log. Defaults to `False`.
|
||||
"""
|
||||
"""
|
||||
|
||||
global log_folder
|
||||
|
||||
@ -58,10 +68,11 @@ def logAppend(message: str, debug=False):
|
||||
else:
|
||||
log_file = "latest.log"
|
||||
|
||||
log = open(path.join(log_folder, log_file), 'a')
|
||||
log.write(f'{message_formatted}\n')
|
||||
log = open(path.join(log_folder, log_file), "a")
|
||||
log.write(f"{message_formatted}\n")
|
||||
log.close()
|
||||
|
||||
|
||||
# Print to stdout and then to log
|
||||
def logWrite(message: str, debug=False):
|
||||
"""Write message to stdout and log file
|
||||
@ -69,7 +80,7 @@ def logWrite(message: str, debug=False):
|
||||
### Args:
|
||||
* message (`str`): Message to print and write
|
||||
* debug (`bool`, *optional*): Whether this is a debug log. Defaults to `False`.
|
||||
"""
|
||||
"""
|
||||
# save to log file and rotation is to be done
|
||||
logAppend(f'{message}', debug=debug)
|
||||
print(f"{message}", flush=True)
|
||||
logAppend(f"{message}", debug=debug)
|
||||
print(f"{message}", flush=True)
|
||||
|
@ -8,7 +8,11 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
from datetime import datetime, timedelta
|
||||
from ujson import dumps
|
||||
from app import app
|
||||
from pyrogram.types import BotCommand, BotCommandScopeChat, BotCommandScopeChatAdministrators
|
||||
from pyrogram.types import (
|
||||
BotCommand,
|
||||
BotCommandScopeChat,
|
||||
BotCommandScopeChatAdministrators,
|
||||
)
|
||||
from pyrogram.errors import bad_request_400
|
||||
from pyrogram.enums.chat_members_filter import ChatMembersFilter
|
||||
from classes.holo_user import HoloUser
|
||||
@ -21,18 +25,27 @@ from requests import get
|
||||
scheduler = AsyncIOScheduler()
|
||||
|
||||
if configGet("enabled", "scheduler", "cache_members"):
|
||||
@scheduler.scheduled_job(trigger="interval", seconds=configGet("interval", "scheduler", "cache_members"))
|
||||
|
||||
@scheduler.scheduled_job(
|
||||
trigger="interval", seconds=configGet("interval", "scheduler", "cache_members")
|
||||
)
|
||||
async def cache_group_members():
|
||||
list_of_users = []
|
||||
async for member in app.get_chat_members(configGet("users", "groups")):
|
||||
list_of_users.append(member.user.id)
|
||||
makedirs(configGet("cache", "locations"), exist_ok=True)
|
||||
jsonSave(list_of_users, path.join(configGet("cache", "locations"), "group_members"))
|
||||
jsonSave(
|
||||
list_of_users, path.join(configGet("cache", "locations"), "group_members")
|
||||
)
|
||||
if configGet("debug") is True:
|
||||
logWrite("User group caching performed", debug=True)
|
||||
|
||||
|
||||
if configGet("enabled", "scheduler", "cache_admins"):
|
||||
@scheduler.scheduled_job(trigger="interval", seconds=configGet("interval", "scheduler", "cache_admins"))
|
||||
|
||||
@scheduler.scheduled_job(
|
||||
trigger="interval", seconds=configGet("interval", "scheduler", "cache_admins")
|
||||
)
|
||||
async def cache_admins():
|
||||
list_of_users = []
|
||||
async for member in app.get_chat_members(configGet("admin", "groups")):
|
||||
@ -42,37 +55,66 @@ if configGet("enabled", "scheduler", "cache_admins"):
|
||||
if configGet("debug") is True:
|
||||
logWrite("Admin group caching performed", debug=True)
|
||||
|
||||
|
||||
# Cache the avatars of group members
|
||||
if configGet("enabled", "scheduler", "cache_avatars"):
|
||||
@scheduler.scheduled_job(trigger="date", run_date=datetime.now()+timedelta(seconds=15))
|
||||
@scheduler.scheduled_job(trigger="interval", hours=configGet("interval", "scheduler", "cache_avatars"))
|
||||
|
||||
@scheduler.scheduled_job(
|
||||
trigger="date", run_date=datetime.now() + timedelta(seconds=15)
|
||||
)
|
||||
@scheduler.scheduled_job(
|
||||
trigger="interval", hours=configGet("interval", "scheduler", "cache_avatars")
|
||||
)
|
||||
async def cache_avatars():
|
||||
list_of_users = []
|
||||
async for member in app.get_chat_members(configGet("users", "groups"), filter=ChatMembersFilter.SEARCH, query=""):
|
||||
async for member in app.get_chat_members(
|
||||
configGet("users", "groups"), filter=ChatMembersFilter.SEARCH, query=""
|
||||
):
|
||||
list_of_users.append(member.user)
|
||||
for user in list_of_users:
|
||||
if user.photo != None:
|
||||
if not path.exists(f'{configGet("cache", "locations")}{sep}avatars{sep}{user.photo.big_file_id}'):
|
||||
print(f'Pre-cached avatar {user.photo.big_file_id} of {user.id}', flush=True)
|
||||
await app.download_media(user.photo.big_file_id, file_name=path.join(configGet("cache", "locations"), "avatars", user.photo.big_file_id))
|
||||
if not path.exists(
|
||||
f'{configGet("cache", "locations")}{sep}avatars{sep}{user.photo.big_file_id}'
|
||||
):
|
||||
print(
|
||||
f"Pre-cached avatar {user.photo.big_file_id} of {user.id}",
|
||||
flush=True,
|
||||
)
|
||||
await app.download_media(
|
||||
user.photo.big_file_id,
|
||||
file_name=path.join(
|
||||
configGet("cache", "locations"),
|
||||
"avatars",
|
||||
user.photo.big_file_id,
|
||||
),
|
||||
)
|
||||
logWrite("Avatars caching performed")
|
||||
|
||||
|
||||
# Check for birthdays
|
||||
if configGet("enabled", "features", "applications") is True:
|
||||
if configGet("enabled", "scheduler", "birthdays") is True:
|
||||
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "birthdays"))
|
||||
|
||||
@scheduler.scheduled_job(
|
||||
trigger="cron", hour=configGet("time", "scheduler", "birthdays")
|
||||
)
|
||||
async def check_birthdays():
|
||||
for entry in col_applications.find():
|
||||
if entry["application"]["2"].strftime("%d.%m") == datetime.now().strftime("%d.%m"):
|
||||
if entry["application"]["2"].strftime(
|
||||
"%d.%m"
|
||||
) == datetime.now().strftime("%d.%m"):
|
||||
try:
|
||||
if entry["user"] not in jsonLoad(path.join(configGet("cache", "locations"), "group_members")):
|
||||
if entry["user"] not in jsonLoad(
|
||||
path.join(configGet("cache", "locations"), "group_members")
|
||||
):
|
||||
continue
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
await app.send_message( configGet("admin", "groups"), locale("birthday", "message").format(str(tg_user.first_name), str(tg_user.username), str(relativedelta(datetime.now(), entry["application"]["2"], '%d.%m.%Y').years)) ) # type: ignore
|
||||
await app.send_message(configGet("admin", "groups"), locale("birthday", "message").format(str(tg_user.first_name), str(tg_user.username), str(relativedelta(datetime.now(), entry["application"]["2"], "%d.%m.%Y").years))) # type: ignore
|
||||
logWrite(f"Notified admins about {entry['user']}'s birthday")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} to send a message about birthday due to '{exp}'")
|
||||
logWrite(
|
||||
f"Could not find user {entry['user']} to send a message about birthday due to '{exp}'"
|
||||
)
|
||||
continue
|
||||
logWrite("Birthdays check performed")
|
||||
|
||||
@ -80,42 +122,71 @@ if configGet("enabled", "features", "applications") is True:
|
||||
# Check for expired sponsorships
|
||||
if configGet("enabled", "features", "sponsorships") is True:
|
||||
if configGet("enabled", "scheduler", "sponsorships") is True:
|
||||
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "sponsorships"))
|
||||
|
||||
@scheduler.scheduled_job(
|
||||
trigger="cron", hour=configGet("time", "scheduler", "sponsorships")
|
||||
)
|
||||
async def check_sponsors():
|
||||
for entry in col_sponsorships.find({"sponsorship.expires": {"$lt": datetime.now()+timedelta(days=2)}}):
|
||||
for entry in col_sponsorships.find(
|
||||
{"sponsorship.expires": {"$lt": datetime.now() + timedelta(days=2)}}
|
||||
):
|
||||
try:
|
||||
if entry["user"] not in jsonLoad(path.join(configGet("cache", "locations"), "group_members")):
|
||||
if entry["user"] not in jsonLoad(
|
||||
path.join(configGet("cache", "locations"), "group_members")
|
||||
):
|
||||
continue
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
until_expiry = abs(relativedelta(datetime.now(), entry["sponsorship"]["expires"]).days)+1
|
||||
await app.send_message( tg_user.id, locale("sponsorships_expires", "message").format(until_expiry) ) # type: ignore
|
||||
logWrite(f"Notified user {entry['user']} that sponsorship expires in {until_expiry} days")
|
||||
until_expiry = (
|
||||
abs(
|
||||
relativedelta(
|
||||
datetime.now(), entry["sponsorship"]["expires"]
|
||||
).days
|
||||
)
|
||||
+ 1
|
||||
)
|
||||
await app.send_message(tg_user.id, locale("sponsorships_expires", "message").format(until_expiry)) # type: ignore
|
||||
logWrite(
|
||||
f"Notified user {entry['user']} that sponsorship expires in {until_expiry} days"
|
||||
)
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} notify about sponsorship expiry due to '{exp}'")
|
||||
logWrite(
|
||||
f"Could not find user {entry['user']} notify about sponsorship expiry due to '{exp}'"
|
||||
)
|
||||
continue
|
||||
for entry in col_sponsorships.find({"sponsorship.expires": {"$lt": datetime.now()-timedelta(days=1)}}):
|
||||
for entry in col_sponsorships.find(
|
||||
{"sponsorship.expires": {"$lt": datetime.now() - timedelta(days=1)}}
|
||||
):
|
||||
try:
|
||||
holo_user = HoloUser(entry["user"])
|
||||
col_sponsorships.find_one_and_delete({"user": holo_user.id})
|
||||
if entry["user"] not in jsonLoad(path.join(configGet("cache", "locations"), "group_members")):
|
||||
if entry["user"] not in jsonLoad(
|
||||
path.join(configGet("cache", "locations"), "group_members")
|
||||
):
|
||||
continue
|
||||
await app.send_message( entry["user"], locale("sponsorships_expired", "message") ) # type: ignore
|
||||
await app.send_message(entry["user"], locale("sponsorships_expired", "message")) # type: ignore
|
||||
await holo_user.label_reset(configGet("users", "groups"))
|
||||
try:
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
logWrite(f"Notified user {entry['user']} that sponsorship expired")
|
||||
logWrite(
|
||||
f"Notified user {entry['user']} that sponsorship expired"
|
||||
)
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} notify about sponsorship expired due to '{exp}'")
|
||||
logWrite(
|
||||
f"Could not find user {entry['user']} notify about sponsorship expired due to '{exp}'"
|
||||
)
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not reset label of user {entry['user']} due to '{exp}'")
|
||||
logWrite(
|
||||
f"Could not reset label of user {entry['user']} due to '{exp}'"
|
||||
)
|
||||
continue
|
||||
logWrite("Sponsorships check performed")
|
||||
|
||||
|
||||
# Register all bot commands
|
||||
@scheduler.scheduled_job(trigger="date", run_date=datetime.now()+timedelta(seconds=10))
|
||||
@scheduler.scheduled_job(
|
||||
trigger="date", run_date=datetime.now() + timedelta(seconds=10)
|
||||
)
|
||||
async def commands_register():
|
||||
|
||||
commands = {
|
||||
"users": [],
|
||||
"admins": [],
|
||||
@ -123,7 +194,7 @@ async def commands_register():
|
||||
"group_users": [],
|
||||
"group_admins": [],
|
||||
"group_users_admins": [],
|
||||
"locales": {}
|
||||
"locales": {},
|
||||
}
|
||||
|
||||
commands_raw = {
|
||||
@ -133,7 +204,7 @@ async def commands_register():
|
||||
"group_users": [],
|
||||
"group_admins": [],
|
||||
"group_users_admins": [],
|
||||
"locales": {}
|
||||
"locales": {},
|
||||
}
|
||||
|
||||
valid_locales = []
|
||||
@ -148,7 +219,7 @@ async def commands_register():
|
||||
"owner": [],
|
||||
"group_users": [],
|
||||
"group_admins": [],
|
||||
"group_users_admins": []
|
||||
"group_users_admins": [],
|
||||
}
|
||||
if configGet("debug") is True:
|
||||
commands_raw["locales"][".".join(entry.split(".")[:-1])] = {
|
||||
@ -157,15 +228,13 @@ async def commands_register():
|
||||
"owner": [],
|
||||
"group_users": [],
|
||||
"group_admins": [],
|
||||
"group_users_admins": []
|
||||
"group_users_admins": [],
|
||||
}
|
||||
|
||||
config_modules = configGet("features")
|
||||
config_commands = configGet("commands")
|
||||
|
||||
|
||||
for command in config_commands:
|
||||
|
||||
enabled = False
|
||||
|
||||
for module in config_commands[command]["modules"]:
|
||||
@ -177,22 +246,27 @@ async def commands_register():
|
||||
continue
|
||||
|
||||
for permission in config_commands[command]["permissions"]:
|
||||
|
||||
commands[permission].append(BotCommand(command, locale("commands")[command]))
|
||||
commands[permission].append(
|
||||
BotCommand(command, locale("commands")[command])
|
||||
)
|
||||
|
||||
if configGet("debug") is True:
|
||||
commands_raw[permission].append({f"{command}": locale("commands")[command]})
|
||||
commands_raw[permission].append(
|
||||
{f"{command}": locale("commands")[command]}
|
||||
)
|
||||
logWrite(f"Registering {command} for {permission}")
|
||||
|
||||
for lc in valid_locales:
|
||||
|
||||
commands["locales"][lc][permission].append(BotCommand(command, locale("commands", locale=lc)[command]))
|
||||
commands["locales"][lc][permission].append(
|
||||
BotCommand(command, locale("commands", locale=lc)[command])
|
||||
)
|
||||
|
||||
if configGet("debug") is True:
|
||||
commands_raw["locales"][lc][permission].append({f"{command}": locale("commands", locale=lc)[command]})
|
||||
commands_raw["locales"][lc][permission].append(
|
||||
{f"{command}": locale("commands", locale=lc)[command]}
|
||||
)
|
||||
logWrite(f"Registering {command} for {permission} [{lc}]")
|
||||
|
||||
|
||||
# Registering user commands
|
||||
await app.set_bot_commands(commands["users"])
|
||||
logWrite("Registered user commands for default locale")
|
||||
@ -205,55 +279,94 @@ async def commands_register():
|
||||
# Registering admin commands
|
||||
for admin in configGet("admins"):
|
||||
try:
|
||||
await app.set_bot_commands(commands["admins"]+commands["users"], scope=BotCommandScopeChat(chat_id=admin))
|
||||
await app.set_bot_commands(
|
||||
commands["admins"] + commands["users"],
|
||||
scope=BotCommandScopeChat(chat_id=admin),
|
||||
)
|
||||
logWrite(f"Registered admin commands for admin {admin}")
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
pass
|
||||
|
||||
# Registering owner commands
|
||||
try:
|
||||
await app.set_bot_commands(commands["admins"]+commands["owner"]+commands["users"], scope=BotCommandScopeChat(chat_id=configGet("owner")))
|
||||
await app.set_bot_commands(
|
||||
commands["admins"] + commands["owner"] + commands["users"],
|
||||
scope=BotCommandScopeChat(chat_id=configGet("owner")),
|
||||
)
|
||||
for lc in valid_locales:
|
||||
await app.set_bot_commands(commands["locales"][lc]["admins"]+commands["locales"][lc]["owner"]+commands["locales"][lc]["users"], scope=BotCommandScopeChat(chat_id=configGet("owner")))
|
||||
await app.set_bot_commands(
|
||||
commands["locales"][lc]["admins"]
|
||||
+ commands["locales"][lc]["owner"]
|
||||
+ commands["locales"][lc]["users"],
|
||||
scope=BotCommandScopeChat(chat_id=configGet("owner")),
|
||||
)
|
||||
logWrite(f"Registered admin commands for owner {configGet('owner')}")
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
pass
|
||||
|
||||
# Registering admin group commands
|
||||
try:
|
||||
await app.set_bot_commands(commands["group_admins"], scope=BotCommandScopeChat(chat_id=configGet("admin", "groups")))
|
||||
await app.set_bot_commands(
|
||||
commands["group_admins"],
|
||||
scope=BotCommandScopeChat(chat_id=configGet("admin", "groups")),
|
||||
)
|
||||
logWrite("Registered admin group commands for default locale")
|
||||
except bad_request_400.ChannelInvalid:
|
||||
logWrite(f"Could not register commands for admin group. Bot is likely not in the group.")
|
||||
logWrite(
|
||||
f"Could not register commands for admin group. Bot is likely not in the group."
|
||||
)
|
||||
|
||||
# Registering destination group commands
|
||||
try:
|
||||
await app.set_bot_commands(commands["group_users"], scope=BotCommandScopeChat(chat_id=configGet("users", "groups")))
|
||||
await app.set_bot_commands(
|
||||
commands["group_users"],
|
||||
scope=BotCommandScopeChat(chat_id=configGet("users", "groups")),
|
||||
)
|
||||
logWrite("Registered destination group commands")
|
||||
except bad_request_400.ChannelInvalid:
|
||||
logWrite(f"Could not register commands for destination group. Bot is likely not in the group.")
|
||||
logWrite(
|
||||
f"Could not register commands for destination group. Bot is likely not in the group."
|
||||
)
|
||||
|
||||
# Registering destination group admin commands
|
||||
try:
|
||||
await app.set_bot_commands(commands["group_users_admins"], scope=BotCommandScopeChatAdministrators(chat_id=configGet("users", "groups")))
|
||||
await app.set_bot_commands(
|
||||
commands["group_users_admins"],
|
||||
scope=BotCommandScopeChatAdministrators(
|
||||
chat_id=configGet("users", "groups")
|
||||
),
|
||||
)
|
||||
logWrite("Registered destination group admin commands")
|
||||
except bad_request_400.ChannelInvalid:
|
||||
logWrite(f"Could not register admin commands for destination group. Bot is likely not in the group.")
|
||||
logWrite(
|
||||
f"Could not register admin commands for destination group. Bot is likely not in the group."
|
||||
)
|
||||
|
||||
|
||||
if configGet("debug") is True:
|
||||
print(commands, flush=True)
|
||||
logWrite(f"Complete commands registration:\n{dumps(commands_raw, indent=4, ensure_ascii=False, encode_html_chars=False)}", debug=True)
|
||||
logWrite(
|
||||
f"Complete commands registration:\n{dumps(commands_raw, indent=4, ensure_ascii=False, encode_html_chars=False)}",
|
||||
debug=True,
|
||||
)
|
||||
|
||||
|
||||
if configGet("enabled", "scheduler", "channels_monitor"):
|
||||
@scheduler.scheduled_job(trigger="interval", minutes=configGet("interval", "scheduler", "channels_monitor"))
|
||||
|
||||
@scheduler.scheduled_job(
|
||||
trigger="interval",
|
||||
minutes=configGet("interval", "scheduler", "channels_monitor"),
|
||||
)
|
||||
async def channels_monitor():
|
||||
for channel in configGet("channels", "scheduler", "channels_monitor"):
|
||||
if configGet("debug") is True:
|
||||
logWrite(f'Processing videos of {channel["name"]} ({channel["id"]})', debug=True)
|
||||
logWrite(
|
||||
f'Processing videos of {channel["name"]} ({channel["id"]})',
|
||||
debug=True,
|
||||
)
|
||||
try:
|
||||
req = get(f'https://www.youtube.com/feeds/videos.xml?channel_id={channel["id"]}')
|
||||
req = get(
|
||||
f'https://www.youtube.com/feeds/videos.xml?channel_id={channel["id"]}'
|
||||
)
|
||||
parsed = parse(req.content)
|
||||
if "feed" not in parsed:
|
||||
continue
|
||||
@ -262,11 +375,33 @@ if configGet("enabled", "scheduler", "channels_monitor"):
|
||||
for entry in parsed["feed"]["entry"]:
|
||||
if "yt:videoId" not in entry:
|
||||
continue
|
||||
if col_youtube.find_one( {"channel": channel["id"], "video": entry["yt:videoId"]} ) is None:
|
||||
col_youtube.insert_one( {"channel": channel["id"], "video": entry["yt:videoId"], "date": datetime.fromisoformat(entry["published"])} )
|
||||
await app.send_message(configGet("users", "groups"), locale("youtube_video", "message").format(channel["name"], channel["link"], entry["title"], entry["link"]["@href"]), disable_web_page_preview=False)
|
||||
if (
|
||||
col_youtube.find_one(
|
||||
{"channel": channel["id"], "video": entry["yt:videoId"]}
|
||||
)
|
||||
is None
|
||||
):
|
||||
col_youtube.insert_one(
|
||||
{
|
||||
"channel": channel["id"],
|
||||
"video": entry["yt:videoId"],
|
||||
"date": datetime.fromisoformat(entry["published"]),
|
||||
}
|
||||
)
|
||||
await app.send_message(
|
||||
configGet("users", "groups"),
|
||||
locale("youtube_video", "message").format(
|
||||
channel["name"],
|
||||
channel["link"],
|
||||
entry["title"],
|
||||
entry["link"]["@href"],
|
||||
),
|
||||
disable_web_page_preview=False,
|
||||
)
|
||||
await sleep(2)
|
||||
except Exception as exp:
|
||||
logWrite(f'Could not get last videos of {channel["name"]} ({channel["id"]}) due to {exp}: {format_exc()}')
|
||||
logWrite(
|
||||
f'Could not get last videos of {channel["name"]} ({channel["id"]}) due to {exp}: {format_exc()}'
|
||||
)
|
||||
if configGet("debug") is True:
|
||||
logWrite("Admin group caching performed", debug=True)
|
||||
logWrite("Admin group caching performed", debug=True)
|
||||
|
@ -16,23 +16,29 @@ from classes.errors.geo import PlaceNotFoundError
|
||||
|
||||
from modules.logging import logWrite
|
||||
|
||||
|
||||
def jsonLoad(filename):
|
||||
"""Loads arg1 as json and returns its contents"""
|
||||
with open(filename, "r", encoding='utf8') as file:
|
||||
with open(filename, "r", encoding="utf8") as file:
|
||||
try:
|
||||
output = loads(file.read())
|
||||
except JSONDecodeError:
|
||||
logWrite(f"Could not load json file {filename}: file seems to be incorrect!\n{print_exc()}")
|
||||
logWrite(
|
||||
f"Could not load json file {filename}: file seems to be incorrect!\n{print_exc()}"
|
||||
)
|
||||
raise
|
||||
except FileNotFoundError:
|
||||
logWrite(f"Could not load json file {filename}: file does not seem to exist!\n{print_exc()}")
|
||||
logWrite(
|
||||
f"Could not load json file {filename}: file does not seem to exist!\n{print_exc()}"
|
||||
)
|
||||
raise
|
||||
return output
|
||||
|
||||
|
||||
def jsonSave(contents, filename):
|
||||
"""Dumps dict/list arg1 to file arg2"""
|
||||
try:
|
||||
with open(filename, "w", encoding='utf8') as file:
|
||||
with open(filename, "w", encoding="utf8") as file:
|
||||
file.write(dumps(contents, ensure_ascii=False, indent=4))
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not save json file {filename}: {exp}\n{print_exc()}")
|
||||
@ -52,6 +58,7 @@ def nested_set(dic, keys, value, create_missing=True):
|
||||
d[keys[-1]] = value
|
||||
return dic
|
||||
|
||||
|
||||
def configSet(keys: list, value: Any, file: str = "config", create_missing=True):
|
||||
"""Set config's value to provided one
|
||||
|
||||
@ -60,7 +67,7 @@ def configSet(keys: list, value: Any, file: str = "config", create_missing=True)
|
||||
* value (`Any`): Needed value
|
||||
* file (`str`, optional): File (if not config). Defaults to "config".
|
||||
* create_missing (`bool`, optional): Create missing items on the way. Defaults to True.
|
||||
"""
|
||||
"""
|
||||
if file == "config":
|
||||
filepath = ""
|
||||
this_dict = jsonLoad(f"{filepath}{file}.json")
|
||||
@ -69,15 +76,19 @@ def configSet(keys: list, value: Any, file: str = "config", create_missing=True)
|
||||
this_dict = jsonLoad("config_debug.json")
|
||||
file = "config_debug"
|
||||
except FileNotFoundError:
|
||||
print("Debug mode is set but config_debug.json is not there! Falling back to config.json", flush=True)
|
||||
print(
|
||||
"Debug mode is set but config_debug.json is not there! Falling back to config.json",
|
||||
flush=True,
|
||||
)
|
||||
else:
|
||||
filepath = f"data{sep}users{sep}"
|
||||
this_dict = jsonLoad(f"{filepath}{file}.json")
|
||||
|
||||
|
||||
this_dict = nested_set(this_dict, keys, value, create_missing=create_missing)
|
||||
jsonSave(this_dict, f"{filepath}{file}.json")
|
||||
return
|
||||
|
||||
|
||||
def configGet(key: str, *args: str, file: str = "config"):
|
||||
"""Get value of the config key
|
||||
### Args:
|
||||
@ -86,18 +97,24 @@ def configGet(key: str, *args: str, file: str = "config"):
|
||||
* file (`str`): User ID to load. Loads config if not provided. Defaults to "config".
|
||||
### Returns:
|
||||
* any: Value of provided key
|
||||
"""
|
||||
"""
|
||||
if file == "config":
|
||||
try:
|
||||
this_dict = jsonLoad("config.json")
|
||||
except FileNotFoundError:
|
||||
print("Config file not found! Copy config_example.json to config.json, configure it and rerun the bot!", flush=True)
|
||||
print(
|
||||
"Config file not found! Copy config_example.json to config.json, configure it and rerun the bot!",
|
||||
flush=True,
|
||||
)
|
||||
exit()
|
||||
if this_dict["debug"] is True:
|
||||
try:
|
||||
this_dict = jsonLoad("config_debug.json")
|
||||
except FileNotFoundError:
|
||||
print("Debug mode is set but config_debug.json is not there! Falling back to config.json", flush=True)
|
||||
print(
|
||||
"Debug mode is set but config_debug.json is not there! Falling back to config.json",
|
||||
flush=True,
|
||||
)
|
||||
else:
|
||||
this_dict = jsonLoad(f"data{sep}users{sep}{file}.json")
|
||||
this_key = this_dict
|
||||
@ -105,6 +122,7 @@ def configGet(key: str, *args: str, file: str = "config"):
|
||||
this_key = this_key[dict_key]
|
||||
return this_key[key]
|
||||
|
||||
|
||||
def locale(key: str, *args: str, locale: Union[str, User] = configGet("locale")) -> Any:
|
||||
"""Get value of locale string
|
||||
### Args:
|
||||
@ -113,32 +131,35 @@ def locale(key: str, *args: str, locale: Union[str, User] = configGet("locale"))
|
||||
* locale (`Union[str, User, HoloUser]`): Locale to looked up in. Provide User to get his `.language_code`. Defaults to config's locale value.
|
||||
### Returns:
|
||||
* any: Value of provided locale key. In normal case must be `str`, `dict` or `list`.
|
||||
"""
|
||||
"""
|
||||
if isinstance(locale, User):
|
||||
locale = locale.language_code
|
||||
elif hasattr(locale, "locale"):
|
||||
locale = locale.locale
|
||||
|
||||
|
||||
if locale is None:
|
||||
locale = configGet("locale")
|
||||
|
||||
|
||||
try:
|
||||
this_dict = jsonLoad(f'{configGet("locale", "locations")}{sep}{locale}.json')
|
||||
except FileNotFoundError:
|
||||
try:
|
||||
this_dict = jsonLoad(f'{configGet("locale", "locations")}{sep}{configGet("locale")}.json')
|
||||
this_dict = jsonLoad(
|
||||
f'{configGet("locale", "locations")}{sep}{configGet("locale")}.json'
|
||||
)
|
||||
except FileNotFoundError:
|
||||
return f'⚠️ Locale in config is invalid: could not get "{key}" in {str(args)} from locale "{locale}"'
|
||||
|
||||
this_key = this_dict
|
||||
for dict_key in args:
|
||||
this_key = this_key[dict_key]
|
||||
|
||||
|
||||
try:
|
||||
return this_key[key]
|
||||
except KeyError:
|
||||
return f'⚠️ Locale in config is invalid: could not get "{key}" in {str(args)} from locale "{locale}"'
|
||||
|
||||
|
||||
def all_locales(key: str, *args: str) -> list:
|
||||
"""Get value of the provided key and path in all available locales
|
||||
|
||||
@ -148,7 +169,7 @@ def all_locales(key: str, *args: str) -> list:
|
||||
|
||||
### Returns:
|
||||
* `list`: List of all values in all locales
|
||||
"""
|
||||
"""
|
||||
|
||||
output = []
|
||||
valid_locales = []
|
||||
@ -156,7 +177,7 @@ def all_locales(key: str, *args: str) -> list:
|
||||
files_locales = listdir(f'{configGet("locale", "locations")}')
|
||||
for entry in files_locales:
|
||||
valid_locales.append(".".join(entry.split(".")[:-1]))
|
||||
|
||||
|
||||
for lc in valid_locales:
|
||||
try:
|
||||
this_dict = jsonLoad(f'{configGet("locale", "locations")}{sep}{lc}.json')
|
||||
@ -166,7 +187,7 @@ def all_locales(key: str, *args: str) -> list:
|
||||
this_key = this_dict
|
||||
for dict_key in args:
|
||||
this_key = this_key[dict_key]
|
||||
|
||||
|
||||
try:
|
||||
output.append(this_key[key])
|
||||
except KeyError:
|
||||
@ -174,6 +195,7 @@ def all_locales(key: str, *args: str) -> list:
|
||||
|
||||
return output
|
||||
|
||||
|
||||
def find_location(query: str) -> dict:
|
||||
"""Find location on geonames.org by query. Search is made with feature classes A and P.
|
||||
|
||||
@ -185,14 +207,22 @@ def find_location(query: str) -> dict:
|
||||
|
||||
### Returns:
|
||||
* `dict`: One instance of geonames response
|
||||
"""
|
||||
"""
|
||||
try:
|
||||
result = (get(f"http://api.geonames.org/searchJSON?q={query}&maxRows=1&countryBias=UA&lang=uk&orderby=relevance&featureClass=P&featureClass=A&username={configGet('username', 'geocoding')}")).json()
|
||||
result = (
|
||||
get(
|
||||
f"http://api.geonames.org/searchJSON?q={query}&maxRows=1&countryBias=UA&lang=uk&orderby=relevance&featureClass=P&featureClass=A&username={configGet('username', 'geocoding')}"
|
||||
)
|
||||
).json()
|
||||
return result["geonames"][0]
|
||||
except (ValueError, KeyError, IndexError):
|
||||
raise PlaceNotFoundError(query)
|
||||
|
||||
def create_tmp(bytedata: Union[bytes, bytearray], kind: Union[Literal["image", "video"], None] = None) -> str:
|
||||
|
||||
def create_tmp(
|
||||
bytedata: Union[bytes, bytearray],
|
||||
kind: Union[Literal["image", "video"], None] = None,
|
||||
) -> str:
|
||||
"""Create temporary file to help uploading it
|
||||
|
||||
### Args:
|
||||
@ -201,7 +231,7 @@ def create_tmp(bytedata: Union[bytes, bytearray], kind: Union[Literal["image", "
|
||||
|
||||
### Returns:
|
||||
* `str`: Path to temporary file
|
||||
"""
|
||||
"""
|
||||
filename = str(uuid1())
|
||||
if kind == "image":
|
||||
filename += ".jpg"
|
||||
@ -212,6 +242,7 @@ def create_tmp(bytedata: Union[bytes, bytearray], kind: Union[Literal["image", "
|
||||
file.write(bytedata)
|
||||
return path.join("tmp", filename)
|
||||
|
||||
|
||||
async def download_tmp(app: Client, file_id: str) -> Tuple[str, bytes]:
|
||||
"""Download file by its ID and return its bytes
|
||||
|
||||
@ -221,7 +252,7 @@ async def download_tmp(app: Client, file_id: str) -> Tuple[str, bytes]:
|
||||
|
||||
### Returns:
|
||||
* `Tuple[str, bytes]`: First is a filepath and the second is file's bytes
|
||||
"""
|
||||
"""
|
||||
filename = str(uuid1())
|
||||
makedirs("tmp", exist_ok=True)
|
||||
await app.download_media(file_id, path.join("tmp", filename))
|
||||
@ -229,24 +260,31 @@ async def download_tmp(app: Client, file_id: str) -> Tuple[str, bytes]:
|
||||
bytedata = f.read()
|
||||
return path.join("tmp", filename), bytedata
|
||||
|
||||
|
||||
try:
|
||||
from psutil import Process
|
||||
except ModuleNotFoundError:
|
||||
# print(locale("deps_missing", "console", locale=configGet("locale")), flush=True)
|
||||
print("Missing dependencies! Please install all needed dependencies and run the bot again!")
|
||||
print(
|
||||
"Missing dependencies! Please install all needed dependencies and run the bot again!"
|
||||
)
|
||||
exit()
|
||||
|
||||
|
||||
def killProc(pid):
|
||||
if osname == "posix":
|
||||
from signal import SIGKILL
|
||||
|
||||
kill(pid, SIGKILL)
|
||||
else:
|
||||
p = Process(pid)
|
||||
p.kill()
|
||||
|
||||
|
||||
def should_quote(msg):
|
||||
return True if msg.chat.type is not ChatType.PRIVATE else False
|
||||
|
||||
|
||||
async def find_user(app: Client, query: Union[str, int]):
|
||||
try:
|
||||
result = await app.get_users(int(query))
|
||||
@ -261,4 +299,4 @@ async def find_user(app: Client, query: Union[str, int]):
|
||||
return None
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
return None
|
||||
return result
|
||||
return result
|
||||
|
Reference in New Issue
Block a user