Telegram. Телеграм-бот на собственном сервере Linux

Материал из Wiki
Перейти к навигации Перейти к поиску

Ссылки

https://winitpro.ru/index.php/2020/01/23/telegram-bot-dlya-linux-servera/

Настройка

Создаем нового бота

  • в телеграме ищем botfather
  • /newbot
  • ищем нового бота
  • делаем боту /start
  • получаем chat_id: https://api.telegram.org/bot<ТОКЕН_НОВОГО_БОТА>/getUpdates

Сервер с ботом (Debian Linux):

apt install python3 python3-pip

pip3 install python-telegram-bot --upgrade

wget https://winitpro.ru/wp-content/uploads/tools/telegram_bot.zip

nano /opt/bot/bot.sh 

#!/bin/bash
python3 /var/www/alert/bot/bot.py

nano /opt/bot/config.py 

token = '131***126:AAH********LcsEWJdpuFXoQ'
admin = ['250***73'] # сюда вписываем ID пользователя, которому бот будет отправлять сообщения
#dir1 = '/var/www/'
#dir_backup = '/backup/' #путь к папке с файлом бэкапа

nano /opt/bot/bot.py

#!/usr/bin/python
import config
import telegram
import os
import subprocess
import sys
import shlex
import datetime
from subprocess import Popen, PIPE
from telegram.ext import CommandHandler
from imp import reload

#bot = telegram.Bot(token = config.token)
#Проверка бота
#print(bot.getMe())
from telegram.ext import Updater
updater = Updater(token=config.token)
dispatcher = updater.dispatcher


def run_command(command):
    process = subprocess.Popen(shlex.split(command), stdout=subprocess.PIPE)
    global textoutput
    textoutput = ''
    while True:
        global output
        output = process.stdout.readline()
        output = output.decode('utf8')
        if output == '' and process.poll() is not None:
            break
        if output:
            print (output.strip())
        textoutput = textoutput + '\n' + output.strip()
    rc = process.poll()
    return rc


def start(bot, update):
    bot.sendMessage(chat_id=update.message.chat_id, text="Привет, я бот, жду команды")


def help(bot, update):
    reload(config)
    bot.sendMessage(chat_id=update.message.chat_id, text='''список доступных команд:
    /apachestatus - нагрузка на сервере
    /id - id пользователя
    /ifconfig - сетевые настройки
    /df - информация о дисковом пространстве (df -h)
    /free - информация о памяти
    /mpstat - информация о нагрузке на процессор
    /dir1 - объем папки''' + config.dir1 + '''
    /dirbackup - размер файла бэкапа за текущий день в папке ''' + config.dir_backup + '''

    ''')

#функция команады id
def myid(bot, update):
    userid = update.message.from_user.id
    bot.sendMessage(chat_id=update.message.chat_id, text=userid)


#функция команады ifconfig
def ifconfig(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        run_command("ifconfig")
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)

#функция команады df
def df(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        run_command("df -h")
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)

#функция команады free
def free(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        run_command("free -m")
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)

#функция команады mpstat
def mpstat(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        run_command("mpstat")
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)

#функция команады top
def apachestatus(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        run_command("apachestatus")
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)

#функция команады dir1
def dir1(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        dir1_command = "du -sh "+ config.dir1
        run_command(dir1_command)
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)

#функция команады dirbackup - проверяет наличие файла по дате
def dirbackup(bot, update):
    reload(config)
    user = str(update.message.from_user.id)
    if user in config.admin: #если пользовательский id в списке admin то команда выполняется
        now_date = datetime.date.today() # Текущая дата
        cur_year = str(now_date.year) # Год текущий
        cur_month = now_date.month # Месяц текущий
        if cur_month < 10:
            cur_month = str(now_date.month)
            cur_month = '0'+ cur_month
        else:
            cur_month = str(now_date.month)
        cur_day = str(now_date.day) # День текущий
        filebackup = config.dir_backup + cur_year + '-' + cur_month + '-' + cur_day + '.03.00.co.7z'  #формируем имя файла для поиска
        print (filebackup)
        filebackup_command = "ls -lh "+ filebackup
        run_command(filebackup_command)
        bot.sendMessage(chat_id=update.message.chat_id, text=textoutput)




start_handler = CommandHandler('start', start)
dispatcher.add_handler(start_handler)

ifconfig_handler = CommandHandler('ifconfig', ifconfig)
dispatcher.add_handler(ifconfig_handler)

df_handler = CommandHandler('df', df)
dispatcher.add_handler(df_handler)

free_handler = CommandHandler('free', free)
dispatcher.add_handler(free_handler)

mpstat_handler = CommandHandler('mpstat', mpstat)
dispatcher.add_handler(mpstat_handler)

apachestatus_handler = CommandHandler('apachestatus', apachestatus)
dispatcher.add_handler(apachestatus_handler)

dir1_handler = CommandHandler('dir1', dir1)
dispatcher.add_handler(dir1_handler)

dirbackup_handler = CommandHandler('dirbackup', dirbackup)
dispatcher.add_handler(dirbackup_handler)

myid_handler = CommandHandler('id', myid)
dispatcher.add_handler(myid_handler)

help_handler = CommandHandler('help', help)
dispatcher.add_handler(help_handler)


updater.start_polling()

chmod +x /opt/bot/bot.py

Служба

touch /etc/systemd/system/telegram-bot.service

chmod 664 /etc/systemd/system/telegram-bot.service

nano /etc/systemd/system/telegram-bot.service 

[Unit]
Description=Telegram bot
After=network.target

[Service]
ExecStart=/opt/bot/bot.sh

[Install]
WantedBy=multi-user.target

systemctl start telegram-bot.service

systemctl enable telegram-bot.service

systemctl status telegram-bot.service

Отправляем боту команды

- Список команд: /help

- Дополнительные команды добавляем в /opt/bot/bot.py

Источник — https://wiki.cless.pro/index.php?title=Telegram._Телеграм-бот_на_собственном_сервере_Linux&oldid=1534