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

fix: "cache" scrapes

Browse source
This commit is contained in:
Barrett Ruth 2025-09-12 23:06:26 -05:00
parent 20712dd427
commit 8201a11a42
2 changed files with 27 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

18
lua/cp/scrape.lua
View file

@ -41,6 +41,24 @@ end
function M.scrape_problem(contest, problem_id, problem_letter)
ensure_io_directory()
local full_problem_id = problem_id:lower()
if contest == "atcoder" or contest == "codeforces" then
if problem_letter then
full_problem_id = full_problem_id .. problem_letter:upper()
end
end
local input_file = "io/" .. full_problem_id .. ".in"
local expected_file = "io/" .. full_problem_id .. ".expected"
if vim.fn.filereadable(input_file) == 1 and vim.fn.filereadable(expected_file) == 1 then
return {
success = true,
problem_id = full_problem_id,
test_count = 1,
}
end
if not setup_python_env() then
return {
success = false,