barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 13

feat(scrapers): cses soft too

Browse source
This commit is contained in:
Barrett Ruth 2025-10-05 13:42:06 -04:00
parent ee88450b3b
commit 25fde26943
1 changed files with 12 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

29
scrapers/cses.py
View file

@ -221,23 +221,18 @@ class CSESScraper(BaseScraper):
html = await fetch_text(client, task_path(pid))
tests = parse_tests(html)
timeout_ms, memory_mb = parse_limits(html)
if not tests:
return {
"problem_id": pid,
"error": f"{self.platform_name}: no tests found",
}
return {
"problem_id": pid,
"tests": [
{"input": t.input, "expected": t.expected}
for t in tests
],
"timeout_ms": timeout_ms,
"memory_mb": memory_mb,
"interactive": False,
}
except Exception as e:
return {"problem_id": pid, "error": str(e)}
except Exception:
tests = []
timeout_ms, memory_mb = 0, 0
return {
"problem_id": pid,
"tests": [
{"input": t.input, "expected": t.expected} for t in tests
],
"timeout_ms": timeout_ms,
"memory_mb": memory_mb,
"interactive": False,
}
tasks = [run_one(p.id) for p in problems]
for coro in asyncio.as_completed(tasks):