Простейший бот «на движке» WEB WhatsApp

Доброго времени суток всем!

Продолжу изгаляться над Web версией Whatsapp и попробую превратить его в подобие телеграмм-бота. Тем более, что все это бесплатно и не требует никаких внешних сервисов.

Я не буду повторять все действия, которые нужны для для запуска python c необходимыми пакетами, они описаны в моей предыдущей статье.

Только упомяну, что в данном боте (его код будет ниже) не требуется подключение autoit, так как он не предусматривает отправку файлов в ответ на запрос.

Теперь опишу суть задумки. Так как нет специализированного сервера, то придётся использовать выделенный компьютер (виртуальную машину, docker-образ & etc). На нём в цикле запускать робота, который будет просматривать группу, где подключены все пользователи бота, и отвечать на их запросы. Данный робот ничего умного не умеет, кроме как отправить назад, в группу текст сообщения с префиксом «Echo:». Но с учетом предыдущего поста, думаю, что можно превратить его в полноценного «ответчика».

А сейчас, подробнее по структуре бота. Он реализован в классе, который называется whatapp(). В нём есть следующие методы:

• startBrowser — запускает браузер;

• finish — закрывает браузер;

• searchGroup — ищет группу, в которой будут обрабатывать сообщения;

• sendMessage — отсылает сообщение в текущую группу или в текущий чат пользователя;

• searchLastMessageAndProcessMessages — осуществляет поиск последнего обработанного сообщения и затем приступает к обработке новых.

Отмечу встреченные проблемы при построении бота. Первое, для просмотра сообщений нужно блокировать работу JavaScript, чтобы не появлялись новые сообщения и не обновлялся DOM, а для отправки сообщений необходимо разрешить JavaScript. Второе, нужно где-то хранить маркер для последнего обработанного сообщения, я для простоты сохраняю его в файле. Конечно же лучше хранить эти маркеры в базе данных и для каждого пользователя, но я не стал загромождать логику бота. Третье, пришлось отбрасывать сообщения, которые содержат смайлики, так в этом случае пришлось бы еще как-то обрабатывать эти картинки.

И еще одно замечание, ответы отправляются в ту же самую группу, где просматриваются запросы. Лучше бы было поискать чат запросившего пользователя и отправить ответ только ему, но я, опять же, не стал загромождать логику работы бота.

Подчеркну, что я не рассматриваю данное приложение как коммерческий проект, но для небольшой компании, в которой фаворитом среди мессенджеров является Whatsapp, это вполне приемлемое решение.

Теперь текст бота. Комментарии, как обычно, по ходу текста и на двух языках.

from selenium import webdriver from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.common.by import By from selenium.webdriver.chrome.service import Service from webdriver_manager.chrome import ChromeDriverManager from selenium.webdriver.support.ui import WebDriverWait from time import sleep import argparse from datetime import datetime   # Константы для Selenium # Constants for Selenium url = f"https://web.whatsapp.com/" xpath = "//button[@data-testid='compose-btn-send']" xpathAttach = "//div[@data-testid='conversation-clip']" cssIdOfDocument = "[aria-label='Документ']" cssIdOfSendButton = "[aria-label='Отправить']" xpathSearchField = "//div[@data-testid='chat-list-search']" xpathFoundItem = "//span[contains(@class,'matched-text')]" xpathInputLineForText = "//div[@data-testid='conversation-compose-box-input']" xpathMessage = "//div[@role = 'row']" xpathAuthorTime = "span[contains(@class,'_3FuDI')]" xpathText1 = "span[contains(@class,'selectable-text')]" xpathText2 = "span[contains(@class,'copyable-text')]"  commandEnableOrDisableJavaScript = "Emulation.setScriptExecutionDisabled"  # Константы для обработки сообщений # Constants for message process messageIsOutgoing = "message-out" messageIsText = "data-pre-plain-text" messageClassOfBody = "selectable-text copyable-text" messageBodyStart = "<span>" messageBodyStop = "</span>" messageIsPicture = "img" messageWarrningAboutPicture = "Смайлики не обрабатываю!  I don't process emoticons !" fileNameWithLastMessage = "lastmessage.txt"  class whatapp():     """ Это класс бота для Whatsapp        This class is bot for Whatsapp"""       def startBrowser(self):         options = webdriver.ChromeOptions() # если у вас другой браузер, например FireFox, мы просто пишем options = webdriver.FirefoxOptions() . # if you have the Mozilla Fifefox just write                   options = webdriver.FirefoxOptions()         options.add_argument('--allow-profiles-outside-user-dir')         options.add_argument('--enable-profile-shortcut-manager') # УКАЖИТЕ ПУТЬ ГДЕ ЛЕЖИТ ВАШ python ФАЙЛ. Советую создать отдельную папку для него # Specify the path to where your python file is located. I suggest you create a separate folder for it         options.add_argument(r'user-data-dir=X:\\YYYYYY\\ZZZZZZ\\')         options.add_argument('--profile-directory=Profile 1')         options.add_argument('--profiling-flush=n')         options.add_argument('--enable-aggressive-domstorage-flushing')  # эти опции нужны чтобы подавить любые сообщения об ошибках  SSL, сертификатов и т.п. Но работает только последняя :( # these options need to disabled any messages about bad ssl, certification & etc         options.add_argument('--ignore-certificate-errors-spki-list')         options.add_argument('--ignore-certificate-errors')         options.add_argument('--ignore-ssl-errors')         options.add_argument('log-level=3') # INFO = 0, # WARNING = 1, # LOG_ERROR = 2, # LOG_FATAL = 3. # default is 0. # Мы запускаем браузер # We are starting a browser         self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options) # ждём его загрузки # it takes some time to load it         wait = WebDriverWait(self.driver, 30) # идём туда # go to there         self.driver.get(url) # ждём загрузки страницы Whatsapp # we are waiting for Whatsapp page to load         wait.until(EC.element_to_be_clickable((By.XPATH, xpathSearchField)))       def finish(self): # закрыть все # close all         self.driver.quit()      def searchGroup(self, group_name): # Найти группу для обработки # В поле поиска вводим название группы и нажимаем на кнопку "Искать" # Find a group to process # In the search field, enter the name of the group and click "Search".         self.driver.find_element(By.XPATH, xpathSearchField).send_keys(group_name)         self.driver.find_element(By.XPATH, xpathFoundItem).click()         sleep(5)       def sendMessage(self, message): # Разрешаем JavaSctipt # Enable JavaSctipt         self.driver.execute_cdp_cmd(commandEnableOrDisableJavaScript, {'value': False}) # Ищем строку ввода и отправляем туда текст сообщения # Find the input line and send the message text there         self.driver.find_element(By.XPATH, xpathInputLineForText).send_keys(message) # теперь ищем кнопку "Отправить" и нажимаем на нее # now we look for the Send Button and click on it         self.driver.find_element(By.XPATH, xpath).click()         sleep(5)       def searchLastMessageAndProcessMessages(self, dt): # Запрещаем JavaSctipt # Disable   JavaSctipt         self.driver.execute_cdp_cmd(commandEnableOrDisableJavaScript, {'value': True}) # Выберем все сообщения # Select all messages         messages = self.driver.find_elements(By.XPATH, xpathMessage)         print(str(dt) + '. I found %s messages' % len(messages)) # Получим последнеe обработанное сообщение из файла # Get the last processed message from file         f = open(fileNameWithLastMessage, "r")         last_message = f.read()         f.close()         need_process = False # просмотрим все сообщения # review all messages         for item in messages: # получим тело сообщения # get body of the message             line = str(item.get_attribute('innerHTML')) # это исходящее сообщение. пропускаем # this is outgoing message... skip it             if (messageIsOutgoing in line):                  continue # в сообщении нет текста. пропускаем # this message without text... skip it             if not (messageIsText in line):                  continue # извлечем дату, время и автора сообщения # extract date, time and author from the message             index1 = line.index(messageIsText)             index2 = line.index("]", index1 + len(messageIsText))             index3 = line.index(":", index2)             author_datetime = line[index1 + len(messageIsText) + 2 : index3] # извлечем текст сообщения # extract text from the message             index1 = line.index(messageClassOfBody)             index2 = line.index(messageBodyStart, index1 + len(messageClassOfBody))             index3 = line.index(messageBodyStop, index2)             message = line[index2 + len(messageBodyStart): index3] # сформируем полное сообщение для сравнения с последним обработанным # create a complete message for comparison with the last processed message             full_message = author_datetime + " " + message # если сообшение последнее обработанное, то начинаем обработку, пропустив это # if the message was last processed, we start processing by skipping this             if (full_message == last_message):                 need_process = True                 continue # обработка сообщений. Посылаем "эхо" # process messages. Send echo             if need_process: # не будем отвечать на сообщения со смайликами # block a response for any pictures                 if (messageIsPicture in message):  ## smile                     message = messageWarrningAboutPicture                 self.sendMessage("Echo: " + message) # Запрещаем JavaSctipt # Disable   JavaSctipt                 self.driver.execute_cdp_cmd(commandEnableOrDisableJavaScript, {'value': True}) # запишем последнее обработанное сообщение # write last processed message to file         if not (author_datetime is None):             f = open(fileNameWithLastMessage, "w")             f.write(full_message)             f.close() # Разрешаем JavaSctipt # Enable    JavaSctipt         self.driver.execute_cdp_cmd(commandEnableOrDisableJavaScript, {'value': False})   def main(args): # основной бесконечный цикл. Прервать его - Ctrl+C # the main infinite loop. interrupt it with Ctrl+C     try:         while True:             wa = whatapp()             wa.startBrowser()             wa.searchGroup(args.group)             dt = datetime.now()             wa.searchLastMessageAndProcessMessages(dt);             wa.finish()     except KeyboardInterrupt:         wa.finish()     return   if __name__ == '__main__': # мы разбираем параметры командной строки # we are parsing command line parameters     parser = argparse.ArgumentParser(description='Process messages by Whatsapp')     parser.add_argument('--group', help='Text for send', required=True)     args = parser.parse_args() # начать обработку # start  processing     main(args)     

Запуск бота осуществляется следующим образом

python whatsapp.py --group "Тест"

Всем удачи!