Inline implemented
This commit is contained in:
parent
2619ecd408
commit
cfc8167663
@ -64,7 +64,7 @@ After all of that you're good to go! Happy using :)
|
|||||||
## To-Do
|
## To-Do
|
||||||
|
|
||||||
* [x] Check if link belongs to admin
|
* [x] Check if link belongs to admin
|
||||||
* [ ] Inline query user's application
|
* [x] Inline query user's application
|
||||||
* [x] Get applications .json
|
* [x] Get applications .json
|
||||||
* [ ] Get application by id and user_id
|
* [ ] Get application by id and user_id
|
||||||
* [x] Age as a DD.MM.YYYY
|
* [x] Age as a DD.MM.YYYY
|
||||||
|
12
app.py
Normal file
12
app.py
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
from modules.utils import configGet
|
||||||
|
from pyrogram.client import Client
|
||||||
|
|
||||||
|
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):
|
||||||
|
if admin_id == configGet("owner") or admin_id in configGet("admins"):
|
||||||
|
return True
|
||||||
|
async for member in app.get_chat_members(configGet("admin_group")):
|
||||||
|
if member.user.id == admin_id:
|
||||||
|
return True
|
||||||
|
return False
|
@ -124,5 +124,22 @@
|
|||||||
"sus_allowed": "✅ Доступ {0} дозволено",
|
"sus_allowed": "✅ Доступ {0} дозволено",
|
||||||
"sus_refused": "❌ Доступ {0} заборонено",
|
"sus_refused": "❌ Доступ {0} заборонено",
|
||||||
"nothing": "🔔 Дія вже виконана"
|
"nothing": "🔔 Дія вже виконана"
|
||||||
|
},
|
||||||
|
"inline": {
|
||||||
|
"forbidden": {
|
||||||
|
"title": "Дія недоступна",
|
||||||
|
"description": "Ти не маєш дозволу переглядати це",
|
||||||
|
"message_content": "Немає дозволу на перегляд."
|
||||||
|
},
|
||||||
|
"not_pm": {
|
||||||
|
"title": "Дія недоступна",
|
||||||
|
"description": "Ця команда доступна тільки в приватних повідомленнях в чаті з самим ботом",
|
||||||
|
"message_content": "Дія доступна тільки в ПП."
|
||||||
|
},
|
||||||
|
"user": {
|
||||||
|
"title": "",
|
||||||
|
"description": "Переглянути анкету {0} (@{1})",
|
||||||
|
"message_content": "{0} (@{1})\n\n**Дані анкети:**\n{2}"
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
114
main.py
114
main.py
@ -5,9 +5,10 @@ from time import time
|
|||||||
from os import getpid, path
|
from os import getpid, path
|
||||||
from modules.birthdays import check_birthdays
|
from modules.birthdays import check_birthdays
|
||||||
from modules.utils import *
|
from modules.utils import *
|
||||||
|
from modules.inline import *
|
||||||
from schedule import run_pending, every
|
from schedule import run_pending, every
|
||||||
|
from app import app
|
||||||
|
|
||||||
from pyrogram.client import Client
|
|
||||||
from pyrogram import filters
|
from pyrogram import filters
|
||||||
from pyrogram.enums.parse_mode import ParseMode
|
from pyrogram.enums.parse_mode import ParseMode
|
||||||
from pyrogram.enums.chat_action import ChatAction
|
from pyrogram.enums.chat_action import ChatAction
|
||||||
@ -17,7 +18,6 @@ from pyrogram.errors.exceptions import bad_request_400
|
|||||||
|
|
||||||
|
|
||||||
pid = getpid()
|
pid = getpid()
|
||||||
app = Client("holochecker", bot_token=configGet("bot_token", "bot"), api_id=configGet("api_id", "bot"), api_hash=configGet("api_hash", "bot"))
|
|
||||||
|
|
||||||
|
|
||||||
for entry in [f"{configGet('data', 'locations')}{sep}applications.json"]:
|
for entry in [f"{configGet('data', 'locations')}{sep}applications.json"]:
|
||||||
@ -28,16 +28,6 @@ for entry in [f"{configGet('data', 'locations')}{sep}applications.json"]:
|
|||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
async def isAnAdmin(admin_id):
|
|
||||||
if admin_id == configGet("owner") or admin_id in configGet("admins"):
|
|
||||||
return True
|
|
||||||
async for member in app.get_chat_members(configGet("admin_group")):
|
|
||||||
if member.user.id == admin_id:
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
# Start command ================================================================================================================
|
# Start command ================================================================================================================
|
||||||
@app.on_message(~ filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"]))
|
@app.on_message(~ filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"]))
|
||||||
async def cmd_start(app, msg):
|
async def cmd_start(app, msg):
|
||||||
@ -472,63 +462,65 @@ async def get_contact(app, msg):
|
|||||||
@app.on_message(~ filters.scheduled & filters.private)
|
@app.on_message(~ filters.scheduled & filters.private)
|
||||||
async def any_stage(app, msg):
|
async def any_stage(app, msg):
|
||||||
|
|
||||||
user_stage = configGet("stage", file=str(msg.from_user.id))
|
if msg.via_bot == None:
|
||||||
|
|
||||||
if user_stage == 1:
|
|
||||||
await msg.reply_text(locale(f"question{user_stage+1}", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage+1}", "force_reply"))))
|
|
||||||
logWrite(f"User {msg.from_user.id} completed stage {user_stage} of application")
|
|
||||||
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
|
||||||
configSet("stage", user_stage+1, file=str(msg.from_user.id))
|
|
||||||
|
|
||||||
elif user_stage == 2:
|
user_stage = configGet("stage", file=str(msg.from_user.id))
|
||||||
|
|
||||||
try:
|
if user_stage == 1:
|
||||||
|
|
||||||
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
|
||||||
|
|
||||||
input_dt = datetime.strptime(msg.text, "%d.%m.%Y")
|
|
||||||
|
|
||||||
if datetime.now() <= input_dt:
|
|
||||||
logWrite(f"User {msg.from_user.id} failed stage {user_stage} due to joking")
|
|
||||||
await msg.reply_text(locale("question2_joke", "message"), reply_markup=ForceReply(placeholder=str(locale("question2", "force_reply"))))
|
|
||||||
|
|
||||||
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 {user_stage} due to being underage")
|
|
||||||
await msg.reply_text(locale("question2_underage", "message").format(str(configGet("age_allowed"))), reply_markup=ForceReply(placeholder=str(locale("question2", "force_reply")))) # type: ignore
|
|
||||||
|
|
||||||
else:
|
|
||||||
logWrite(f"User {msg.from_user.id} completed stage {user_stage} of application")
|
|
||||||
await msg.reply_text(locale(f"question{user_stage+1}", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage+1}", "force_reply"))))
|
|
||||||
configSet("stage", user_stage+1, file=str(msg.from_user.id))
|
|
||||||
|
|
||||||
except ValueError:
|
|
||||||
logWrite(f"User {msg.from_user.id} failed stage {user_stage} due to sending invalid date format")
|
|
||||||
await msg.reply_text(locale(f"question2_invalid", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage}", "force_reply"))))
|
|
||||||
|
|
||||||
else:
|
|
||||||
if user_stage <= 9:
|
|
||||||
logWrite(f"User {msg.from_user.id} completed stage {user_stage} of application")
|
|
||||||
await msg.reply_text(locale(f"question{user_stage+1}", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage+1}", "force_reply"))))
|
await msg.reply_text(locale(f"question{user_stage+1}", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage+1}", "force_reply"))))
|
||||||
|
logWrite(f"User {msg.from_user.id} completed stage {user_stage} of application")
|
||||||
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
||||||
configSet("stage", user_stage+1, file=str(msg.from_user.id))
|
configSet("stage", user_stage+1, file=str(msg.from_user.id))
|
||||||
|
|
||||||
|
elif user_stage == 2:
|
||||||
|
|
||||||
|
try:
|
||||||
|
|
||||||
|
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
||||||
|
|
||||||
|
input_dt = datetime.strptime(msg.text, "%d.%m.%Y")
|
||||||
|
|
||||||
|
if datetime.now() <= input_dt:
|
||||||
|
logWrite(f"User {msg.from_user.id} failed stage {user_stage} due to joking")
|
||||||
|
await msg.reply_text(locale("question2_joke", "message"), reply_markup=ForceReply(placeholder=str(locale("question2", "force_reply"))))
|
||||||
|
|
||||||
|
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 {user_stage} due to being underage")
|
||||||
|
await msg.reply_text(locale("question2_underage", "message").format(str(configGet("age_allowed"))), reply_markup=ForceReply(placeholder=str(locale("question2", "force_reply")))) # type: ignore
|
||||||
|
|
||||||
|
else:
|
||||||
|
logWrite(f"User {msg.from_user.id} completed stage {user_stage} of application")
|
||||||
|
await msg.reply_text(locale(f"question{user_stage+1}", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage+1}", "force_reply"))))
|
||||||
|
configSet("stage", user_stage+1, file=str(msg.from_user.id))
|
||||||
|
|
||||||
|
except ValueError:
|
||||||
|
logWrite(f"User {msg.from_user.id} failed stage {user_stage} due to sending invalid date format")
|
||||||
|
await msg.reply_text(locale(f"question2_invalid", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage}", "force_reply"))))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
if not configGet("sent", file=str(msg.from_user.id)):
|
if user_stage <= 9:
|
||||||
if not configGet("confirmed", file=str(msg.from_user.id)):
|
logWrite(f"User {msg.from_user.id} completed stage {user_stage} of application")
|
||||||
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
await msg.reply_text(locale(f"question{user_stage+1}", "message"), reply_markup=ForceReply(placeholder=str(locale(f"question{user_stage+1}", "force_reply"))))
|
||||||
application_content = []
|
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
||||||
i = 1
|
configSet("stage", user_stage+1, file=str(msg.from_user.id))
|
||||||
for question in configGet("application", file=str(msg.from_user.id)):
|
else:
|
||||||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles')} {configGet('application', file=str(msg.from_user.id))[question]}")
|
if not configGet("sent", file=str(msg.from_user.id)):
|
||||||
i += 1
|
if not configGet("confirmed", file=str(msg.from_user.id)):
|
||||||
await msg.reply_text(locale("confirm", "message").format("\n".join(application_content)), reply_markup=ReplyKeyboardMarkup(locale("confirm", "keyboard"), resize_keyboard=True)) # type: ignore
|
configSet(str(user_stage), str(msg.text), "application", file=str(msg.from_user.id))
|
||||||
#configSet("sent", True, file=str(msg.from_user.id))
|
application_content = []
|
||||||
#configSet("application_date", int(time()), file=str(msg.from_user.id))
|
i = 1
|
||||||
|
for question in configGet("application", file=str(msg.from_user.id)):
|
||||||
|
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles')} {configGet('application', file=str(msg.from_user.id))[question]}")
|
||||||
|
i += 1
|
||||||
|
await msg.reply_text(locale("confirm", "message").format("\n".join(application_content)), reply_markup=ReplyKeyboardMarkup(locale("confirm", "keyboard"), resize_keyboard=True)) # type: ignore
|
||||||
|
#configSet("sent", True, file=str(msg.from_user.id))
|
||||||
|
#configSet("application_date", int(time()), file=str(msg.from_user.id))
|
||||||
|
else:
|
||||||
|
if not configGet("approved", file=str(msg.from_user.id)) and not configGet("refused", file=str(msg.from_user.id)):
|
||||||
|
await msg.reply_text(locale("already_sent", "message"))
|
||||||
else:
|
else:
|
||||||
if not configGet("approved", file=str(msg.from_user.id)) and not configGet("refused", file=str(msg.from_user.id)):
|
if not configGet("approved", file=str(msg.from_user.id)) and not configGet("refused", file=str(msg.from_user.id)):
|
||||||
await msg.reply_text(locale("already_sent", "message"))
|
await msg.reply_text(locale("already_sent", "message"))
|
||||||
else:
|
|
||||||
if not configGet("approved", file=str(msg.from_user.id)) and not configGet("refused", file=str(msg.from_user.id)):
|
|
||||||
await msg.reply_text(locale("already_sent", "message"))
|
|
||||||
# ==============================================================================================================================
|
# ==============================================================================================================================
|
||||||
|
|
||||||
|
|
||||||
|
136
modules/inline.py
Normal file
136
modules/inline.py
Normal file
@ -0,0 +1,136 @@
|
|||||||
|
from datetime import datetime
|
||||||
|
from os import path, sep
|
||||||
|
from pyrogram.types import InlineQueryResultArticle, InputTextMessageContent
|
||||||
|
from pyrogram.enums.chat_type import ChatType
|
||||||
|
from pyrogram.enums.chat_members_filter import ChatMembersFilter
|
||||||
|
from dateutil.relativedelta import relativedelta
|
||||||
|
|
||||||
|
from app import app, isAnAdmin
|
||||||
|
from modules.utils import configGet, jsonLoad, locale
|
||||||
|
|
||||||
|
@app.on_inline_query()
|
||||||
|
async def inline_answer(client, inline_query):
|
||||||
|
|
||||||
|
if inline_query.chat_type not in [ChatType.BOT, ChatType.PRIVATE]:
|
||||||
|
await inline_query.answer(
|
||||||
|
results=[
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
title=locale("title", "inline", "not_pm"),
|
||||||
|
input_message_content=InputTextMessageContent(
|
||||||
|
locale("message_content", "inline", "not_pm")
|
||||||
|
),
|
||||||
|
description=locale("description", "inline", "not_pm")
|
||||||
|
)
|
||||||
|
]
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
if (path.exists(f"{configGet('data', 'locations')}{sep}users{sep}{inline_query.from_user.id}.json") and jsonLoad(f"{configGet('data', 'locations')}{sep}users{sep}{inline_query.from_user.id}.json")["approved"]) or (await isAnAdmin(inline_query.from_user.id)):
|
||||||
|
|
||||||
|
list_of_users = []
|
||||||
|
async for m in app.get_chat_members(configGet("admin_group"), filter=ChatMembersFilter.SEARCH, query=inline_query.query):
|
||||||
|
list_of_users.append(m)
|
||||||
|
|
||||||
|
#list_of_names = []
|
||||||
|
|
||||||
|
#list_filtered = [v for v in (list_of_users) if inline_query.query in v]
|
||||||
|
|
||||||
|
results = []
|
||||||
|
|
||||||
|
for match in list_of_users:
|
||||||
|
# if match.user.photo != None:
|
||||||
|
# results.append(
|
||||||
|
# InlineQueryResultArticle(
|
||||||
|
# title=str(match.user.first_name),
|
||||||
|
# input_message_content=InputTextMessageContent(
|
||||||
|
# f"More about @{match.user.username}"
|
||||||
|
# ),
|
||||||
|
# description=f"More info about {match.user.username}",
|
||||||
|
# thumb_url=f'https://api.telegram.org/bot{configGet("bot_token", "bot")}/{match.user.photo.big_file_id}',
|
||||||
|
# thumb_height=640,
|
||||||
|
# thumb_width=640
|
||||||
|
# )
|
||||||
|
# )
|
||||||
|
# else:
|
||||||
|
|
||||||
|
try:
|
||||||
|
user_data = jsonLoad(f"{configGet('data', 'locations')}{sep}users{sep}{match.user.id}.json")
|
||||||
|
application_content = []
|
||||||
|
i = 1
|
||||||
|
for question in configGet("application", file=str(match.user.id)):
|
||||||
|
if i == 2:
|
||||||
|
age = relativedelta(datetime.now(), datetime.strptime(configGet('application', file=str(match.user.id))['2'], '%d.%m.%Y'))
|
||||||
|
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles')} {configGet('application', file=str(match.user.id))['2']} ({age.years} р.)")
|
||||||
|
else:
|
||||||
|
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles')} {configGet('application', file=str(match.user.id))[question]}")
|
||||||
|
i += 1
|
||||||
|
results.append(
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
title=str(match.user.first_name),
|
||||||
|
input_message_content=InputTextMessageContent(
|
||||||
|
locale("message_content", "inline", "user").format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||||||
|
),
|
||||||
|
description=locale("description", "inline", "user").format(match.user.first_name, match.user.username)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
except FileNotFoundError:
|
||||||
|
pass
|
||||||
|
except TypeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
await inline_query.answer(
|
||||||
|
results=results,
|
||||||
|
cache_time=10
|
||||||
|
)
|
||||||
|
|
||||||
|
# await inline_query.answer(
|
||||||
|
# results=[
|
||||||
|
# InlineQueryResultArticle(
|
||||||
|
# title="Installation",
|
||||||
|
# input_message_content=InputTextMessageContent(
|
||||||
|
# "Here's how to install **Pyrogram**"
|
||||||
|
# ),
|
||||||
|
# url="https://docs.pyrogram.org/intro/install",
|
||||||
|
# description="How to install Pyrogram",
|
||||||
|
# reply_markup=InlineKeyboardMarkup(
|
||||||
|
# [
|
||||||
|
# [InlineKeyboardButton(
|
||||||
|
# "Open website",
|
||||||
|
# url="https://docs.pyrogram.org/intro/install"
|
||||||
|
# )]
|
||||||
|
# ]
|
||||||
|
# )
|
||||||
|
# ),
|
||||||
|
# InlineQueryResultArticle(
|
||||||
|
# title="Usage",
|
||||||
|
# input_message_content=InputTextMessageContent(
|
||||||
|
# "Here's how to use **Pyrogram**"
|
||||||
|
# ),
|
||||||
|
# url="https://docs.pyrogram.org/start/invoking",
|
||||||
|
# description="How to use Pyrogram",
|
||||||
|
# reply_markup=InlineKeyboardMarkup(
|
||||||
|
# [
|
||||||
|
# [InlineKeyboardButton(
|
||||||
|
# "Open website",
|
||||||
|
# url="https://docs.pyrogram.org/start/invoking"
|
||||||
|
# )]
|
||||||
|
# ]
|
||||||
|
# )
|
||||||
|
# )
|
||||||
|
# ],
|
||||||
|
# cache_time=1
|
||||||
|
# )
|
||||||
|
|
||||||
|
else:
|
||||||
|
await inline_query.answer(
|
||||||
|
results=[
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
title=locale("title", "inline", "forbidden"),
|
||||||
|
input_message_content=InputTextMessageContent(
|
||||||
|
locale("message_content", "inline", "forbidden")
|
||||||
|
),
|
||||||
|
description=locale("description", "inline", "forbidden")
|
||||||
|
)
|
||||||
|
]
|
||||||
|
)
|
||||||
|
return
|
@ -44,15 +44,15 @@ def configSet(key: str, value, *args: str, file: str = "config"):
|
|||||||
"""
|
"""
|
||||||
if file == "config":
|
if file == "config":
|
||||||
filepath = ""
|
filepath = ""
|
||||||
|
if this_dict["debug"]:
|
||||||
|
try:
|
||||||
|
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)
|
||||||
else:
|
else:
|
||||||
filepath = f"data{sep}users{sep}"
|
filepath = f"data{sep}users{sep}"
|
||||||
this_dict = jsonLoad(f"{filepath}{file}.json")
|
this_dict = jsonLoad(f"{filepath}{file}.json")
|
||||||
if this_dict["debug"]:
|
|
||||||
try:
|
|
||||||
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)
|
|
||||||
string = "this_dict"
|
string = "this_dict"
|
||||||
for arg in args:
|
for arg in args:
|
||||||
string += f'["{arg}"]'
|
string += f'["{arg}"]'
|
||||||
@ -79,11 +79,11 @@ def configGet(key: str, *args: str, file: str = "config"):
|
|||||||
except FileNotFoundError:
|
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()
|
exit()
|
||||||
if this_dict["debug"]:
|
if this_dict["debug"]:
|
||||||
try:
|
try:
|
||||||
this_dict = jsonLoad("config_debug.json")
|
this_dict = jsonLoad("config_debug.json")
|
||||||
except FileNotFoundError:
|
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:
|
else:
|
||||||
this_dict = jsonLoad(f"data{sep}users{sep}{file}.json")
|
this_dict = jsonLoad(f"data{sep}users{sep}{file}.json")
|
||||||
this_key = this_dict
|
this_key = this_dict
|
||||||
|
Reference in New Issue
Block a user