73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import Any, Awaitable, Callable, ParamSpec, cast
|
|
|
|
from .models import ContestListResult, MetadataResult, TestsResult
|
|
|
|
P = ParamSpec("P")
|
|
|
|
|
|
class BaseScraper(ABC):
|
|
@property
|
|
@abstractmethod
|
|
def platform_name(self) -> str: ...
|
|
|
|
@abstractmethod
|
|
async def scrape_contest_metadata(self, contest_id: str) -> MetadataResult: ...
|
|
|
|
@abstractmethod
|
|
async def scrape_contest_list(self) -> ContestListResult: ...
|
|
|
|
@abstractmethod
|
|
async def stream_tests_for_category_async(self, category_id: str) -> None: ...
|
|
|
|
def _create_metadata_error(
|
|
self, error_msg: str, contest_id: str = ""
|
|
) -> MetadataResult:
|
|
return MetadataResult(
|
|
success=False,
|
|
error=f"{self.platform_name}: {error_msg}",
|
|
contest_id=contest_id,
|
|
problems=[],
|
|
url="",
|
|
)
|
|
|
|
def _create_tests_error(
|
|
self, error_msg: str, problem_id: str = "", url: str = ""
|
|
) -> TestsResult:
|
|
return TestsResult(
|
|
success=False,
|
|
error=f"{self.platform_name}: {error_msg}",
|
|
problem_id=problem_id,
|
|
tests=[],
|
|
timeout_ms=0,
|
|
memory_mb=0,
|
|
interactive=False,
|
|
)
|
|
|
|
def _create_contests_error(self, error_msg: str) -> ContestListResult:
|
|
return ContestListResult(
|
|
success=False,
|
|
error=f"{self.platform_name}: {error_msg}",
|
|
contests=[],
|
|
)
|
|
|
|
async def _safe_execute(
|
|
self,
|
|
operation: str,
|
|
func: Callable[P, Awaitable[Any]],
|
|
*args: P.args,
|
|
**kwargs: P.kwargs,
|
|
):
|
|
try:
|
|
return await func(*args, **kwargs)
|
|
except Exception as e:
|
|
if operation == "metadata":
|
|
contest_id = cast(str, args[0]) if args else ""
|
|
return self._create_metadata_error(str(e), contest_id)
|
|
elif operation == "tests":
|
|
problem_id = cast(str, args[1]) if len(args) > 1 else ""
|
|
return self._create_tests_error(str(e), problem_id)
|
|
elif operation == "contests":
|
|
return self._create_contests_error(str(e))
|
|
else:
|
|
raise
|