165 lines
6.5 KiB
Python
165 lines
6.5 KiB
Python
from datetime import datetime
|
||
from app import app, isAnAdmin
|
||
from pyrogram.types import (
|
||
ChatPermissions,
|
||
InlineKeyboardMarkup,
|
||
InlineKeyboardButton,
|
||
ChatMemberUpdated,
|
||
)
|
||
from pyrogram.client import Client
|
||
from modules.utils import configGet, locale
|
||
from modules import custom_filters
|
||
from modules.logging import logWrite
|
||
from modules.database import col_applications
|
||
from classes.holo_user import HoloUser
|
||
from dateutil.relativedelta import relativedelta
|
||
|
||
|
||
@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}"
|
||
)
|
||
|
||
application = col_applications.find_one({"user": holo_user.id})
|
||
application_content = []
|
||
i = 1
|
||
|
||
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} р.)"
|
||
)
|
||
elif i == 3:
|
||
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']})"
|
||
)
|
||
else:
|
||
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),
|
||
),
|
||
)
|
||
|
||
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}"
|
||
)
|
||
|
||
application = col_applications.find_one({"user": holo_user.id})
|
||
|
||
if application is None:
|
||
return
|
||
|
||
application_content = []
|
||
i = 1
|
||
|
||
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} р.)"
|
||
)
|
||
elif i == 3:
|
||
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']})"
|
||
)
|
||
else:
|
||
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),
|
||
),
|
||
)
|
||
|
||
return
|
||
|
||
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(
|
||
[
|
||
[
|
||
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(
|
||
can_send_messages=False,
|
||
can_send_media_messages=False,
|
||
can_send_other_messages=False,
|
||
can_send_polls=False,
|
||
),
|
||
)
|
||
return
|
||
|
||
if member.new_chat_member is None:
|
||
await app.send_message(
|
||
configGet("users", "groups"),
|
||
locale("user_left", "message").format(
|
||
member.old_chat_member.user.first_name
|
||
),
|
||
)
|
||
logWrite(
|
||
f"User {member.old_chat_member.user.first_name} ({member.old_chat_member.user.id}) left the destination group"
|
||
)
|
||
return
|