From bf4f7797b006185cbf62cdc60e3419a6887d2921 Mon Sep 17 00:00:00 2001 From: Barrett Ruth Date: Fri, 6 Mar 2026 21:34:43 -0500 Subject: [PATCH] ci: format --- lua/cp/pickers/fzf_lua.lua | 5 +- lua/cp/pickers/telescope.lua | 5 +- lua/cp/submit.lua | 5 +- lua/cp/utils.lua | 15 ++++- scrapers/atcoder.py | 121 ++++++++++++++++++++++++++++------- 5 files changed, 121 insertions(+), 30 deletions(-) diff --git a/lua/cp/pickers/fzf_lua.lua b/lua/cp/pickers/fzf_lua.lua index 218ba69..65a600a 100644 --- a/lua/cp/pickers/fzf_lua.lua +++ b/lua/cp/pickers/fzf_lua.lua @@ -10,7 +10,10 @@ local function contest_picker(platform, refresh, language) local contests = picker_utils.get_platform_contests(platform, refresh) if vim.tbl_isempty(contests) then - logger.log(("No contests found for platform '%s'"):format(platform_display_name), { level = vim.log.levels.WARN }) + logger.log( + ("No contests found for platform '%s'"):format(platform_display_name), + { level = vim.log.levels.WARN } + ) return end diff --git a/lua/cp/pickers/telescope.lua b/lua/cp/pickers/telescope.lua index b0a00b3..890ae6a 100644 --- a/lua/cp/pickers/telescope.lua +++ b/lua/cp/pickers/telescope.lua @@ -15,7 +15,10 @@ local function contest_picker(opts, platform, refresh, language) local contests = picker_utils.get_platform_contests(platform, refresh) if vim.tbl_isempty(contests) then - logger.log(('No contests found for platform: %s'):format(platform_display_name), { level = vim.log.levels.WARN }) + logger.log( + ('No contests found for platform: %s'):format(platform_display_name), + { level = vim.log.levels.WARN } + ) return end diff --git a/lua/cp/submit.lua b/lua/cp/submit.lua index c95f95f..7302d1d 100644 --- a/lua/cp/submit.lua +++ b/lua/cp/submit.lua @@ -89,7 +89,10 @@ function M.submit(opts) creds, function(ev) vim.schedule(function() - logger.log(STATUS_MSGS[ev.status] or ev.status, { level = vim.log.levels.INFO, override = true }) + logger.log( + STATUS_MSGS[ev.status] or ev.status, + { level = vim.log.levels.INFO, override = true } + ) end) end, function(result) diff --git a/lua/cp/utils.lua b/lua/cp/utils.lua index e93161a..c27763f 100644 --- a/lua/cp/utils.lua +++ b/lua/cp/utils.lua @@ -142,7 +142,10 @@ local function discover_nix_submit_cmd() local plugin_path = M.get_plugin_path() vim.cmd.redraw() - logger.log('Building submit environment...', { level = vim.log.levels.INFO, override = true, sync = true }) + logger.log( + 'Building submit environment...', + { level = vim.log.levels.INFO, override = true, sync = true } + ) vim.cmd.redraw() local result = vim .system( @@ -209,7 +212,10 @@ local function discover_nix_python() end local plugin_path = M.get_plugin_path() - logger.log('Building Python environment with nix...', { level = vim.log.levels.INFO, override = true, sync = true }) + logger.log( + 'Building Python environment with nix...', + { level = vim.log.levels.INFO, override = true, sync = true } + ) vim.cmd.redraw() local result = vim .system( @@ -263,7 +269,10 @@ function M.setup_python_env() if not on_nixos and vim.fn.executable('uv') == 1 then local plugin_path = M.get_plugin_path() logger.log('Python env: uv sync (dir=' .. plugin_path .. ')') - logger.log('Setting up Python environment...', { level = vim.log.levels.INFO, override = true, sync = true }) + logger.log( + 'Setting up Python environment...', + { level = vim.log.levels.INFO, override = true, sync = true } + ) vim.cmd.redraw() local env = vim.fn.environ() diff --git a/scrapers/atcoder.py b/scrapers/atcoder.py index 92252fa..8645beb 100644 --- a/scrapers/atcoder.py +++ b/scrapers/atcoder.py @@ -37,28 +37,95 @@ from .timeouts import ( ) _LANGUAGE_ID_EXTENSION: dict[str, str] = { - "6002": "ada", "6003": "apl", "6004": "asm", "6005": "asm", - "6006": "awk", "6008": "sh", "6009": "bas", "6010": "bc", - "6012": "bf", "6013": "c", "6014": "c", - "6015": "cs", "6016": "cs", - "6017": "cc", "6021": "clj", "6022": "clj", "6023": "clj", "6025": "cljs", - "6026": "cob", "6027": "lisp", "6028": "cr", "6030": "d", "6031": "d", - "6032": "d", "6033": "dart", "6038": "ex", "6039": "el", "6041": "erl", - "6042": "fs", "6043": "factor", "6044": "fish", "6045": "fth", - "6046": "f90", "6047": "f90", "6048": "f", "6049": "gleam", - "6050": "go", "6051": "go", "6052": "hs", "6053": "hx", "6054": "cc", - "6056": "java", "6057": "js", "6058": "js", "6059": "js", - "6060": "jule", "6061": "kk", "6062": "kt", "6065": "lean", - "6066": "ll", "6067": "lua", "6068": "lua", "6071": "nim", "6072": "nim", - "6073": "ml", "6074": "m", "6075": "pas", "6076": "pl", "6077": "php", - "6079": "pony", "6080": "ps1", "6081": "pro", - "6082": "py", "6083": "py", "6084": "r", "6085": "re", - "6086": "rb", "6087": "rb", "6088": "rs", "6089": "py", - "6090": "scala", "6091": "scala", "6092": "scm", "6093": "scm", - "6094": "sd7", "6095": "swift", "6096": "tcl", - "6100": "ts", "6101": "ts", "6102": "ts", - "6105": "v", "6106": "vala", "6107": "v", "6109": "wat", - "6111": "zig", "6114": "jl", "6115": "py", "6116": "cc", "6118": "sql", + "6002": "ada", + "6003": "apl", + "6004": "asm", + "6005": "asm", + "6006": "awk", + "6008": "sh", + "6009": "bas", + "6010": "bc", + "6012": "bf", + "6013": "c", + "6014": "c", + "6015": "cs", + "6016": "cs", + "6017": "cc", + "6021": "clj", + "6022": "clj", + "6023": "clj", + "6025": "cljs", + "6026": "cob", + "6027": "lisp", + "6028": "cr", + "6030": "d", + "6031": "d", + "6032": "d", + "6033": "dart", + "6038": "ex", + "6039": "el", + "6041": "erl", + "6042": "fs", + "6043": "factor", + "6044": "fish", + "6045": "fth", + "6046": "f90", + "6047": "f90", + "6048": "f", + "6049": "gleam", + "6050": "go", + "6051": "go", + "6052": "hs", + "6053": "hx", + "6054": "cc", + "6056": "java", + "6057": "js", + "6058": "js", + "6059": "js", + "6060": "jule", + "6061": "kk", + "6062": "kt", + "6065": "lean", + "6066": "ll", + "6067": "lua", + "6068": "lua", + "6071": "nim", + "6072": "nim", + "6073": "ml", + "6074": "m", + "6075": "pas", + "6076": "pl", + "6077": "php", + "6079": "pony", + "6080": "ps1", + "6081": "pro", + "6082": "py", + "6083": "py", + "6084": "r", + "6085": "re", + "6086": "rb", + "6087": "rb", + "6088": "rs", + "6089": "py", + "6090": "scala", + "6091": "scala", + "6092": "scm", + "6093": "scm", + "6094": "sd7", + "6095": "swift", + "6096": "tcl", + "6100": "ts", + "6101": "ts", + "6102": "ts", + "6105": "v", + "6106": "vala", + "6107": "v", + "6109": "wat", + "6111": "zig", + "6114": "jl", + "6115": "py", + "6116": "cc", + "6118": "sql", } MIB_TO_MB = 1.048576 @@ -422,10 +489,16 @@ def _submit_headless( f'select[name="data.LanguageId"] option[value="{language_id}"]' ).wait_for(state="attached", timeout=BROWSER_ELEMENT_WAIT) page.select_option('select[name="data.LanguageId"]', language_id) - ext = _LANGUAGE_ID_EXTENSION.get(language_id, Path(file_path).suffix.lstrip(".") or "txt") + ext = _LANGUAGE_ID_EXTENSION.get( + language_id, Path(file_path).suffix.lstrip(".") or "txt" + ) page.set_input_files( "#input-open-file", - {"name": f"solution.{ext}", "mimeType": "text/plain", "buffer": Path(file_path).read_bytes()}, + { + "name": f"solution.{ext}", + "mimeType": "text/plain", + "buffer": Path(file_path).read_bytes(), + }, ) page.wait_for_timeout(BROWSER_SETTLE_DELAY) page.locator('button[type="submit"]').click(no_wait_after=True)