238 lines
11 KiB
Python
238 lines
11 KiB
Python
import sys
|
|
import asyncio
|
|
import logging
|
|
import os
|
|
import psutil
|
|
import time
|
|
from datetime import datetime
|
|
|
|
# --- НАСТРОЙКА ЛОГИРОВАНИЯ ---
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s [%(levelname)s] %(name)s - %(message)s",
|
|
handlers=[
|
|
logging.FileHandler("bot_log.txt", encoding="utf-8"),
|
|
logging.StreamHandler(sys.stdout)
|
|
]
|
|
)
|
|
logging.getLogger('trueconf.client.chatbot').propagate = False
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Добавляем путь к проекту в sys.path
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from trueconf import Bot, Dispatcher, Router, Message
|
|
from config.config import *
|
|
from utils.states import set_state, clear_state, get_state
|
|
from utils import states
|
|
|
|
# 🎯 ИСПРАВЛЕНО: Импортируем оба варианта меню (короткий и приветственный)
|
|
from utils.menu import MENU_TEXT, WELCOME_TEXT, MENU_MAP
|
|
|
|
# =========================================================
|
|
# 🔄 МОДУЛЬ АВТОМАТИЧЕСКОГО КОНТРОЛЯ СЕССИЙ И ТАЙМАУТОВ
|
|
# =========================================================
|
|
USER_LAST_ACTIVITY = {}
|
|
USER_MENU_SHOWN = {}
|
|
DEFAULT_TIMEOUT = 1800
|
|
|
|
_orig_get_state = states.get_state
|
|
|
|
def patched_get_state(user_id):
|
|
current_time = time.time()
|
|
last_activity = USER_LAST_ACTIVITY.get(user_id)
|
|
session_timeout = getattr(states.config, 'SESSION_TIMEOUT', DEFAULT_TIMEOUT)
|
|
if last_activity and (current_time - last_activity > session_timeout):
|
|
states.clear_state(user_id)
|
|
USER_MENU_SHOWN[user_id] = False
|
|
logger.info(f"⏳ Сессия пользователя {user_id} сброшена по таймауту бездействия ({session_timeout} сек).")
|
|
USER_LAST_ACTIVITY[user_id] = current_time
|
|
return _orig_get_state(user_id)
|
|
|
|
states.get_state = patched_get_state
|
|
get_state = patched_get_state
|
|
|
|
# =========================================================
|
|
# 🔌 КОРПОРАТИВНЫЕ ШАБЛОНЫ ТЕКСТОВ КЛЕВЕР
|
|
# =========================================================
|
|
from utils.texts import UNKNOWN_MAIN_CMD_TEXT
|
|
from utils.stats_logger import log_menu_stats
|
|
|
|
# =========================================================
|
|
# ФУНКЦИЯ МОНИТОРИНГА ОПЕРАТИВНОЙ ПАМЯТИ
|
|
# =========================================================
|
|
def log_ram_usage(stage_name: str):
|
|
process = psutil.Process(os.getpid())
|
|
ram_mb = process.memory_info().rss / 1024 / 1024
|
|
logger.info(f"📊 [RAM MONITOR] {stage_name}: {ram_mb:.2f} MB")
|
|
|
|
log_ram_usage("Старт скрипта (базовый вес)")
|
|
|
|
from photo_bot.handlers import router as photo_router, PHOTO_MAIN_MENU_TEXT
|
|
log_ram_usage("После импорта photo_bot")
|
|
|
|
from service_desk.handlers import router as sd_router, SD_MAIN_MENU_TEXT
|
|
log_ram_usage("После импорта service_desk")
|
|
|
|
from transcription_bot.handlers import router as transcription_router, TRANSCRIPTION_MAIN_MENU_TEXT
|
|
log_ram_usage("После импорта transcription_router (Виспер)")
|
|
|
|
from search_bot.handlers import router as search_router, SEARCH_MAIN_MENU_TEXT
|
|
log_ram_usage("После импорта search_router")
|
|
|
|
from lk.handlers import router as lk_router
|
|
log_ram_usage("После импорта lk_router (Личный кабинет)")
|
|
|
|
# =========================================================
|
|
# ГЛАВНЫЙ РОУТЕР
|
|
# =========================================================
|
|
main_router = Router()
|
|
|
|
@main_router.message()
|
|
async def main_menu(msg: Message):
|
|
if ENABLE_WHITELIST and msg.from_user.id not in ALLOWED_USERS:
|
|
logger.info(f"Доступ закрыт (Whitelist ON): Игнорирование пользователя {msg.from_user.id}")
|
|
return
|
|
user_id = msg.from_user.id
|
|
if getattr(msg, "handled", False):
|
|
return
|
|
if get_state(user_id) is not None:
|
|
return
|
|
is_attachment = hasattr(msg.type, "name") and msg.type.name == "ATTACHMENT"
|
|
if is_attachment:
|
|
await msg.answer("⚠️ <b>Сначала выберите режим работы из меню.</b>", parse_mode="html")
|
|
await msg.answer(WELCOME_TEXT, parse_mode="html")
|
|
USER_MENU_SHOWN[user_id] = True
|
|
return
|
|
menu_shown = USER_MENU_SHOWN.get(user_id, False)
|
|
if msg.text:
|
|
cmd = msg.text.strip().lower()
|
|
action = MENU_MAP.get(cmd)
|
|
if action == "SD":
|
|
USER_MENU_SHOWN[user_id] = True
|
|
set_state(user_id, "SD_MODE")
|
|
log_menu_stats(user_id, "Service Desk", "Вход")
|
|
# Загружаем шаблоны и встраиваем их в текст
|
|
try:
|
|
import json, asyncio, httpx, urllib3
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
# Импортируем SD_URL и SD_TOKEN из config
|
|
import importlib.util
|
|
spec = importlib.util.spec_from_file_location("custom_config", "/opt/trueconf_bot/config/config.py")
|
|
custom_config = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(custom_config)
|
|
SD_URL = custom_config.SD_URL
|
|
SD_TOKEN = custom_config.SD_TOKEN
|
|
headers = {"authtoken": SD_TOKEN, "Accept": "application/vnd.manageengine.sdp.v3+json"}
|
|
input_data = {"list_info": {"row_count": 100, "start_index": 1}}
|
|
params = {"input_data": json.dumps(input_data)}
|
|
url = f"{SD_URL}/api/v3/request_templates"
|
|
async with httpx.AsyncClient(verify=False) as client:
|
|
resp = await client.get(url, headers=headers, params=params, timeout=30.0)
|
|
if resp.status_code == 200:
|
|
data = resp.json()
|
|
all_templates = data.get("request_templates", [])
|
|
pre_filtered = [
|
|
t for t in all_templates
|
|
if t.get("status") == "ACTIVE"
|
|
and not t.get("inactive", False)
|
|
and str(t.get("id")) != "2"
|
|
]
|
|
async def check_template(t, cl):
|
|
try:
|
|
t_resp = await client.get(url + "/" + str(t["id"]), headers=headers, timeout=15.0)
|
|
if t_resp.status_code == 200:
|
|
detailed = t_resp.json().get("request_template", {})
|
|
return t if detailed.get("show_to_requester") is True else None
|
|
except Exception:
|
|
pass
|
|
return None
|
|
tasks = [check_template(t, client) for t in pre_filtered]
|
|
results = await asyncio.gather(*tasks)
|
|
templates = [t for t in results if t is not None]
|
|
# Пагинация по 10
|
|
per_page = 10
|
|
total_pages = (len(templates) + per_page - 1) // per_page if templates else 1
|
|
start = 0
|
|
end = min(per_page, len(templates))
|
|
page_items = templates[start:end]
|
|
template_lines = ""
|
|
for idx, t in enumerate(page_items):
|
|
num = idx + 1
|
|
digit = f"{num}⃣"
|
|
template_lines += f"{digit} - {t['name']}\n"
|
|
page_info = f"\n📄 <b>Страница 1 из {total_pages}</b>"
|
|
template_text = f"\n\n📋 <b>Выберите шаблон:</b>\n{template_lines}{page_info}"
|
|
full_text = SD_MAIN_MENU_TEXT + template_text
|
|
await msg.answer(full_text, parse_mode="html")
|
|
else:
|
|
await msg.answer(SD_MAIN_MENU_TEXT, parse_mode="html")
|
|
except Exception as e:
|
|
await msg.answer(SD_MAIN_MENU_TEXT, parse_mode="html")
|
|
return
|
|
elif action == "PHOTO":
|
|
USER_MENU_SHOWN[user_id] = True
|
|
set_state(user_id, "PHOTO_MODE")
|
|
log_menu_stats(user_id, "Photo Bot", "Вход")
|
|
await msg.answer(PHOTO_MAIN_MENU_TEXT, parse_mode="html")
|
|
return
|
|
elif action == "TRANSCRIPTION":
|
|
USER_MENU_SHOWN[user_id] = True
|
|
set_state(user_id, "TRANSCRIPTION_MODE")
|
|
log_menu_stats(user_id, "Speech-to-Text", "Вход")
|
|
await msg.answer(TRANSCRIPTION_MAIN_MENU_TEXT, parse_mode="html")
|
|
return
|
|
elif action == "SEARCH":
|
|
USER_MENU_SHOWN[user_id] = True
|
|
set_state(user_id, "SEARCH_MODE")
|
|
log_menu_stats(user_id, "Search Bot", "Вход")
|
|
await msg.answer(SEARCH_MAIN_MENU_TEXT, parse_mode="html")
|
|
return
|
|
elif action == "LK":
|
|
USER_MENU_SHOWN[user_id] = True
|
|
set_state(user_id, "LK_MODE")
|
|
log_menu_stats(user_id, "Личный кабинет", "Вход")
|
|
from lk.handlers import send_dynamic_lk_main_menu
|
|
await send_dynamic_lk_main_menu(msg, user_id)
|
|
return
|
|
else:
|
|
if not menu_shown:
|
|
USER_MENU_SHOWN[user_id] = True
|
|
await msg.answer(WELCOME_TEXT, parse_mode="html")
|
|
return
|
|
await msg.answer(UNKNOWN_MAIN_CMD_TEXT, parse_mode="html")
|
|
return
|
|
USER_MENU_SHOWN[user_id] = True
|
|
await msg.answer(WELCOME_TEXT, parse_mode="html")
|
|
|
|
async def main():
|
|
logger.info("Инициализация Dispatcher и подключение роутеров...")
|
|
dp = Dispatcher()
|
|
dp.include_router(photo_router)
|
|
dp.include_router(sd_router)
|
|
dp.include_router(transcription_router)
|
|
dp.include_router(search_router)
|
|
dp.include_router(lk_router)
|
|
dp.include_router(main_router)
|
|
bot = Bot.from_credentials(server=TC_SERVER, username=TC_LOGIN, password=TC_PASSWORD, dispatcher=dp, verify_ssl=False)
|
|
states.tc_bot = bot
|
|
logger.info("Бот запущен и готов к работе!")
|
|
target_chat = globals().get('ADMIN_CHAT_ID', "533a9784afdfe80c8d80b971a732cd2520a71dc6")
|
|
async def send_startup_notification(bot_instance, chat_id):
|
|
await asyncio.sleep(10)
|
|
try:
|
|
await bot_instance.send_message(chat_id, "✅ <b>Я - online!</b> Запуск прошел успешно.", parse_mode="html")
|
|
logger.info(f"Стартовое сообщение успешно отправлено в чат {chat_id}")
|
|
except Exception as e:
|
|
logger.error(f"Не удалось отправить стартовое сообщение администратору: {e}")
|
|
asyncio.create_task(send_startup_notification(bot, target_chat))
|
|
logger.info("Подключение к серверу TrueConf...")
|
|
await bot.run()
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main_router.message()
|
|
asyncio.run(main())
|
|
except KeyboardInterrupt:
|
|
logger.info("Бот остановлен вручную.")
|