Как задать переменную для команды в боте Telegram?
Здравствуйте.
Имеется бот, написан на питоне, в котором прописана команда /status - подключается по ssh к серверу используя paramiko, выполняет определенные действия на сервере и ответом присылает статус действий, которые прописаны в paramiko.
Вопрос следующий, так как серверов достаточное количество, каким образом осуществить следующее:
Отправляя боту /status(IP-сервера) он он понимал что нужно менять IP в зависимости от того что мы ему напишем к примеру /status192.168.15.5 или /status192.168.15.8
Возможно ли такое? Или нужно писать команды для каждого из серверов?
Можно пожалуйста подробнее, не совсем понял как осуществить.
Сейчас код для одного сервера:
import paramiko
host = 'ip'
user = 'myuser'
psw = '1234'
port = 22
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# Подключение
ssh.connect(hostname=host, port=port, username=user, password=psw)
# Выполнение команды
stdin, stdout, stderr = ssh.exec_command("uptime -p")
# Читаем результат:
result = stdout.read() + stderr.read()
ssh.close()
@bot.message_handler(commands=['status'])
def send_start(message):
# отправка простого сообщения
bot.send_message(message.chat.id, result)
Соответственно выполнив команду боту /status прилетает инфа по нему.
А как теперь что бы выполнять команды по типу /statusIP-сервера, выполнять по конкретно введенному IP?
enabl3,
1) Оберните код в тег code
2) Вынесите логику работы с ssh в отдельную функцию, принимающую на вход ip и используйте её в в хендлере /status:
def get_status(host, ...):
<общение с сервером>
...
# Вариант 1
# Используете команды вида /status 1.1.1.1
def send_start(msg):
# получаем адрес сервера
host = msg.text.split()[1]
# Общаемся
result = get_status(host)
# Отправляем ответ
send_m(uid, result)
# Вариант 2
# Команды вида /status1.1.1.1
# ctrl + c, ctrl + v
В первом случае вам достаточно будет написать /status <host>, чтобы получить результат.
enabl3, я предлагаю вместо создания большого числа команд /status{IP} использовать одну команду /status, которой вы будете передавать ip в качестве параметра, e.g. /status 192.168.16.5
xdgadd, Спасибо большое, все работает.
Можете еще подсказать, возможно ли осуществить следующее, если сервер вдруг недоступен что бы приходил ответ что он недоступен.
И как можно этот код использовать для других команд?
Если мы его используем для /status IP
Этот же код, только с другими ssh запросами для команды /start IP?
enabl3,
>приходил ответ
Скорее всего, в таком случае ssh выкинет исключение. Вам нужно его поймать и вернуть в качестве результат сообщение о недоступности сервера.
>для других команд
Сначала разберитесь, как он работает.
Отрравляя команду боту, например /start, вы отправляете самое обычное сообщение. В коде происходит разбиение текста сообщения по пробелу и получается последний элемент. Например, если вы просто отправите боту /status, он упадёт с ошибкой.
Вообще, чуть-чуть изменив код, вы можете сделать команду вида /execute host command_to_execute, которая будет выполнять command_to_execute на сервере с адресом ip.
xdgadd, Спасибо, буду ковырять и изучать. Если можно пожалуйста последнюю подсказочку, что именно определяет разделение команды /status IP с пробелом или без
xdgadd, Вы ОООЧень мне помогли. Уже поставил, если можно пожалуйста последнюю подсказочку, что именно определяет разделение команды /status IP с пробелом или без
enabl3, я вам выше ссылку дал. message.text - текст сообщения, представленный питоновским объектом str, который имеет метод split, разбивающий строку по разделителю.
xdgadd, Подскажите еще пожалуйста, как можно на обычную команду /status которая без ИП, что бы отправляло сообщение, что команда не распознана, что бы бот не падал, и понятно было тому кто отправляет что не верно она введена.
xdgadd, Сейчас у меня есть проверка на доступность сервера, как сюда еще добавить проверку на корректность?
def listener(messages):
@bot.message_handler(commands=['status'])
def handle_status_request(message):
# получение из чата IP сервака
ip = message.text.split()[-1]
# проверка его на доступность
r = pyping.ping(ip)
if r.ret_code == 0:
result = ssh.get_status(ip)
bot.send_message(message.chat.id, result)
else:
result = ("Указанный сервер недоступен")
bot.send_message(message.chat.id, result)
xdgadd, если данных нет, тогда отправлять сообщение, если есть, проверять пинг и идти дальше, но пока не могу сообразить как это написать на питоне, так как только недавно начал его ковырять ((
enabl3, вы спросили, как уведомить пользователя он неправильном использовании команды статус. Ошибки, которые вы можете обработать сами - это команда без IP и команда с неверным IP (e.g. 256.1.1.1). В первом случае можно воспользоваться функцией len, а во втором - регулярными выражениями.
enabl3, регексп лучше скомпилировать вне функции, либо юзать сразу re.match:
ip_pat = re.compile("^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$")
...
def sample_f(msg):
ip = msg.text.split()[-1]
if ip_pat.match(ip) is not None:
do_something()
Еще вы можете использовать `re.findall()` чтобы найти всех вхождения ip адреса в сообщении.
enabl3, ваш код вообще не будет работать. Если вы планируете командой получать только один ip адрес, то лучше использовать re.match. По вашему коду:
1. if ip in valid_ip проверяет вхождение ip в списке найденных адресов. Это довольно глупо выглядит. Либо итерируйтесь по возвращаемой коллекции, либо используйте методы для поиска единичного вхождения.
2. Ваше регулярное выражение написано не верно и будет работать только со строкой "192.168.".
Вот как будет выглядеть правильное, если вам нужно получать любые адреса, начинающиеся с 192.168: 192\.168\.\d{1,3}\.\d{1,3}. Для тестирования регулярных выражение удобно использовать regex101.
В итоге, ваш код должен выглядеть примерно так:
ip = message.text.split()[-1]
valid_ip = re.match(r'192\.168\.\d{1,3}\.\d{1,3}', ip)
if valid_ip is not None:
valid_ip = valid_ip.group()
...
Перед тем как писать код, прочитайте документацию.
Простой