mirror of
https://github.com/EDeev/npr_meteobot.git
synced 2026-06-15 19:11:06 +03:00
262 lines
13 KiB
Python
262 lines
13 KiB
Python
import config, asyncio, info
|
||
import logging, pymorphy2, time
|
||
|
||
from aiogram import Bot, Dispatcher, executor, types
|
||
from sql import SQLighter
|
||
from datetime import datetime as dt
|
||
|
||
# задаем уровень логов
|
||
logging.basicConfig(level=logging.INFO)
|
||
|
||
# инициализируем бота
|
||
bot = Bot(token=config.API_TOKEN)
|
||
dp = Dispatcher(bot)
|
||
keyboard = types.InlineKeyboardMarkup()
|
||
morph = pymorphy2.MorphAnalyzer()
|
||
|
||
# инициализируем соединение с БД
|
||
db = SQLighter('sub.db')
|
||
|
||
|
||
# СТАРТОВЫЕ КОМАНДЫ
|
||
@dp.message_handler(commands=['start'])
|
||
async def start(message: types.Message):
|
||
if not db.user_exists(message.from_user.id):
|
||
# если юзера нет в базе, добавляем его
|
||
db.add_user(message.from_user.id)
|
||
|
||
buttons = [types.InlineKeyboardButton(text="1-я смена", callback_data="prt1"),
|
||
types.InlineKeyboardButton(text="2-я смена", callback_data="prt2")]
|
||
|
||
keyboard = types.InlineKeyboardMarkup(row_width=2)
|
||
keyboard.add(*buttons)
|
||
|
||
await message.answer(f"Привет, {message.from_user.first_name}!\nЯ - погодный бот, созданный для выдачи информации о"
|
||
f" вероятных актировках. Прошу вас выбрать вашу смену обучения!\n\n<i>Для последующей "
|
||
f"смены: города или смены обучения воспользуйтесь командой <b>/town</b></i>",
|
||
parse_mode=types.ParseMode.HTML, reply_markup=keyboard)
|
||
|
||
|
||
@dp.message_handler(content_types=['sticker'])
|
||
async def stick(message: types.Message):
|
||
print(message.sticker.file_id)
|
||
|
||
|
||
@dp.message_handler(commands=['town'])
|
||
async def town(message: types.Message):
|
||
|
||
buttons = [types.InlineKeyboardButton(text="1-я смена", callback_data="prt1"),
|
||
types.InlineKeyboardButton(text="2-я смена", callback_data="prt2")]
|
||
|
||
keyboard = types.InlineKeyboardMarkup(row_width=2)
|
||
keyboard.add(*buttons)
|
||
|
||
await message.answer("<b>Прошу вас выбрать вашу смену обучения!</b>", parse_mode=types.ParseMode.HTML,
|
||
reply_markup=keyboard)
|
||
|
||
|
||
# ИЛАЙН КЛАВИАТУРА
|
||
@dp.callback_query_handler(text="prt1")
|
||
async def part_one(call: types.CallbackQuery):
|
||
db.update_user(call.message.chat.id, 3, 1)
|
||
|
||
buttons = [types.InlineKeyboardButton(text="Норильск", callback_data="nor"),
|
||
types.InlineKeyboardButton(text="Талнах", callback_data="tal"),
|
||
types.InlineKeyboardButton(text="Оганер", callback_data="oga"),
|
||
types.InlineKeyboardButton(text="Каеркан", callback_data="kae")]
|
||
|
||
keyboard = types.InlineKeyboardMarkup(row_width=2)
|
||
keyboard.add(*buttons)
|
||
|
||
await call.message.answer(text="<b><i>Отлично!</i></b> Теперь вам надо выбрать район в котором вы проживаете "
|
||
"на территории НПР!",
|
||
parse_mode=types.ParseMode.HTML, reply_markup=keyboard)
|
||
|
||
|
||
@dp.callback_query_handler(text="prt2")
|
||
async def part_two(call: types.CallbackQuery):
|
||
db.update_user(call.message.chat.id, 3, 2)
|
||
|
||
buttons = [types.InlineKeyboardButton(text="Норильск", callback_data="nor"),
|
||
types.InlineKeyboardButton(text="Талнах", callback_data="tal"),
|
||
types.InlineKeyboardButton(text="Оганер", callback_data="oga"),
|
||
types.InlineKeyboardButton(text="Каеркан", callback_data="kae")]
|
||
|
||
keyboard = types.InlineKeyboardMarkup(row_width=2)
|
||
keyboard.add(*buttons)
|
||
|
||
await call.message.answer(text="<b><i>Конечно я вам не завидую, но куда деваться)</i></b> Теперь вам надо выбрать "
|
||
"район в котором вы проживаете на территории НПР!",
|
||
parse_mode=types.ParseMode.HTML, reply_markup=keyboard)
|
||
|
||
|
||
@dp.callback_query_handler(text=["nor", "tal", "oga", "kae"])
|
||
async def function(call: types.CallbackQuery):
|
||
txt = call.data
|
||
if txt == "nor":
|
||
db.update_user(call.message.chat.id, 4, 1)
|
||
elif txt == "tal":
|
||
db.update_user(call.message.chat.id, 4, 2)
|
||
elif txt == "oga":
|
||
db.update_user(call.message.chat.id, 4, 3)
|
||
elif txt == "kae":
|
||
db.update_user(call.message.chat.id, 4, 4)
|
||
|
||
await call.message.answer(text='<b>Прекрасно!</b> Вы заполнили данные, теперь если вы хотите подписаться на '
|
||
'утреннюю рассылку об актировках воспользуйтесь командой <b>/letter</b>.\n\n'
|
||
'<i><b>Для полного ознакомления можете воспользоваться командой /help</b></i>',
|
||
parse_mode=types.ParseMode.HTML)
|
||
|
||
|
||
# КОМАНДЫ АКТИВАЦИИ ПОДПИСОК
|
||
@dp.message_handler(commands=['notice'])
|
||
async def notification(message: types.Message):
|
||
if str(message.chat.id) not in db.get_send():
|
||
db.update_user(message.chat.id, 2, 1)
|
||
await message.answer("Увидомление приходит каждый день не зависимо от наличия актировки!")
|
||
else:
|
||
db.update_user(message.chat.id, 2, 0)
|
||
await message.answer("Уведомление приходит, только если есть актировка!")
|
||
|
||
|
||
@dp.message_handler(commands=['letter'])
|
||
async def letter(message: types.Message):
|
||
if str(message.chat.id) not in db.get_subs():
|
||
db.update_user(message.chat.id, 1, 1)
|
||
await message.answer("Вы подписались на рассылку!")
|
||
else:
|
||
db.update_user(message.chat.id, 1, 0)
|
||
await message.answer("Вы отписались от расслыки!")
|
||
|
||
|
||
@dp.message_handler(commands=['storm'])
|
||
async def storm(message: types.Message):
|
||
await message.answer(info.storm(), parse_mode=types.ParseMode.MARKDOWN)
|
||
|
||
|
||
@dp.message_handler(commands=['acta'])
|
||
async def act(message: types.Message):
|
||
mail = f"<b>Актировка за {'.'.join(reversed(str(db.get_data()[0]).split('-')))}</b>\n"
|
||
|
||
for part in [1, 2]:
|
||
mail = mail + f"\n<b>Для {part}-ой смены:</b>\n"
|
||
if db.check_acta(part):
|
||
cities = config.NAMES
|
||
city = db.get_acta(part)
|
||
|
||
for i, town in enumerate(city):
|
||
if str(town) != "нет":
|
||
mail = mail + f"<b>*</b> В {morph.parse(cities[i])[0].inflect({'loct'}).word.title()} актировка " \
|
||
f"с 1 по {town} классы!\n"
|
||
else:
|
||
mail = mail + f"<b>*</b> В {morph.parse(cities[i])[0].inflect({'loct'}).word.title()} актировки нет!\n"
|
||
|
||
else:
|
||
mail = mail + "На данный момент информации на сайте нету!\n\n"
|
||
await message.answer(mail, parse_mode=types.ParseMode.HTML)
|
||
|
||
|
||
@dp.message_handler(commands=['weather'])
|
||
async def storm(message: types.Message):
|
||
text = message.text.split()
|
||
try:
|
||
day = int(text[1])
|
||
if 0 <= day <= 9:
|
||
weather = db.get_weather(day)
|
||
|
||
txt = "<b>Погода:</b>\n\n"
|
||
for hour in weather:
|
||
dat = weather[hour]
|
||
if len(dat[1]) == 2:
|
||
par = f"{dat[1][0]}-{dat[1][1]}"
|
||
else:
|
||
par = str(dat[1][0])
|
||
txt = txt + f"<b>В {hour}:</b>\n Температура: {dat[0]}\n Скорость и порывы ветра: {par}\n Направление ветра: {dat[2]}\n Осадки: {dat[3]}\n\n"
|
||
|
||
await message.answer(txt, parse_mode=types.ParseMode.HTML)
|
||
else:
|
||
raise NameError('Число не вподходящем диапозоне!')
|
||
except Exception:
|
||
await message.answer("Введи число дня по счёту от сегодня, погода которого тебе нужна!\n\nТо есть:\nСегодня - {/waether 0}\nЗавтра - {/weather 1}",
|
||
parse_mode=types.ParseMode.MARKDOWN)
|
||
|
||
|
||
@dp.message_handler(commands=['help'])
|
||
async def main(message: types.Message):
|
||
await message.answer('*Инструкция использования:*\n\n'
|
||
'*/town* - с помощью этой команды вы можете изменить данные о городе и смене обучения\n'
|
||
'*/acta* - команда, выдающая данные об актировке на все города на данный момент\n'
|
||
'*/storm* - команда, выдающая данные о штормовом предупреждении\n'
|
||
'*/letter* - команда, позваляющая подписаться на утреннюю рассылку о наличии актировки\n'
|
||
'*/notice* - команда, дающая выбор - получать уведомления в любом случае или только при '
|
||
'наличии актировки\n', types.ParseMode.MARKDOWN)
|
||
|
||
|
||
# УВЕДОМЛЕНИЕ
|
||
async def timer(wait_for):
|
||
while True:
|
||
await asyncio.sleep(wait_for)
|
||
city, part = info.act()
|
||
|
||
if city:
|
||
date_old, part_old = db.get_data()
|
||
date_now, part_num = str(dt.now().date()), int(part[0])
|
||
|
||
if (part_num == 1 and date_old != date_now) or (part_num == 2 and date_old == date_now):
|
||
if part_old != part_num:
|
||
db.update_data(dt.now().date(), part_num)
|
||
db.save_acta(city, part_num)
|
||
for twn in city:
|
||
town = morph.parse(twn[0][1])[0].inflect({'datv'}).word
|
||
if isinstance(twn[1], int):
|
||
for s in db.get_users(part_num, twn[0][0]):
|
||
try:
|
||
await bot.send_sticker(chat_id=s, sticker=config.ID_STICKERS + config.STICKERS[1])
|
||
await bot.send_message(chat_id=s, text=f"*Для {part} по {town.title()} объявлена актировка c 1 по {twn[1]} классы!*",
|
||
parse_mode=types.ParseMode.MARKDOWN)
|
||
except Exception as e:
|
||
print(repr(e))
|
||
db.update_status(s)
|
||
else:
|
||
for s in db.get_sends(part_num, twn[0][0]):
|
||
try:
|
||
await bot.send_sticker(chat_id=s, sticker=config.ID_STICKERS + config.STICKERS[6])
|
||
await bot.send_message(chat_id=s, text=f"*Для {part} по {town.title()} актировка отсутствует!",
|
||
parse_mode=types.ParseMode.MARKDOWN)
|
||
except Exception as e:
|
||
print(repr(e))
|
||
db.update_status(s)
|
||
print("---------------------------------")
|
||
|
||
|
||
# ОБНОВЛЕНИЕ ДАННЫХ О ПОГОДЫ
|
||
async def wind(wait_for):
|
||
while True:
|
||
await asyncio.sleep(wait_for)
|
||
for moment in db.get_moments():
|
||
weather = info.weather(moment)
|
||
|
||
if type(weather) == dict:
|
||
for tm in weather:
|
||
db.update_weather(weather[tm], moment, tm)
|
||
else:
|
||
print(weather)
|
||
time.sleep(10)
|
||
|
||
|
||
# ОЧИСТКА ДАННЫХ ОБ АКТИРОВКАХ
|
||
async def clear(wait_for):
|
||
while True:
|
||
await asyncio.sleep(wait_for)
|
||
|
||
if (db.get_data()[0] != str(dt.now().date())) and (db.check_acta(1) or db.check_acta(2)):
|
||
db.del_acta()
|
||
|
||
|
||
# ЗАПУСК ЛОНГ ПОЛЛИНГА
|
||
if __name__ == '__main__':
|
||
loop = asyncio.get_event_loop()
|
||
loop.create_task(timer(60)) # ПРОВЕРКА МЕТЕО НОРИЛЬСК КАЖДУЮ МИНУТУ
|
||
loop.create_task(clear(300)) # ПРОВЕРКА НА НОВЫЙ ДЕНЬ КАЖДЫЕ 10 МИНУТ
|
||
loop.create_task(wind(1200)) # ПРОВЕРКА НА ОБНОВЛЕНИЕ ПОГОДЫ КАЖДЫЕ 20 МИНУТ
|
||
executor.start_polling(dp, skip_updates=True)
|