189 lines
7.9 KiB
Python
189 lines
7.9 KiB
Python
import asyncio
|
||
import logging
|
||
import os
|
||
from datetime import datetime
|
||
|
||
from aiogram import Bot, Dispatcher, F, types
|
||
from aiogram.filters import Command
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton, ReplyKeyboardRemove
|
||
from aiogram.client.default import DefaultBotProperties
|
||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||
from dotenv import load_dotenv
|
||
|
||
from weather import get_daily_weather_summary, get_coords_by_city
|
||
import database as db # Импортируем нашу базу данных
|
||
|
||
load_dotenv()
|
||
BOT_TOKEN = os.getenv("BOT_TOKEN")
|
||
|
||
logging.basicConfig(level=logging.INFO)
|
||
|
||
bot = Bot(token=BOT_TOKEN, default=DefaultBotProperties(parse_mode="Markdown"))
|
||
dp = Dispatcher()
|
||
scheduler = AsyncIOScheduler()
|
||
|
||
class WeatherSetup(StatesGroup):
|
||
waiting_for_location = State()
|
||
waiting_for_time = State()
|
||
|
||
def get_main_keyboard():
|
||
return ReplyKeyboardMarkup(
|
||
keyboard=[
|
||
[KeyboardButton(text="🌤 Погода сейчас")],
|
||
[
|
||
KeyboardButton(text="📍 Изменить город"),
|
||
KeyboardButton(text="⏰ Изменить время")
|
||
]
|
||
],
|
||
resize_keyboard=True
|
||
)
|
||
|
||
def get_location_keyboard():
|
||
return ReplyKeyboardMarkup(
|
||
keyboard=[[KeyboardButton(text="📍 Отправить текущую геопозицию", request_location=True)]],
|
||
resize_keyboard=True,
|
||
input_field_placeholder="Или напишите название города..."
|
||
)
|
||
|
||
async def send_weather_update(user_id: int):
|
||
"""Отправка погоды по расписанию"""
|
||
user = await db.get_user(user_id)
|
||
if not user or not user["lat"]:
|
||
return
|
||
|
||
weather_text = await get_daily_weather_summary(user["lat"], user["lon"])
|
||
try:
|
||
await bot.send_message(chat_id=user_id, text=weather_text)
|
||
except Exception as e:
|
||
logging.error(f"Ошибка отправки пользователю {user_id}: {e}")
|
||
|
||
@dp.message(Command("start"))
|
||
async def cmd_start(message: types.Message, state: FSMContext):
|
||
await state.set_state(WeatherSetup.waiting_for_location)
|
||
await message.answer(
|
||
"Привет! Я погодный бот.\n\nОтправь мне свою геопозицию кнопкой ниже или **просто напиши название своего города**:",
|
||
reply_markup=get_location_keyboard()
|
||
)
|
||
|
||
@dp.message(F.text == "📍 Изменить город")
|
||
async def cmd_change_city(message: types.Message, state: FSMContext):
|
||
await state.set_state(WeatherSetup.waiting_for_location)
|
||
await message.answer("Напиши название нового города или отправь геопозицию:", reply_markup=get_location_keyboard())
|
||
|
||
@dp.message(WeatherSetup.waiting_for_location)
|
||
async def process_location_input(message: types.Message, state: FSMContext):
|
||
user_id = message.from_user.id
|
||
city_name_display = "Выбранная локация"
|
||
|
||
if message.location:
|
||
lat = message.location.latitude
|
||
lon = message.location.longitude
|
||
elif message.text:
|
||
lat, lon, found_name = await get_coords_by_city(message.text)
|
||
if not lat:
|
||
await message.answer("К сожалению, я не нашел такой город 😔. Попробуй уточнить название.")
|
||
return
|
||
city_name_display = found_name
|
||
else:
|
||
return
|
||
|
||
# Сохраняем локацию в БД
|
||
await db.save_user_location(user_id, lat, lon)
|
||
|
||
# Проверяем, есть ли уже у пользователя настроенное время
|
||
user = await db.get_user(user_id)
|
||
if user and user["time"]:
|
||
await message.answer(f"📍 Локация успешно обновлена на: **{city_name_display}**!", reply_markup=get_main_keyboard())
|
||
await state.clear()
|
||
else:
|
||
await state.set_state(WeatherSetup.waiting_for_time)
|
||
await message.answer(
|
||
f"Отлично! Локация **{city_name_display}** найдена.\n\nТеперь напиши время (в формате ЧЧ:ММ):",
|
||
reply_markup=ReplyKeyboardRemove()
|
||
)
|
||
|
||
@dp.message(F.text == "⏰ Изменить время")
|
||
async def cmd_change_time(message: types.Message, state: FSMContext):
|
||
user = await db.get_user(message.from_user.id)
|
||
if not user:
|
||
await message.answer("Сначала отправьте локацию через команду /start.")
|
||
return
|
||
|
||
await state.set_state(WeatherSetup.waiting_for_time)
|
||
await message.answer("Введите новое время для ежедневного прогноза (в формате ЧЧ:ММ):", reply_markup=ReplyKeyboardRemove())
|
||
|
||
@dp.message(WeatherSetup.waiting_for_time, F.text)
|
||
async def process_time(message: types.Message, state: FSMContext):
|
||
time_text = message.text.strip()
|
||
user_id = message.from_user.id
|
||
|
||
try:
|
||
dt = datetime.strptime(time_text, "%H:%M")
|
||
|
||
# Сохраняем время в БД
|
||
await db.update_user_time(user_id, time_text)
|
||
|
||
# Обновляем или создаем задачу. id задачи = строковый user_id.
|
||
# replace_existing=True автоматически заменит старую задачу, если она была.
|
||
scheduler.add_job(
|
||
send_weather_update,
|
||
trigger='cron',
|
||
hour=dt.hour,
|
||
minute=dt.minute,
|
||
args=[user_id],
|
||
id=str(user_id),
|
||
replace_existing=True
|
||
)
|
||
|
||
await message.answer(f"✅ Готово! Ежедневный прогноз установлен на {time_text}.", reply_markup=get_main_keyboard())
|
||
await state.clear()
|
||
|
||
except ValueError:
|
||
await message.answer("Неверный формат времени. Пожалуйста, используй формат ЧЧ:ММ.")
|
||
|
||
@dp.message(F.text == "🌤 Погода сейчас")
|
||
async def cmd_weather_now(message: types.Message):
|
||
user = await db.get_user(message.from_user.id)
|
||
|
||
if not user or not user["lat"]:
|
||
await message.answer("Сначала отправьте локацию через команду /start.")
|
||
return
|
||
|
||
loading_msg = await message.answer("⏳ Узнаю погоду...")
|
||
weather_text = await get_daily_weather_summary(user["lat"], user["lon"])
|
||
await loading_msg.edit_text(weather_text)
|
||
|
||
# --- ВОССТАНОВЛЕНИЕ ПЛАНИРОВЩИКА ---
|
||
async def restore_jobs():
|
||
"""Загружает задачи из БД в планировщик после перезапуска бота"""
|
||
users = await db.get_all_users_with_time()
|
||
count = 0
|
||
for user in users:
|
||
try:
|
||
dt = datetime.strptime(user["time"], "%H:%M")
|
||
scheduler.add_job(
|
||
send_weather_update,
|
||
trigger='cron',
|
||
hour=dt.hour,
|
||
minute=dt.minute,
|
||
args=[user["user_id"]],
|
||
id=str(user["user_id"]),
|
||
replace_existing=True
|
||
)
|
||
count += 1
|
||
except Exception as e:
|
||
logging.error(f"Не удалось восстановить задачу для {user['user_id']}: {e}")
|
||
logging.info(f"Восстановлено задач из БД: {count}")
|
||
|
||
async def main():
|
||
await db.init_db() # 1. Инициализируем БД
|
||
await restore_jobs() # 2. Восстанавливаем расписание из БД
|
||
scheduler.start() # 3. Запускаем планировщик
|
||
|
||
await bot.delete_webhook(drop_pending_updates=True)
|
||
await dp.start_polling(bot) # 4. Запускаем бота
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main()) |