Weather_Bot/bot.py at master · AlexMaximenko/Weather_Bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
import telebot
import pyowm
import apiai, json
import pyowm.exceptions
import config
import time as tm
import datetime
from datetime import date
from datetime import timedelta
from weather import *


bot = telebot.TeleBot(config.BOT_TOKEN, threaded=False)

@bot.message_handler(commands=['start'])
def starting(message):
    start_markup = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True,
                                                     one_time_keyboard=False)
    start_markup.row('/forecast', '/weather', '/hide')
    bot.send_message(message.chat.id, "Привет!\n")
    bot.send_message(message.chat.id, "Чтобы скрыть клавиатуру, нажмите /hide\n")
    bot.send_message(message.chat.id,
                     "Чтобы узнать прогноз на определенный день, нажмите /forecast\n")
    bot.send_message(message.chat.id,
                     "Чтобы узнать текущую погоду, нажмите /weather",
                     reply_markup=start_markup)


@bot.message_handler(commands=['hide'])
def hiding(message):
    bot.send_message(message.chat.id,
                     'Клавиатура спрятана',
                     reply_markup=telebot.types.ReplyKeyboardRemove())


@bot.message_handler(commands=['help'])
def help(message):
    start_markup = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True,
                                                     one_time_keyboard=False)
    start_markup.row('/forecast', '/weather', '/hide')
    bot.send_message(message.chat.id, "Чтобы скрыть клавиатуру, введите /hide\n")
    bot.send_message(message.chat.id,
                     "Чтобы узнать прогноз на определенный день, введите /forecast\n")
    bot.send_message(message.chat.id,
                     "Чтобы узнать текущую погоду, введите /weather",
                     reply_markup=start_markup)


@bot.message_handler(commands=['weather'])
def weather_city_receive(message):
    city_request = bot.send_message(message.chat.id,
                                    'Введите город, в котором хотите узнать погоду')
    bot.register_next_step_handler(city_request, send_weather)


def send_weather(message):
    try:
        get_weather(message.text)
    except pyowm.exceptions.api_response_error.NotFoundError:
        bot.send_message(message.chat.id,
                         '❌Город не найден, убедитесь в правильности написания')
    bot.send_message(message.chat.id, get_weather(message.text))


@bot.message_handler(commands=['forecast'])
def forecast_day_request(message):
    day_markup = telebot.types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=False)
    today = date.today()
    possible_days = []

    for i in range(1, 6):
        delta = timedelta(days = i)
        day = months[(today+delta).month] + ', '+(today + delta).strftime(("%d"))
        possible_days.append(day)

    day_markup.row(*possible_days)
    request =  bot.send_message(message.chat.id,
                                    'В какой день вы хотите узнать погоду?',
                                    reply_markup=day_markup)

    bot.register_next_step_handler(request, forecast_city_request)

def forecast_city_request(message):
    today = date.today()
    possible_days = []

    for i in range(1, 6):
        delta = timedelta(days = i)
        day = months[(today+delta).month] + ', '+(today + delta).strftime(("%d"))
        possible_days.append(day)

    if not any(message.text in day for day in possible_days):
        bot.send_message(message.chat.id,
                         "❌Выбирайте день только из предложенных ниже!",
                         reply_markup=telebot.types.ReplyKeyboardRemove())
    else:
        handler = {possible_days[0]: send_forecast(1),
                possible_days[1]: send_forecast(2),
                possible_days[2]: send_forecast(3),
                possible_days[3]: send_forecast(4),
                possible_days[4]: send_forecast(5)}
        request = bot.send_message(message.chat.id,
                                        'Введите город, в котором хотите узнать погоду')
        bot.register_next_step_handler(request, handler.get(message.text))

def send_forecast(days):
    def forecast(message):
        try:
            get_forecast(message.text, days)
        except pyowm.exceptions.api_response_error.NotFoundError:
            bot.send_message(message.chat.id,
                             '❌Город не найден, убедитесь в правильности написания',
                             reply_markup=telebot.types.ReplyKeyboardRemove())

        bot.send_message(message.chat.id,
                     get_forecast(city=message.text, days=2),
                     reply_markup=telebot.types.ReplyKeyboardRemove())
    return forecast


@bot.message_handler(content_types=['text'])
def send_text(message):
    if message.text.lower().find('прогноз') > -1:
        forecast_day_request(message)
        return
    elif message.text.lower().find('погод') > -1:
        city_request = bot.send_message(message.chat.id,
                                        'Введите город, в котором хотите узнать погоду')
        bot.register_next_step_handler(city_request, send_weather)
        return

    request = apiai.ApiAI(config.APIAI_TOKEN).text_request()
    request.lang = 'ru'
    request.session_id = 'BatlabAIBot'
    request.query = message.text
    responseJson = json.loads(request.getresponse().read().decode('utf-8'))
    response = responseJson['result']['fulfillment']['speech']
    if response:
        bot.send_message(message.chat.id, response)
    else:
        bot.send_message(message.chat.id, 'Я Вас не совсем понял!')

while True:
	try:
		bot.infinity_polling(True)
	except Exception:
		tm.sleep(1)