גישה לשרת לינוקס באמצעות בוט Telegram ב-Python

לעתים קרובות יש מצבים שבהם יש צורך בגישה לשרת כאן ועכשיו. עם זאת, חיבור באמצעות SSH הוא לא תמיד הדרך הנוחה ביותר, מכיוון שאולי אין לך לקוח SSH, כתובת שרת או שילוב משתמש/סיסמה בהישג יד. כמובן שיש Webmin, מה שמפשט את הניהול, אך הוא גם אינו מספק גישה מיידית.

אז החלטתי ליישם פתרון פשוט אך מעניין. כלומר, לכתוב בוט של טלגרם שכאשר יופעל בשרת עצמו, יבצע את הפקודות שנשלחו אליו ויחזיר את התוצאה. לאחר שלמד כמה מאמרים בנושא זה, הבנתי שאף אחד עדיין לא תיאר יישומים כאלה.

יישמתי את הפרויקט הזה על אובונטו 16.04, אבל לריצה ללא בעיות בהפצות אחרות, ניסיתי לעשות הכל בצורה כללית.

רישום בוט

רשום בוט חדש עם @BotFather. אנחנו שולחים אותו /newbot ועוד בטקסט. נצטרך אסימון בוט חדש ואת המזהה שלך (תוכל לקבל אותו, למשל, מ @userinfobot).

הכנה לפייתון

כדי להפעיל את הבוט, נשתמש בספרייה telebot (pip install pytelegrambotapi). שימוש בספרייה subprocess נבצע פקודות בשרת.

השקת בוט

צור קובץ bot.py בשרת:
nano bot.py

ותדביק את הקוד לתוכו:

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)#в случае падения

אנו מחליפים את אסימון הבוט בו בזה שהונפק על ידי @BotFather, ואת user_id בערך המזהה של חשבונך. יש צורך בבדיקת מזהה המשתמש כדי שהבוט יספק גישה לשרת שלך רק לך. פוּנקצִיָה check_output() מבצע את הפקודה הנתונה ומחזיר את התוצאה.

נותר רק להפעיל את הבוט. להפעלת תהליכים על השרת, אני מעדיף להשתמש screen (sudo apt-get install screen):

screen -dmS ServerBot python3 bot.py

(כאשר "ServerBot" הוא מזהה התהליך)

התהליך יפעל אוטומטית ברקע. בוא נלך לדיאלוג עם הבוט ונבדוק שהכל עובד כמו שצריך:

מזל טוב! הבוט מבצע את הפקודות שנשלחות אליו. כעת, על מנת לגשת לשרת, אתה רק צריך לפתוח דיאלוג עם הבוט.

חזרה על הפקודה

לעתים קרובות, כדי לפקח על מצב השרת, אתה צריך לבצע את אותן פקודות. לכן, יישום פקודות חוזרות מבלי לשלוח אותן מחדש יהיה מאוד לא במקום.

ניישם את זה באמצעות כפתורים מוטבעים מתחת להודעות:

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)#в случае падения

הפעלה מחדש של הבוט:

killall python3
screen -dmS ServerBot python3 bot.py

בוא נבדוק שוב שהכל עובד כמו שצריך:

בלחיצה על הכפתור מתחת להודעה, על הבוט לחזור על הפקודה שממנה נשלחה הודעה זו.

במקום מסקנה

כמובן ששיטה זו אינה מתיימרת להיות תחליף לשיטות החיבור הקלאסיות, אולם היא מאפשרת לברר במהירות על מצב השרת ולשלוח אליו פקודות שאינן דורשות פלט מורכב.

מקור: www.habr.com