[Python] Telegram 응답 받기

https://goodbyeanma.tistory.com/157

[Python] Telegram 메시지 보내주기

위의 글에서는 Python 외부라이브러리를 사용해서 telegram 봇이 메시지를 보내주는 기능을 구현해 보았다.

 

이번 글에서는 사용자의 메시지에 Telegram 봇이 응답해 주는 기능을 구현해 볼 것이다.

 

 

 

라이브러리 로직 기본 구조

 

이번에는 안정적인 구현을 위해서 python-telegram-bot 13.8 버전을 다운로드하여 줄 것이다.

 

# 기존 라이브러리 제거
pip3 uninstall python-telegram-bot

# python-telegram-bot 13.8 버전 설치
pip3 install python-telegram-bot==13.8

 

봇과 응답을 주고받기 위해서는 아래의 객체구조를 이해해야 한다.

 

 

python-telegram-bot 라이브러리에서는 Updater와 Dispatcher 객체를 사용해서

 

Telegram Bot과 상호작용 할 수 있게 해 준다.

 

Updater는 사용자로부터 새로운 메시지가 왔는지를 주기적으로 확인하는 기능을 수행한다.

 

그리고 만약, 사용자로부터 메시지 혹은 특정 명령어가 왔다면 이를 Queue에 저장을 해준다.

 

Dispatcher는 Updater 가 Queue에 넣어둔 사용자 입력 데이터를 가져가서 처리한다.

 

각 요청에 대한 처리를 담당할 핸들러 객체를 미리 지정해 두고 

 

특정 요청이 들어오게 되면, 지정된 핸들러를 통해서 명령이나 메시지에 대한 

 

처리를 수행한다.

 

 

 

 

 

Telegram Bot 응답하기

 

처음에는 간단하게 사용자가 '/command' 명령어를 적어주면,

 

Telegram Bot 이 반응을 하는 기능을 구현해 보자.

 

from telegram.ext import *
import json

""" 
Read telegram related information.
""" 
def telegram_read_infos(telegram_info_path):
    
    telegram_dict = {}

    with open(telegram_info_path, "r") as f:
        data = json.load(f)

    telegram_dict["token"] = data["token"]

    return telegram_dict


# command handler
def start(update, context):
    user_id = update.effective_chat.id
    context.bot.send_message(chat_id=user_id, text="안녕하세요 Telegram bot 입니다.")
    
    

""" 
main function
""" 
def main():
    
    tele_info_path = './tele_info.json'
    telegram_dict = telegram_read_infos(tele_info_path)

    updater = Updater(token=telegram_dict['token'], use_context=True)
    dispatcher = updater.dispatcher
    
    start_handler = CommandHandler('command', start)
    dispatcher.add_handler(start_handler)

    
    #polling
    updater.start_polling()
    updater.idle()
    

""" 
main function - execute
""" 
if __name__ == '__main__':
    main()

 

json 파일에서 Telegram bot에 관한 토큰 정보를 읽어와서

 

사용자가 '/command'라는 특정 메시지를 보내주면 응답해 주는 구조이다.

 

코드를 실행하면 아래와 같이 정상적으로 기능을 수행하는 것을 볼 수 있다.

 

그럼 이번에는 추가적으로, 사용자가 보내는 메시지를 

 

따라서 응답해주는 코드를 추가해 주자.

 

python-telegram-bot Handler는 크게 두 개로 나뉜다.

 

하나는 '/~'와 같은 명령어를 처리해 주는 'CommandHandler' 

 

또 하나는, 명령어가 아닌 사용자의 메시지를 받아서 처리해주는

 

'MessageHandler'가 존재한다.

 

이번에 추가할 기능은 사용자의 메시지를 받아서 그대로 사용자에게 출력해줘야 하므로

 

'MessageHandler' 객체를 사용하면 된다.

 

from telegram.ext import *
import json

""" 
Read telegram related information.
""" 
def telegram_read_infos(telegram_info_path):
    
    telegram_dict = {}

    with open(telegram_info_path, "r") as f:
        data = json.load(f)

    telegram_dict["token"] = data["token"]

    return telegram_dict


# command handler
def start(update, context):
    user_id = update.effective_chat.id
    context.bot.send_message(chat_id=user_id, text="안녕하세요 Telegram bot 입니다.")
    

# message handler
def echo(update, context):
    user_id = update.effective_chat.id
    user_text = update.message.text
    context.bot.send_message(chat_id=user_id, text=user_text)


""" 
main function
""" 
def main():
    
    tele_info_path = './tele_info.json'
    telegram_dict = telegram_read_infos(tele_info_path)

    updater = Updater(token=telegram_dict['token'], use_context=True)
    dispatcher = updater.dispatcher
    
    start_handler = CommandHandler('command', start)
    dispatcher.add_handler(start_handler)

    echo_handler = MessageHandler(Filters.text & (~Filters.command), echo)
    dispatcher.add_handler(echo_handler)
    
    #polling
    updater.start_polling()
    updater.idle()
    

""" 
main function - execute
""" 
if __name__ == '__main__':
    main()

 

해당 코드를 실행해 주면 아래와 같이 정상적으로 기능하는 것을 확인할 수 있다.

 

 

 

 

 

 

Telegram Bot 응답 응용 - 환율정보 받아오기

 

이번에는 Telegram Bot을 사용해서 환율 정보를 가져오는 코드를 구성해 보자.

 

"미국 환율"이라고 치면 1달러당 원화로 얼마인지 봇이 응답해 주는 방식이다.

 

웹 스크랩핑 전용 라이브러리인 'beautifulsoup4' 라이브러리를 설치해 주자.

 

pip3 install beautifulsoup4

 

그리고 아래와 같이 구현해 주면 된다.

 

from telegram.ext import *
import json
from bs4 import BeautifulSoup
import requests

""" 
Read telegram related information.
""" 
def telegram_read_infos(telegram_info_path):
    
    telegram_dict = {}

    with open(telegram_info_path, "r") as f:
        data = json.load(f)

    telegram_dict["token"] = data["token"]

    return telegram_dict


"""
Function to get exchange rate information
"""
def exchange_rate(query):

    currency_dict = {}

    # 검색 결과 페이지에 접속하여 HTML 코드를 가져옴
    url = f"https://www.google.com/search?q={query}"
    response = requests.get(url)
    html = response.text
    
    # HTML 코드를 분석하여 원하는 데이터 추출
    soup = BeautifulSoup(html, "html.parser")
    
    target_currency = soup.find("span", {"class": "r0bn4c rQMQod"}).get_text()
    korea_won = soup.find("div", {"class": "BNeawe iBp4i AP7Wnd"}).get_text()

    currency_dict['target_currency'] = target_currency
    currency_dict['korea_won'] = korea_won

    return currency_dict


# command handler
def start(update, context):
    user_id = update.effective_chat.id
    context.bot.send_message(chat_id=user_id, text="안녕하세요 Telegram bot 입니다.")
    

# message handler
def echo(update, context):
    user_id = update.effective_chat.id
    user_text = update.message.text
    currency_dict = exchange_rate(user_text)

    result = '{} {}'.format(currency_dict['target_currency'],currency_dict['korea_won'])

    context.bot.send_message(chat_id=user_id, text=result)


""" 
main function
""" 
def main():
    
    tele_info_path = './tele_info.json'
    telegram_dict = telegram_read_infos(tele_info_path)

    updater = Updater(token=telegram_dict['token'], use_context=True)
    dispatcher = updater.dispatcher
    
    start_handler = CommandHandler('command', start)
    dispatcher.add_handler(start_handler)

    echo_handler = MessageHandler(Filters.text & (~Filters.command), echo)
    dispatcher.add_handler(echo_handler)
    
    #polling
    updater.start_polling()
    updater.idle()
    

""" 
main function - execute
""" 
if __name__ == '__main__':
    main()

 

 

웹 스크랩핑은 HTML, XML, JSON 등의 마크업 언어를 파싱하고 분석하는 것으로,

 

환율 정보가 어떤 tag / class 내에 있는지 확인해보아야 한다.

    target_currency = soup.find("span", {"class": "r0bn4c rQMQod"}).get_text()
    korea_won = soup.find("div", {"class": "BNeawe iBp4i AP7Wnd"}).get_text()

 

위의 코드 부분은 파이썬 코드에서 print(soup) 의 결과를 분석하여

 

환율 정보를 찾아내어 tag 종류와 class 명을 기입해 준 것이다.

 

코드를 실행해 주고 환율 정보를 물어봐주면 아래와 같이 정확하게 응답한다.