feat: add CAPTCHA detection for submissions (#311)

* feat: add CAPTCHA detection for submissions

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

* Fix styles

---------

Co-authored-by: Claude <[email protected]>

Author: kyuridenamida

atcodertools/client/atcoder.py

@@ -25,6 +25,10 @@ class PageNotFoundError(Exception):
     pass
 
 
+class CaptchaError(Exception):
+    pass
+
+
 default_cookie_path = get_cache_file_path('cookie.txt')
 
 
@@ -167,6 +171,10 @@ def submit_source_code(self, contest: Contest, problem: Problem, lang: Union[str
 
         resp = self._request(contest.get_submit_url())
 
+        if "https://challenges.cloudflare.com" in resp.text:
+            raise CaptchaError(
+                "CAPTCHA detected. Cannot submit automatically. Please submit manually through the web interface.")
+
         soup = BeautifulSoup(resp.text, "html.parser")
         session_id = soup.find("input", attrs={"type": "hidden"}).get("value")
         task_select_area = soup.find(

atcodertools/tools/submit.py

@@ -8,7 +8,7 @@
 from atcodertools.tools.tester import USER_FACING_JUDGE_TYPE_LIST, DEFAULT_EPS
 from atcodertools.tools.utils import with_color
 
-from atcodertools.client.atcoder import AtCoderClient, LoginError
+from atcodertools.client.atcoder import AtCoderClient, LoginError, CaptchaError
 from atcodertools.tools import tester
 from atcodertools.common.logging import logger
 from atcodertools.config.config import Config, ConfigType, USER_CONFIG_PATH
@@ -170,11 +170,15 @@ def main(prog, args, credential_supplier=None, use_local_session_cache=True, cli
                 logger.warning("code wasn't recognized as {}".format(encoding))
         logger.info(
             "Submitting {} as {}".format(code_path, metadata.lang.name))
-        submission = client.submit_source_code(
-            metadata.problem.contest, metadata.problem, metadata.lang, source)
-        logger.info("{} {}".format(
-            with_color("Done!", Fore.LIGHTGREEN_EX),
-            metadata.problem.contest.get_submissions_url(submission)))
+        try:
+            submission = client.submit_source_code(
+                metadata.problem.contest, metadata.problem, metadata.lang, source)
+            logger.info("{} {}".format(
+                with_color("Done!", Fore.LIGHTGREEN_EX),
+                metadata.problem.contest.get_submissions_url(submission)))
+        except CaptchaError as e:
+            logger.error(with_color(str(e), Fore.LIGHTRED_EX))
+            return False
         if config.submit_config.exec_after_submit:
             run_command(config.submit_config.exec_after_submit, args.dir)
 

tests/resources/test_atcoder_client_mock/submit/after_get.html

@@ -1084,12 +1084,6 @@ <h2>提出</h2>
                     <div class="form-group">
                         <label class="control-label col-sm-3 col-md-2"></label>
                         <div class="col-sm-5">
-
-                            <script src="https://challenges.cloudflare.com/turnstile/v0/api.js" async defer></script>
-
-                            <div class="cf-challenge" data-sitekey="0x4AAAAAAA6HJUmmLP7mLxx0" data-theme="light"></div>
-
-
                             <button id="submit" type="submit" class="btn btn-primary">提出</button>
                         </div>
                     </div>