telegram_bot_kievradar.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Simple Bot to reply to Telegram messages
# This program is dedicated to the public domain under the CC0 license.
"""
This Bot uses the Updater class to handle the bot.

First, a few handler functions are defined. Then, those functions are passed to
the Dispatcher and registered at their respective places.
Then, the bot is started and runs until we press Ctrl-C on the command line.

Usage:
Basic Echobot example, repeats messages, Cand sends
weather map with radar meteo data in Kiev, Ukraine

Press Ctrl-C on the command line or send a signal to the process to stop the
bot.
"""

import telegram
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
import logging
import syslog
import requests
import shutil
import re
import random
import os
import sys
import requests
import hashlib
import random
from bs4 import BeautifulSoup

import lxml.html


from PIL import Image
import numpy as np

import urllib

tmp_imagename = '/tmp/RADAR_KIEV.png'
news_hashes = ["0"]

# Enable logging
logging.basicConfig(level=logging.INFO,
                    format='%(asctime)s - %(name)s \
                    - %(levelname)s - %(message)s')

logger = logging.getLogger(__name__)

syslog.syslog(syslog.LOG_INFO, 'telegram bot started')

# Define a few command handlers. These usually take the two arguments bot and
# update. Error handlers also receive the raised TelegramError object in error.


def start(bot, update):
    update.message.reply_text('Hi!')


def help(bot, update):
    update.message.reply_text('Help!')


def kiev_news_text():
    url = "https://news.yandex.ua/Kyiv/index.html?lang=ru"
    i = 0
    hexdigest = "0"
    request = requests.get(url)
    soup = BeautifulSoup(request.text, "lxml")
    titles = [h2.text for h2 in soup.findAll(
        'div', attrs={'class': 'story__text'})]
    while hexdigest in news_hashes and i < 10:
        i = i + 1
        index = random.randrange(len(titles))
        news_string = titles[index]
        hash_object = hashlib.md5(news_string.encode('utf-8'))
        hexdigest = hash_object.hexdigest()

        if hexdigest not in news_hashes \
                and "пробки" not in news_string \
                and "синоптик" not in news_string:
                    news_hashes.append(hexdigest)
                    i = 10  # break

    return news_string


def kiev_news(bot, update):
    # send typing...
    bot.sendChatAction(action=telegram.ChatAction.TYPING,
                       chat_id=update.message.chat_id)
    update.message.reply_text(kiev_news_text())


def radar_kiev(bot, update):

    # send typing...
    bot.sendChatAction(action=telegram.ChatAction.TYPING,
                       chat_id=update.message.chat_id)

    meteoby_url = "http://meteoinfo.by/radar/UKBB/UKBB_latest.png"
    pngfile = "/tmp/UKBB_latest.png"
    pngtransfile = "/tmp/UKBB_transparent.png"
    output = "/tmp/output.png"
    mapfile = "/tmp/map.png"

    # get radar black-white png
#    r = requests.get(meteoby_url, stream = True)
#    if r.status_code == 200:
#        with open(pngfile, 'wb') as f:
#            r.raw.decode_content = True
#            shutil.copyfileobj(r.raw, f)

    urllib.request.urlretrieve(
        "http://meteoinfo.by/radar/UKBB/UKBB_latest.png", pngfile)

    orig_color = (204, 204, 204, 255)
    replacement_color = (255, 255, 255, 0)
    img = Image.open(pngfile).convert('RGBA')
    data = np.array(img)

    # replace gray colors with transparent
    colors = [204, 192]

    for color in colors:
        orig_color = (color, color, color, 255)
        data[(data == orig_color).all(axis=-1)] = replacement_color

    img2 = Image.fromarray(data, mode='RGBA')
    img2.save(pngtransfile)

    background = Image.open(mapfile).convert('RGBA')
    foreground = Image.open(pngtransfile)

    # merge map with radar png
    background.paste(foreground, (3, 10), foreground)
    background.save(output)

    # send it as answer
    bot.sendPhoto(chat_id=update.message.chat_id, photo=open(output, 'rb'))
#    with open(output,'rb') as photo_file:
#          update.message.reply_photo(photo_file)


def echo(bot, update):
    update.message.reply_text("v17 02" + update.message.text)


def error(bot, update, error):
    logger.warn('Update "%s" caused error "%s"' % (update, error))


def main():
    if "TOKEN_BOT" not in os.environ:
        print("Please set the environment variable TOKEN_BOT")
        sys.exit(1)

    # Create the EventHandler and pass it your bot's token.
    updater = Updater(os.environ["TOKEN_BOT"])

    # Get the dispatcher to register handlers
    dp = updater.dispatcher

    # on different commands - answer in Telegram
    dp.add_handler(CommandHandler("kiev_radar", radar_kiev))
    dp.add_handler(CommandHandler("kiev_news", kiev_news))

    # on noncommand i.e message - echo the message on Telegram
    dp.add_handler(MessageHandler(Filters.text, echo))

    # log all errors
    dp.add_error_handler(error)
    
    logging.getLogger().setLevel(logging.DEBUG)
    logger.debug("acid232 dev version")

    # Start the Bot
    updater.start_polling()

    # Run the bot until the you presses Ctrl-C or the process receives SIGINT,
    # SIGTERM or SIGABRT. This should be used most of the time, since
    # start_polling() is non-blocking and will stop the bot gracefully.
    updater.idle()


if __name__ == '__main__':
    main()