mirror of
https://github.com/MuRuLOSE/limoka.git
synced 2026-06-18 23:24:18 +02:00
Added and updated repositories 2025-08-10 01:17:09
This commit is contained in:
github-actions[bot]
345
mead0wsss/mead0wsMods/InfoPresets.py
Normal file
345
mead0wsss/mead0wsMods/InfoPresets.py
Normal file
@@ -0,0 +1,345 @@
|
||||
__version__ = (1, 1, 0)
|
||||
|
||||
# ███╗░░░███╗███████╗░█████╗░██████╗░░█████╗░░██╗░░░░░░░██╗░██████╗░██████╗
|
||||
# ████╗░████║██╔════╝██╔══██╗██╔══██╗██╔══██╗░██║░░██╗░░██║██╔════╝██╔════╝
|
||||
# ██╔████╔██║█████╗░░███████║██║░░██║██║░░██║░╚██╗████╗██╔╝╚█████╗░╚█████╗░
|
||||
# ██║╚██╔╝██║██╔══╝░░██╔══██║██║░░██║██║░░██║░░████╔═████║░░╚═══██╗░╚═══██╗
|
||||
# ██║░╚═╝░██║███████╗██║░░██║██████╔╝╚█████╔╝░░╚██╔╝░╚██╔╝░██████╔╝██████╔╝
|
||||
# ╚═╝░░░░░╚═╝╚══════╝╚═╝░░╚═╝╚═════╝░░╚════╝░░░░╚═╝░░░╚═╝░░╚═════╝░╚═════╝░
|
||||
# © Copyright 2025
|
||||
# ✈ https://t.me/mead0wssMods
|
||||
|
||||
# scope: heroku_only
|
||||
# meta developer: @mead0wssMods
|
||||
# meta banner: https://x0.at/GHOP.png
|
||||
|
||||
import json
|
||||
import os
|
||||
from telethon.tl.types import Message
|
||||
from ..inline.types import InlineCall
|
||||
from .. import loader, utils
|
||||
import logging
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@loader.tds
|
||||
class InfoPresets(loader.Module):
|
||||
"""Управление пресетами для HerokuInfo"""
|
||||
strings = {
|
||||
"name": "InfoPresets",
|
||||
"preset_exists": "🚫 Пресет с таким именем уже существует!",
|
||||
"preset_created": "✅ Пресет '{}' создан. Теперь настройте параметры.",
|
||||
"file_created": "✅ Файл InfoPresets.json создан",
|
||||
"param_set": "✅ Параметр '{}' установлен в '{}' для пресета '{}'",
|
||||
"preset_not_found": "🚫 Пресет '{}' не найден!",
|
||||
"preset_deleted": "✅ Пресет '{}' удален",
|
||||
"no_presets": "🚫 Нет сохраненных пресетов",
|
||||
"preset_loaded": "✅ Пресет '{}' загружен",
|
||||
"enter_value": "✍️ Введите значение для параметра '{}':",
|
||||
"invalid_bool": "🚫 Значение должно быть True или False",
|
||||
"param_not_set": "🚫 Параметр '{}' не установлен в пресете '{}'",
|
||||
"config_menu": "⚙️ Настройка пресета '{}'\nВыберите параметр:",
|
||||
"file_deleted": "✅ Файл с пресетами удален",
|
||||
"file_not_found": "🚫 Файл с пресетами не найден",
|
||||
"preset_list": "📋 Список пресетов:\n\n{}",
|
||||
"preset_info": "🔹 {}:\n{}",
|
||||
"param_info": " • {}: {}",
|
||||
"done": "✅ Готово",
|
||||
"cancel": "❌ Отмена",
|
||||
"form_expired": "⏳ Время действия формы истекло, создайте новую"
|
||||
}
|
||||
|
||||
async def client_ready(self, client, db):
|
||||
self._client = client
|
||||
self.db = db
|
||||
self.presets_file = "InfoPresets.json"
|
||||
self.ensure_presets_file()
|
||||
self._waiting_param = {}
|
||||
self._active_forms = {}
|
||||
|
||||
def ensure_presets_file(self):
|
||||
if not os.path.exists(self.presets_file):
|
||||
with open(self.presets_file, "w", encoding="utf-8") as f:
|
||||
json.dump({}, f)
|
||||
|
||||
async def createprcmd(self, message: Message):
|
||||
"""Создать новый пресет."""
|
||||
args = utils.get_args_raw(message)
|
||||
if not args:
|
||||
return
|
||||
|
||||
with open(self.presets_file, "r+", encoding="utf-8") as f:
|
||||
try:
|
||||
presets = json.load(f)
|
||||
except json.JSONDecodeError:
|
||||
presets = {}
|
||||
|
||||
if args in presets:
|
||||
await utils.answer(message, self.strings["preset_exists"])
|
||||
return
|
||||
|
||||
presets[args] = {}
|
||||
f.seek(0)
|
||||
json.dump(presets, f, indent=4)
|
||||
f.truncate()
|
||||
|
||||
await self.edit_preset(message, args)
|
||||
|
||||
async def edit_preset(self, message: Message, preset_name: str):
|
||||
"""Редактирование пресета с инлайн-кнопками"""
|
||||
buttons = [
|
||||
[
|
||||
{"text": "✏️ custom_message", "callback": self._param_callback, "args": (preset_name, "custom_message")},
|
||||
{"text": "🖼️ pp_to_banner", "callback": self._param_callback, "args": (preset_name, "pp_to_banner")}
|
||||
],
|
||||
[
|
||||
{"text": "🔗 banner_url", "callback": self._param_callback, "args": (preset_name, "banner_url")},
|
||||
{"text": "⚙️ show_heroku", "callback": self._param_callback, "args": (preset_name, "show_heroku")}
|
||||
],
|
||||
[
|
||||
{"text": self.strings["done"], "callback": self._done_callback, "args": (preset_name,)}
|
||||
]
|
||||
]
|
||||
|
||||
form = await self.inline.form(
|
||||
message=message,
|
||||
text=self.strings["config_menu"].format(preset_name),
|
||||
reply_markup=buttons,
|
||||
silent=True
|
||||
)
|
||||
|
||||
self._active_forms[preset_name] = {
|
||||
"form": form,
|
||||
"chat_id": message.chat_id,
|
||||
"user_id": message.sender_id
|
||||
}
|
||||
|
||||
async def _param_callback(self, call: InlineCall, preset_name: str, param: str):
|
||||
"""Обработчик выбора параметра"""
|
||||
if preset_name not in self._active_forms:
|
||||
await call.answer(self.strings["form_expired"])
|
||||
return
|
||||
|
||||
form_info = self._active_forms[preset_name]
|
||||
|
||||
await call.edit(
|
||||
self.strings["enter_value"].format(param),
|
||||
reply_markup=[
|
||||
[{"text": self.strings["cancel"], "callback": self._cancel_callback, "args": (preset_name,)}]
|
||||
]
|
||||
)
|
||||
|
||||
self._waiting_param = {
|
||||
"user_id": call.from_user.id,
|
||||
"chat_id": form_info["chat_id"],
|
||||
"preset_name": preset_name,
|
||||
"param": param,
|
||||
"form_info": form_info
|
||||
}
|
||||
|
||||
async def _cancel_callback(self, call: InlineCall, preset_name: str):
|
||||
"""Обработчик отмены"""
|
||||
if preset_name not in self._active_forms:
|
||||
await call.answer(self.strings["form_expired"])
|
||||
return
|
||||
|
||||
form_info = self._active_forms[preset_name]
|
||||
|
||||
try:
|
||||
await form_info["form"].edit(
|
||||
self.strings["config_menu"].format(preset_name),
|
||||
reply_markup=[
|
||||
[
|
||||
{"text": "✏️ custom_message", "callback": self._param_callback, "args": (preset_name, "custom_message")},
|
||||
{"text": "🖼️ pp_to_banner", "callback": self._param_callback, "args": (preset_name, "pp_to_banner")}
|
||||
],
|
||||
[
|
||||
{"text": "🔗 banner_url", "callback": self._param_callback, "args": (preset_name, "banner_url")},
|
||||
{"text": "⚙️ show_heroku", "callback": self._param_callback, "args": (preset_name, "show_heroku")}
|
||||
],
|
||||
[
|
||||
{"text": self.strings["done"], "callback": self._done_callback, "args": (preset_name,)}
|
||||
]
|
||||
]
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to edit form on cancel: {e}")
|
||||
|
||||
self._waiting_param = {}
|
||||
|
||||
async def _done_callback(self, call: InlineCall, preset_name: str):
|
||||
"""Обработчик завершения"""
|
||||
if preset_name in self._active_forms:
|
||||
try:
|
||||
await call.delete()
|
||||
except:
|
||||
pass
|
||||
del self._active_forms[preset_name]
|
||||
self._waiting_param = {}
|
||||
|
||||
async def watcher(self, message: Message):
|
||||
"""Обработчик ввода значений параметров"""
|
||||
if not self._waiting_param or not isinstance(self._waiting_param, dict):
|
||||
return
|
||||
|
||||
if not isinstance(message, Message) or not message.message or not hasattr(message, "raw_text"):
|
||||
return
|
||||
|
||||
waiting_chat_id = self._waiting_param.get("chat_id")
|
||||
waiting_user_id = self._waiting_param.get("user_id")
|
||||
|
||||
if (not waiting_chat_id or not waiting_user_id or
|
||||
message.chat_id != waiting_chat_id or
|
||||
message.sender_id != waiting_user_id):
|
||||
return
|
||||
|
||||
preset_name = self._waiting_param.get("preset_name")
|
||||
param = self._waiting_param.get("param")
|
||||
form_info = self._waiting_param.get("form_info")
|
||||
|
||||
if not all([preset_name, param, form_info]):
|
||||
self._waiting_param = {}
|
||||
return
|
||||
|
||||
value = message.raw_text.strip()
|
||||
|
||||
if param in ["pp_to_banner", "show_heroku"]:
|
||||
if value.lower() not in ["true", "false"]:
|
||||
return
|
||||
value = value.lower() == "true"
|
||||
|
||||
try:
|
||||
with open(self.presets_file, "r+", encoding="utf-8") as f:
|
||||
presets = json.load(f)
|
||||
if preset_name not in presets:
|
||||
return
|
||||
|
||||
presets[preset_name][param] = value
|
||||
f.seek(0)
|
||||
json.dump(presets, f, indent=4)
|
||||
f.truncate()
|
||||
|
||||
await utils.answer(message, self.strings["param_set"].format(param, value, preset_name))
|
||||
|
||||
try:
|
||||
await form_info["form"].edit(
|
||||
self.strings["config_menu"].format(preset_name),
|
||||
reply_markup=[
|
||||
[
|
||||
{"text": "✏️ custom_message", "callback": self._param_callback, "args": (preset_name, "custom_message")},
|
||||
{"text": "🖼️ pp_to_banner", "callback": self._param_callback, "args": (preset_name, "pp_to_banner")}
|
||||
],
|
||||
[
|
||||
{"text": "🔗 banner_url", "callback": self._param_callback, "args": (preset_name, "banner_url")},
|
||||
{"text": "⚙️ show_heroku", "callback": self._param_callback, "args": (preset_name, "show_heroku")}
|
||||
],
|
||||
[
|
||||
{"text": self.strings["done"], "callback": self._done_callback, "args": (preset_name,)}
|
||||
]
|
||||
]
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to edit form: {e}")
|
||||
|
||||
except Exception as e:
|
||||
logger.exception("Error saving parameter")
|
||||
|
||||
finally:
|
||||
self._waiting_param = {}
|
||||
|
||||
async def delprcmd(self, message: Message):
|
||||
"""Удалить пресет."""
|
||||
args = utils.get_args_raw(message)
|
||||
if not args:
|
||||
return
|
||||
|
||||
with open(self.presets_file, "r+", encoding="utf-8") as f:
|
||||
presets = json.load(f)
|
||||
if args not in presets:
|
||||
return
|
||||
|
||||
del presets[args]
|
||||
f.seek(0)
|
||||
json.dump(presets, f, indent=4)
|
||||
f.truncate()
|
||||
|
||||
await utils.answer(message, self.strings["preset_deleted"].format(args))
|
||||
|
||||
async def delfileprcmd(self, message: Message):
|
||||
"""Удалить файл с пресетами."""
|
||||
if not os.path.exists(self.presets_file):
|
||||
return
|
||||
|
||||
os.remove(self.presets_file)
|
||||
self.ensure_presets_file()
|
||||
await utils.answer(message, self.strings["file_deleted"])
|
||||
|
||||
async def uploadprcmd(self, message: Message):
|
||||
"""Загрузить файл с пресетами."""
|
||||
if not os.path.exists(self.presets_file):
|
||||
return
|
||||
|
||||
with open(self.presets_file, "r", encoding="utf-8") as f:
|
||||
presets = json.load(f)
|
||||
if not presets:
|
||||
return
|
||||
|
||||
await self._client.send_file(
|
||||
message.chat_id,
|
||||
self.presets_file,
|
||||
caption="📁 Файл с пресетами"
|
||||
)
|
||||
await message.delete()
|
||||
|
||||
async def listprcmd(self, message: Message):
|
||||
"""Показать список всех пресетов."""
|
||||
if not os.path.exists(self.presets_file):
|
||||
return
|
||||
|
||||
with open(self.presets_file, "r", encoding="utf-8") as f:
|
||||
try:
|
||||
presets = json.load(f)
|
||||
except json.JSONDecodeError:
|
||||
return
|
||||
|
||||
if not presets:
|
||||
return
|
||||
|
||||
result = []
|
||||
for preset_name, params in presets.items():
|
||||
param_lines = []
|
||||
for param, value in params.items():
|
||||
param_lines.append(self.strings["param_info"].format(param, value))
|
||||
result.append(self.strings["preset_info"].format(
|
||||
preset_name, "\n".join(param_lines) if param_lines else "⏺ Нет параметров"
|
||||
))
|
||||
|
||||
await utils.answer(
|
||||
message,
|
||||
self.strings["preset_list"].format("\n\n".join(result))
|
||||
)
|
||||
|
||||
async def loadprcmd(self, message: Message):
|
||||
"""Загрузить пресет."""
|
||||
args = utils.get_args_raw(message)
|
||||
if not args:
|
||||
return
|
||||
|
||||
with open(self.presets_file, "r", encoding="utf-8") as f:
|
||||
presets = json.load(f)
|
||||
if args not in presets:
|
||||
return
|
||||
|
||||
preset = presets[args]
|
||||
heroku_info = self.lookup("HerokuInfo")
|
||||
|
||||
if not heroku_info:
|
||||
return
|
||||
|
||||
for param, value in preset.items():
|
||||
if param in heroku_info.config:
|
||||
heroku_info.config[param] = value
|
||||
else:
|
||||
logger.warning(f"Параметр {param} не найден в конфиге HerokuInfo")
|
||||
|
||||
await utils.answer(message, self.strings["preset_loaded"].format(args))
|
||||
Reference in New Issue
Block a user