Toegang tot Linux-server met behulp van Telegram-bot in Python

Heel vaak zijn er situaties waarin hier en nu toegang tot de server nodig is. Verbinden via SSH is echter niet altijd de handigste manier, omdat je misschien geen SSH-client, serveradres of gebruiker/wachtwoord-combinatie bij de hand hebt. Natuurlijk hebben Webmin, wat het beheer vereenvoudigt, maar het biedt ook geen directe toegang.

Dus besloot ik een eenvoudige maar interessante oplossing te implementeren. Namelijk om een ​​Telegram-bot te schrijven die, wanneer hij op de server zelf wordt gelanceerd, de opdrachten uitvoert die ernaar worden verzonden en het resultaat retourneert. Na gestudeerd te hebben verscheidene artikels over dit onderwerp realiseerde ik me dat nog niemand dergelijke implementaties heeft beschreven.

Ik heb dit project geïmplementeerd op Ubuntu 16.04, maar voor een probleemloze werking op andere distributies heb ik geprobeerd alles op een algemene manier te maken.

Bot-registratie

Registreer een nieuwe bot bij @BotFather. Wij sturen hem /newbot en verder in de tekst. We hebben een nieuw bot-token en uw id nodig (u kunt deze bijvoorbeeld krijgen van @gebruikersinfobot).

Python-voorbereiding

Om de bot uit te voeren, gebruiken we de bibliotheek telebot (pip install pytelegrambotapi). De bibliotheek gebruiken subprocess we zullen commando's uitvoeren op de server.

Bot-lancering

Maak een bot.py-bestand op de server:
nano bot.py

En plak de code erin:

from subprocess import check_output
import telebot
import time

bot = telebot.TeleBot("XXXXXXXXX:AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA")#токен бота
user_id = 0 #id вашего аккаунта
@bot.message_handler(content_types=["text"])
def main(message):
   if (user_id == message.chat.id): #проверяем, что пишет именно владелец
      comand = message.text  #текст сообщения
      try: #если команда невыполняемая - check_output выдаст exception
         bot.send_message(message.chat.id, check_output(comand, shell = True))
      except:
         bot.send_message(message.chat.id, "Invalid input") #если команда некорректна
if __name__ == '__main__':
    while True:
        try:#добавляем try для бесперебойной работы
            bot.polling(none_stop=True)#запуск бота
        except:
            time.sleep(10)#в случае падения

We vervangen het bot-token erin door het token dat is uitgegeven door @BotFather en user_id door de id-waarde van uw account. Het controleren van de gebruikers-ID is nodig, zodat de bot alleen u toegang geeft tot uw server. Functie check_output() voert de gegeven opdracht uit en retourneert het resultaat.

Het blijft alleen om de bot te starten. Voor lopende processen op de server gebruik ik het liefst screen (sudo apt-get install screen):

screen -dmS ServerBot python3 bot.py

(waarbij "ServerBot" de proces-ID is)

Het proces wordt automatisch op de achtergrond uitgevoerd. Laten we naar de dialoog met de bot gaan en controleren of alles werkt zoals het hoort:

Gefeliciteerd! De bot voert de opdrachten uit die eraan zijn verzonden. Om toegang te krijgen tot de server, hoeft u alleen maar een dialoog met de bot te openen.

Commando herhaling

Om de status van de server te bewaken, moet u vaak dezelfde opdrachten uitvoeren. Daarom zal de implementatie van herhalende opdrachten zonder ze opnieuw te verzenden erg misplaatst zijn.

We zullen het implementeren met behulp van inline-knoppen onder de berichten:

from subprocess import check_output
import telebot
from telebot import types #Добавляем импорт кнопок
import time

bot = telebot.TeleBot("XXXXXXXXX:AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA")#Токен бота
user_id = 0 #id вашего аккаунта
@bot.message_handler(content_types=["text"])
def main(message):
   if (user_id == message.chat.id): #проверяем, что пишет именно владелец
      comand = message.text  #текст сообщения
      markup = types.InlineKeyboardMarkup() #создаем клавиатуру
      button = types.InlineKeyboardButton(text="Повторить", callback_data=comand) #создаем кнопку
      markup.add(button) #добавляем кнопку в клавиатуру
      try: #если команда невыполняемая - check_output выдаст exception
         bot.send_message(user_id, check_output(comand, shell = True,  reply_markup = markup)) #вызываем команду и отправляем сообщение с результатом
      except:
         bot.send_message(user_id, "Invalid input") #если команда некорректна

@bot.callback_query_handler(func=lambda call: True)
def callback(call):
  comand = call.data #считываем команду из поля кнопки data
  try:#если команда не выполняемая - check_output выдаст exception
     markup = types.InlineKeyboardMarkup() #создаем клавиатуру
     button = types.InlineKeyboardButton(text="Повторить", callback_data=comand) #создаем кнопку и в data передаём команду
     markup.add(button) #добавляем кнопку в клавиатуру
     bot.send_message(user_id, check_output(comand, shell = True), reply_markup = markup) #вызываем команду и отправляем сообщение с результатом
  except:
     bot.send_message(user_id, "Invalid input") #если команда некорректна

if __name__ == '__main__':
    while True:
        try:#добавляем try для бесперебойной работы
            bot.polling(none_stop=True)#запуск бота
        except:
            time.sleep(10)#в случае падения

De bot opnieuw opstarten:

killall python3
screen -dmS ServerBot python3 bot.py

Laten we nogmaals controleren of alles correct werkt:

Door op de knop onder het bericht te drukken, moet de bot het commando herhalen van waaruit dit bericht is verzonden.

In plaats Output

Deze methode pretendeert natuurlijk niet een vervanging te zijn voor de klassieke verbindingsmethoden, maar stelt u in staat om snel de status van de server te achterhalen en er opdrachten naar te sturen die geen complexe uitvoer vereisen.

Bron: www.habr.com