ChessCalcNextTour/bots/admin_bot.py

59 lines
1.9 KiB
Python
Raw Normal View History

"""
Админский Telegram-бот chessCalc.
Показывает статистику использования клиентского бота.
"""
import os
import sys
from telegram import Update
from telegram.ext import Application, CommandHandler, ContextTypes
from telegram.constants import ParseMode
from bots.stats import get_stats, record_user
TOKEN = os.environ.get('ADMIN_BOT_TOKEN', '')
START_MSG = (
'<b>🔧 chessCalc — админский бот</b>\n\n'
'Доступные команды:\n'
'/stat — статистика использования клиентского бота\n'
'/start — эта справка'
)
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
user = update.effective_user
record_user(user.id, user.username or '', user.first_name or '',
user.last_name or '')
await update.message.reply_text(START_MSG, parse_mode=ParseMode.HTML)
async def stat(update: Update, context: ContextTypes.DEFAULT_TYPE):
user = update.effective_user
record_user(user.id, user.username or '', user.first_name or '',
user.last_name or '')
stats = get_stats()
text = (
'<b>📊 Статистика chessCalc</b>\n\n'
f'👤 Уникальных пользователей: <b>{stats["unique_users"]}</b>\n'
f'🔢 Всего запросов: <b>{stats["total_requests"]}</b>\n'
f'📅 За сегодня: <b>{stats["today_requests"]}</b>'
)
await update.message.reply_text(text, parse_mode=ParseMode.HTML)
def main():
if not TOKEN:
print('ADMIN_BOT_TOKEN not set', file=sys.stderr)
sys.exit(1)
app = Application.builder().token(TOKEN).build()
app.add_handler(CommandHandler('start', start))
app.add_handler(CommandHandler('stat', stat))
print('Admin bot started', file=sys.stderr)
app.run_polling()
if __name__ == '__main__':
main()