mirror of
https://github.com/EDeev/npr_meteobot.git
synced 2026-06-15 11:01:06 +03:00
start file
This commit is contained in:
Egor Deev
GitHub
parent
5ded1fd8df
commit
6a53ffcfd6
1 changed files with 262 additions and 0 deletions
262
bot.py
Normal file
262
bot.py
Normal file
|
|
@ -0,0 +1,262 @@
|
||||||
|
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)
|
||||||
Loading…
Add table
Reference in a new issue