{body}"
+
+ async def is_published_for(self, day_offset: int = 0) -> bool:
+ target = self._resolve_target_date(day_offset)
+ return await self.drive.find_for_date(target) is not None
+
+ async def get_schedule(
+ self, group: str, day_offset: int = 0
+ ) -> Tuple[str, str, int, int]:
+ pdf_bytes, url, day, month = await self._load_pdf_for_date(day_offset)
+
+ folder_url = "https://drive.google.com/drive/folders/" + self.drive.folder_id
+
+ if not pdf_bytes:
+ result = (
+ f"⚠️ Расписание на {day:02d}.{month:02d} ещё не опубликовано "
+ f"в Google Drive"
+ )
+ return result, folder_url, day, month
+
+ schedule_lines = self._parse_group_lines(
+ self._extract_pdf_lines(pdf_bytes), group
+ )
+
+ if not schedule_lines:
+ result = f"⚠️ Расписание для группы {escape(group)} на {day} число не найдено"
+ else:
+ result = self._format_schedule_html(day, schedule_lines)
+
+ return result, url or folder_url, day, month
async def get_pschedule(
- self, group: str, day_offset: int = 0
+ self, group: str, day_offset: int = 0
) -> Tuple[Optional[bytes], str, int, int]:
- url, day, month = self._make_url(day_offset)
+ pdf_bytes, url, day, month = await self._load_pdf_for_date(day_offset)
+ fallback_url = (
+ url
+ or "https://drive.google.com/drive/folders/" + self.drive.folder_id
+ )
- async with async_playwright() as p:
- browser = await p.chromium.launch(headless=True)
- context = await browser.new_context(viewport={"width": 400, "height": 3000})
- page = await context.new_page()
+ if not pdf_bytes:
+ return None, fallback_url, day, month
- try:
- response = await page.goto(url, wait_until="networkidle", timeout=30000)
- if not response or response.status != 200:
- logger.warning(f"Ошибка загрузки страницы: {url}")
- return None, url, day, month
+ try:
+ doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+ regex = self.exact_group_regex(group)
- # 1) сначала пытаемся по более точному селектору (как в HTML-парсере)
- candidates = page.locator("p.MsoPlainText b")
- count = await candidates.count()
+ for page in doc:
+ line_items = []
+ page_dict = page.get_text("dict")
+ for block in page_dict.get("blocks", []):
+ if block.get("type") != 0:
+ continue
+ for line in block.get("lines", []):
+ text = "".join(span["text"] for span in line["spans"]).strip()
+ if text:
+ line_items.append((text, fitz.Rect(line["bbox"])))
- regex = self.exact_group_regex(group)
- target_handle = None
-
- for i in range(count):
- el = candidates.nth(i)
- text = (await el.inner_text()).strip()
- if regex.search(text):
- # нашли b с нужной группой — возьмём родительский p для удобного скрина
- parent_p = await el.locator("xpath=ancestor::p[1]").element_handle()
- target_handle = parent_p or await el.element_handle()
- break
-
- # 2) если не нашли в p.MsoPlainText b, попробуем просто p b или p
- if not target_handle:
- candidates = page.locator("p b")
- count = await candidates.count()
- for i in range(count):
- el = candidates.nth(i)
- text = (await el.inner_text()).strip()
+ found_group = False
+ rects: List[fitz.Rect] = []
+ for text, bbox in line_items:
+ if not found_group:
if regex.search(text):
- parent_p = await el.locator("xpath=ancestor::p[1]").element_handle()
- target_handle = parent_p or await el.element_handle()
+ found_group = True
+ rects.append(bbox)
+ else:
+ if "-----" in text or "+----" in text:
break
+ rects.append(bbox)
- if not target_handle:
- # последний шанс: любые - candidates = page.locator("p") - count = await candidates.count() - for i in range(count): - el = candidates.nth(i) - text = (await el.inner_text()).strip() - if regex.search(text): - target_handle = await el.element_handle() - break + if not rects: + continue - if target_handle: - # скроллим и получаем box - await target_handle.scroll_into_view_if_needed() - box = await target_handle.bounding_box() - if box: - clip_rect = { - "x": float(max(box["x"], 0)), - "y": float(max(box["y"], 0)), - "width": float(box["width"] + 150), - "height": float(box["height"] + 100), - } - img = await page.screenshot(clip=clip_rect) - return img, url, day, month + clip = rects[0] + for rect in rects[1:]: + clip |= rect + clip.x0 = max(0, clip.x0 - 10) + clip.x1 = min(page.rect.width, clip.x1 + 150) + clip.y0 = max(0, clip.y0 - 5) + clip.y1 = min(page.rect.height, clip.y1 + 10) - except Exception as e: - logger.error(f"Ошибка при получении расписания: {e}") - finally: - await context.close() - await browser.close() + pixmap = page.get_pixmap(clip=clip, matrix=fitz.Matrix(2, 2)) + return pixmap.tobytes("png"), fallback_url, day, month - return None, url, day, month + except Exception as e: + logger.error(f"Ошибка при получении расписания из PDF: {e}") + return None, fallback_url, day, month diff --git a/services/watcher_service.py b/services/watcher_service.py index 00044f6..c2df54c 100644 --- a/services/watcher_service.py +++ b/services/watcher_service.py @@ -1,12 +1,12 @@ import asyncio -from datetime import datetime, timedelta from random import randint -from aiogram import Bot, types -from models.state import BotState -from config import Config -from services.schedule_service import ScheduleService -from logging import getLogger +from aiogram import Bot, types + +from config import Config +from logging import getLogger +from models.state import BotState +from services.schedule_service import ScheduleService logger = getLogger(__name__) @@ -40,21 +40,24 @@ class WatcherService: pass logger.info("Watcher остановлен") + @staticmethod + def _next_delay() -> int: + return Config.WATCHER_INTERVAL_SEC + randint( + Config.WATCHER_RANDOM_DELAY_MIN, + Config.WATCHER_RANDOM_DELAY_MAX, + ) + async def _watcher_loop(self): - """Основной цикл слежки""" + """Основной цикл слежки за появлением PDF на Google Drive.""" while self.state.watcher_work: try: - find = await self._check_all_groups() - if find: - # ничего не нашли → ждём - delay = randint( - Config.WATCHER_BASE_DELAY, Config.WATCHER_BASE_DELAY + 30 - ) - logger.info(f"Следующая проверка через {delay}") + nothing_found = await self._check_all_groups() + if nothing_found: + delay = self._next_delay() + logger.info(f"PDF/расписание не найдено, следующая проверка через {delay} с") await asyncio.sleep(delay) else: - # нашли → останавливаемся - logger.info("Расписание найдено, останавливаем watcher") + logger.info("Расписание найдено и отправлено, останавливаем watcher") self.state.watcher_work = False break except asyncio.CancelledError: @@ -63,61 +66,63 @@ class WatcherService: logger.error(f"Ошибка в watcher_loop: {e}") await asyncio.sleep(60) - @staticmethod - def _get_target_day() -> datetime: - """Получение целевого дня""" - now = datetime.now() - target = now + timedelta(days=1) - if target.weekday() == 6: - target += timedelta(days=1) - return target - async def _check_all_groups(self) -> bool: """ - Возвращает True, если НИ в одной группе не найдено расписание. - Возвращает False, если хотя бы в одной группе найдено расписание. + Возвращает True, если расписание ещё недоступно ни для одной группы. + Возвращает False, если хотя бы одной группе отправили расписание. """ - day = self._get_target_day() - found_any = False + target = self.schedule_service._resolve_target_date(0) + logger.info( + f"Проверяем Google Drive на расписание за {target.strftime('%d.%m.%Y')}" + ) + if not await self.schedule_service.is_published_for(0): + return True + + found_any = False for group, chat_id in Config.GROUP_CHATS.items(): logger.info( - f"Проверяем расписание для {group} на {day.strftime('%d.%m.%Y')}" + f"Проверяем расписание для {group} на {target.strftime('%d.%m.%Y')}" ) - found = await self._check_group_schedule(group, chat_id, day.day) - if found: + if await self._check_group_schedule(group, chat_id): found_any = True - return not found_any # <-- вот так правильно + return not found_any - async def _check_group_schedule(self, group: str, chat_id: int, day: int) -> bool: + async def _check_group_schedule(self, group: str, chat_id: int) -> bool: text, url, data_day, data_month = await self.schedule_service.get_schedule( - group, day + group, 0 ) - if text and "не найдено" not in text.lower(): + if not self.schedule_service.is_schedule_missing(text): msg = await self.bot.send_message( chat_id, - f"Авто-расписание для {group} на {data_day:02d}.{data_month:02d}\n\n{text}", - parse_mode="Markdown", + ( + f"🔔 Авто-расписание для {group} " + f"на {data_day:02d}.{data_month:02d}\n\n{text}" + ), + parse_mode="HTML", ) - await self.bot.pin_chat_message( - chat_id, msg.message_id, disable_notification=False + try: + await self.bot.pin_chat_message( + chat_id, msg.message_id, disable_notification=False + ) + except Exception as e: + logger.warning(f"Не удалось закрепить сообщение в {chat_id}: {e}") + return True + + png, url, data_day, data_month = await self.schedule_service.get_pschedule( + group, 0 + ) + if png: + await self.bot.send_photo( + chat_id, + types.BufferedInputFile(png, filename=f"{group}.png"), + caption=( + f"🔔 АВАРИЙНЫЙ РЕЖИМ\n\n" + f"Авто-расписание для {group} " + f"на {data_day:02d}.{data_month:02d}" + ), ) return True - else: - png, url, data_day, data_month = await self.schedule_service.get_pschedule( - group, day - ) - if png: - await self.bot.send_photo( - chat_id, - types.BufferedInputFile(png, filename=f"{group}.png"), - caption=f"АВАРИЙНЫЙ РЕЖИМ\n\nАвто-расписание для {group} на {data_day:02d}.{data_month:02d}\n\nНайдено с ошибкой", - ) - return True + return False - - # clip_hash = hashlib.md5(clip_png).hexdigest() - - # Логика проверки изменений и отправки сообщений - # ... (ваша существующая логика)