feat: add subscribtion scheduler with ai, pagination

This commit is contained in:
Faynot
2026-03-29 11:25:31 +03:00
parent 1db524f757
commit 164acd6307
16 changed files with 688 additions and 358 deletions

121
handlers/search.py Normal file
View File

@@ -0,0 +1,121 @@
import html
from aiogram import Router, F
from aiogram.filters import Command
from aiogram.types import Message, CallbackQuery
from database import Database
from kwork import get_kwork_projects
from ai import filter_vacancies_with_ai
from aiogram import F
from aiogram.types import CallbackQuery
import logging
from keyboards import get_pagination_kb
router = Router()
db = Database("users.db")
@router.message(Command("search"))
async def search_projects(message: Message):
args = message.text.split()
start_p, end_p = 1, 1
if len(args) == 3:
if args[1].isdigit() and args[2].isdigit():
start_p, end_p = int(args[1]), int(args[2])
elif len(args) == 2:
if args[1].isdigit():
start_p, end_p = int(args[1]), int(args[1])
user_data = await db.get_user(message.from_user.id)
if not user_data or user_data[0] is None:
await message.answer("⚠️ Твой профиль еще не настроен.")
return
sphere, lang, prefs = user_data
user_preferences = f"- Сфера: {sphere}\n- Стек: {lang}\n- Доп: {prefs}"
msg = await message.answer(f"⏳ <b>Собираю свежие проекты и анализирую их нейросетью...</b>\n<i>Это может занять около минуты.</i>")
try:
raw_vacancies = await get_kwork_projects(start_page=start_p, end_page=end_p)
if not raw_vacancies:
await msg.edit_text("❌ Проектов не найдено.")
return
filtered_vacancies = await filter_vacancies_with_ai(raw_vacancies, user_preferences)
if not filtered_vacancies:
await msg.edit_text("😔 Подходящих проектов сейчас нет.")
return
await msg.delete()
await message.answer(f"🎯 <b>Найдено {len(filtered_vacancies)} подходящих проектов:</b>")
for vac in filtered_vacancies:
title = html.escape(vac.get('title', 'Без названия'))
price = html.escape(vac.get('price', 'По договоренности'))
desc = html.escape(vac.get('description', ''))
url = vac.get('url', '#')
text = (
f"💼 <b>{title}</b>\n\n"
f"💰 <b>Бюджет:</b> {price}\n"
f"📝 <b>Описание:</b> {desc}...\n\n"
f"🔗 <a href='{url}'>Смотреть на Kwork</a>"
)
# Отправляем, отключая превью ссылок, чтобы не захламлять чат
await message.answer(text, disable_web_page_preview=True)
except Exception as e:
print(f"Ошибка в поиске: {e}")
await msg.edit_text("❌ Произошла ошибка при анализе проектов. Попробуй еще раз чуть позже.")
async def build_pages_text(page_num: int):
raw_vacancies = await get_kwork_projects(start_page=page_num, end_page=page_num)
if not raw_vacancies:
return "На этой странице проектов не найдено."
text = f"📂 <b>Все проекты (Страница {page_num}):</b>\n\n"
for i, vac in enumerate(raw_vacancies, 1):
title = html.escape(vac.get('title', 'Без названия'))
price = html.escape(vac.get('price', 'По договоренности'))
url = vac.get('url', '#')
desc = html.escape(vac.get('description', ''))[:100] + "..."
text += f"{i}. <a href='{url}'>{title}</a>\n💰 {price}\n📝 {desc}\n\n"
return text
@router.message(Command("all"))
async def command_all_vancancies(message: Message):
wait_msg = await message.answer("⏳ Загружаю список проектов...")
try:
page = 1
content = await build_pages_text(page)
await wait_msg.edit_text(
content,
reply_markup=get_pagination_kb(page),
disable_web_page_preview=True
)
except Exception as e:
logging.error(f"Error in /all: {e}")
await wait_msg.edit_text("⚠️ Ошибка при загрузке данных.")
# Обработка нажатий на кнопки пагинации
@router.callback_query(F.data.startswith("browse_"))
async def process_pagination(callback: CallbackQuery):
page = int(callback.data.split("_")[1])
await callback.message.edit_text("🔄 Обновляю список...")
try:
content = await build_pages_text(page)
await callback.message.edit_text(
content,
reply_markup=get_pagination_kb(page),
disable_web_page_preview=True
)
await callback.answer()
except Exception as e:
logging.error(f"Error in pagination: {e}")
await callback.answer("Ошибка при смене страницы", show_alert=True)