Niken
62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
import hashlib
|
|
from datetime import datetime, timedelta
|
|
from typing import Optional, Tuple
|
|
from playwright.async_api import async_playwright, ViewportSize, FloatRect
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ScheduleService:
|
|
def __init__(self):
|
|
self.base_url = "https://college.by/accounts/raspis/{mouth:02d}/{day:02d}-PODNAM.htm"
|
|
|
|
def _make_url(self, day: int = 0) -> Tuple[str, int, int]:
|
|
"""Генерация URL для расписания"""
|
|
d = datetime.now()
|
|
if day == 0:
|
|
if d.hour >= 12:
|
|
d += timedelta(days=1)
|
|
if d.weekday() == 6:
|
|
d += timedelta(days=1)
|
|
return self.base_url.format(day=d.day, mouth=d.month), d.day, d.month
|
|
else:
|
|
return self.base_url.format(day=int(day), mouth=d.month), int(day), int(d.month)
|
|
|
|
async def get_schedule(self, group: str, day_offset: int = 0) -> Tuple[Optional[bytes], str, int, int]:
|
|
"""Получение скриншота расписания"""
|
|
url, day, month = self._make_url(day_offset)
|
|
|
|
async with async_playwright() as p:
|
|
browser = await p.chromium.launch(headless=True)
|
|
context = await browser.new_context(viewport=ViewportSize(width=400, height=3000))
|
|
page = await context.new_page()
|
|
|
|
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
|
|
|
|
locator = page.locator("p", has_text=group).first
|
|
if await locator.count() > 0:
|
|
await locator.scroll_into_view_if_needed()
|
|
box = await locator.bounding_box()
|
|
|
|
if box:
|
|
clip_rect = FloatRect(
|
|
x=float(max(box["x"] - 0, 0)),
|
|
y=float(max(box["y"] - 0, 0)),
|
|
width=float(box["width"] + 150),
|
|
height=float(box["height"] + 100)
|
|
)
|
|
return await page.screenshot(clip=clip_rect), url, day, month
|
|
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при получении расписания: {e}")
|
|
finally:
|
|
await context.close()
|
|
await browser.close()
|
|
|
|
return None, url, day, month |