AtCoderStudyBooster 0.4.0__py3-none-any.whl → 0.4.1__py3-none-any.whl

atcdr/test.py

@@ -26,6 +26,7 @@ from atcdr.util.filetype import (
     detect_language,
     lang2str,
 )
+from atcdr.util.i18n import _
 from atcdr.util.parse import ProblemHTML
 
 
@@ -159,7 +160,7 @@ class TestRunner:
             ]
             self.exe = None
         else:
-            raise ValueError(f'{lang}の適切な言語のランナーが見つかりませんでした.')
+            raise ValueError(_('runner_not_found', lang))
 
         return self
 
@@ -300,9 +301,9 @@ def create_renderable_test_info(
     status_text = STATUS_TEXT_MAP[test_info.summary]
 
     header_text = Text.assemble(
-        Text.from_markup(f'[cyan]{test_info.sourcename}[/]のテスト \n'),
+        Text.from_markup(f'[cyan]{test_info.sourcename}[/]' + _('test_of', '') + '\n'),
         Text.from_markup(
-            f'[italic #0f0f0f]コンパイルにかかった時間: [not italic cyan]{test_info.compile_time}[/] ms[/]\n'
+            '[italic #0f0f0f]' + _('compile_time', test_info.compile_time) + '\n'
         )
         if test_info.compile_time
         else Text(''),
@@ -319,7 +320,7 @@ def create_renderable_test_info(
 
     if test_info.compiler_message:
         rule = Rule(
-            title='コンパイラーのメッセージ',
+            title=_('compiler_message'),
             style=COLOR_MAP[ResultStatus.CE],
         )
         components.append(rule)
@@ -342,27 +343,29 @@ def create_renderable_test_result(
 
     # 以下の部分は if-else ブロックの外に移動
     status_header = Text.assemble(
-        'ステータス ',
+        _('status'),
         STATUS_TEXT_MAP[test_result.result.passed],  # status_text をここに追加
     )
 
     execution_time_text = None
     if test_result.result.executed_time is not None:
         execution_time_text = Text.from_markup(
-            f'実行時間   [cyan]{test_result.result.executed_time}[/cyan] ms'
+            _('execution_time', test_result.result.executed_time)
         )
 
     table = Table(show_header=True, header_style='bold')
-    table.add_column('入力', style='cyan', min_width=10)
+    table.add_column(_('input'), style='cyan', min_width=10)
 
     if test_result.result.passed != ResultStatus.AC:
         table.add_column(
-            '出力',
+            _('output'),
             style=COLOR_MAP[test_result.result.passed],
             min_width=10,
             overflow='fold',
         )
-        table.add_column('正解の出力', style=COLOR_MAP[ResultStatus.AC], min_width=10)
+        table.add_column(
+            _('expected_output'), style=COLOR_MAP[ResultStatus.AC], min_width=10
+        )
         table.add_row(
             escape(test_result.testcase.input),
             escape(test_result.result.output),
@@ -370,7 +373,7 @@ def create_renderable_test_result(
         )
     else:
         table.add_column(
-            '出力', style=COLOR_MAP[test_result.result.passed], min_width=10
+            _('output'), style=COLOR_MAP[test_result.result.passed], min_width=10
         )
         table.add_row(
             escape(test_result.testcase.input), escape(test_result.result.output)
@@ -393,7 +396,9 @@ def render_results(test: TestRunner) -> None:
         SpinnerColumn(style='white', spinner_name='simpleDots'),
         BarColumn(),
     )
-    task_id = progress.add_task(description='テスト進行中', total=test.info.case_number)
+    task_id = progress.add_task(
+        description=_('test_in_progress'), total=test.info.case_number
+    )
 
     current_display = [create_renderable_test_info(test.info, progress)]
 
@@ -404,7 +409,9 @@ def render_results(test: TestRunner) -> None:
             current_display.insert(-1, (create_renderable_test_result(i, result)))
             live.update(Group(*current_display))
 
-        progress.update(task_id, description='テスト完了')  # 完了メッセージに更新
+        progress.update(
+            task_id, description=_('test_completed')
+        )  # 完了メッセージに更新
         current_display[-1] = create_renderable_test_info(test.info, progress)
         live.update(Group(*current_display))
 
@@ -412,9 +419,7 @@ def render_results(test: TestRunner) -> None:
 def run_test(path_of_code: str) -> None:
     html_paths = [f for f in os.listdir('.') if f.endswith('.html')]
     if not html_paths:
-        print(
-            '問題のファイルが見つかりません。\n問題のファイルが存在するディレクトリーに移動してから実行してください。'
-        )
+        print(_('problem_file_not_found'))
         return
 
     with open(html_paths[0], 'r') as file:
@@ -425,7 +430,7 @@ def run_test(path_of_code: str) -> None:
     render_results(test)
 
 
-@click.command(short_help='テストを実行')
+@click.command(short_help=_('cmd_test'), help=_('cmd_test'))
 @add_file_selector('files', filetypes=COMPILED_LANGUAGES + INTERPRETED_LANGUAGES)
 def test(files):
     """指定したソースコードをサンプルケースでテストします。"""

atcdr/util/fileops.py

@@ -7,6 +7,7 @@ import questionary as q
 import rich_click as click
 
 from atcdr.util.filetype import FILE_EXTENSIONS, Lang
+from atcdr.util.i18n import _
 
 
 def collect_files(
@@ -41,9 +42,9 @@ def collect_files(
 
 def select_files_interactively(files: List[str]) -> List[str]:
     target_file = q.select(
-        message='複数のファイルが見つかりました.ファイルを選択してください:',
+        message=_('multiple_files_found'),
         choices=[q.Choice(title=file, value=file) for file in files],
-        instruction='\n 十字キーで移動, [enter]で実行',
+        instruction='\n ' + _('navigate_with_arrows'),
         pointer='❯',
         qmark='',
         style=q.Style(
@@ -78,7 +79,7 @@ def add_file_selector(
             # 2) ファイル収集 (非再帰固定)
             files = collect_files(patterns, tuple(exts), recursive=False)
             if not files:
-                click.echo('対象ファイルが見つかりません。')
+                click.echo(_('target_file_not_found'))
                 ctx.exit(1)
 
             # 3) 候補が1つなら即実行
@@ -89,7 +90,7 @@ def add_file_selector(
             if not patterns:
                 selected = select_files_interactively(files)
                 if not selected:
-                    click.echo('ファイルが選択されませんでした。')
+                    click.echo(_('file_not_selected'))
                     ctx.exit(1)
                 selected_list = [selected]
                 return ctx.invoke(f, **{arg_name: selected_list}, **kwargs)

atcdr/util/gpt.py

@@ -4,6 +4,8 @@ from typing import Dict, List, Optional
 
 import requests
 
+from atcdr.util.i18n import _
+
 
 class Model(Enum):
     GPT4O = 'gpt-4o'
@@ -24,27 +26,23 @@ def set_api_key() -> Optional[str]:
     if api_key and validate_api_key(api_key):
         return api_key
     elif api_key:
-        print('環境変数に設定されているAPIキーの検証に失敗しました ')
+        print(_('api_key_validation_failed'))
     else:
         pass
 
-    api_key = input(
-        'https://platform.openai.com/api-keys からchatGPTのAPIキーを入手しましょう。\nAPIキー入力してください: '
-    )
+    api_key = input(_('get_api_key_prompt'))
     if validate_api_key(api_key):
-        print('APIキーのテストに成功しました。')
-        print('以下, ~/.zshrcにAPIキーを保存しますか? [y/n]')
+        print(_('api_key_test_success'))
+        print(_('save_api_key_prompt'))
         if input() == 'y':
             zshrc_path = os.path.expanduser('~/.zshrc')
             with open(zshrc_path, 'a') as f:
                 f.write(f'export OPENAI_API_KEY={api_key}\n')
-            print(
-                f'APIキーを {zshrc_path} に保存しました。次回シェル起動時に読み込まれます。'
-            )
+            print(_('api_key_saved', zshrc_path))
         os.environ['OPENAI_API_KEY'] = api_key
         return api_key
     else:
-        print('コード生成にはAPIキーが必要です。')
+        print(_('api_key_required'))
         return None
 
 
@@ -59,7 +57,7 @@ def validate_api_key(api_key: str) -> bool:
     if response.status_code == 200:
         return True
     else:
-        print('APIキーの検証に失敗しました。')
+        print(_('api_key_validation_error'))
         return False
 
 
@@ -106,7 +104,7 @@ class ChatGPT:
         try:
             reply = responsej['choices'][0]['message']['content']
         except KeyError:
-            print('Error:レスポンスの形式が正しくありません. \n' + str(responsej))
+            print(_('response_format_error') + str(responsej))
             return 'Error: Unable to retrieve response.'
 
         self.messages.append({'role': 'assistant', 'content': reply})

atcdr/util/i18n.py

@@ -0,0 +1,317 @@
+import locale
+from typing import Dict, Optional
+
+
+class I18n:
+    def __init__(self):
+        self._lang: Optional[str] = None
+        self._messages: Dict[str, Dict[str, str]] = {'ja': {}, 'en': {}}
+        self._load_messages()
+        self._detect_language()
+
+    def _detect_language(self) -> None:
+        try:
+            system_locale = locale.getdefaultlocale()[0]
+            if system_locale and system_locale.startswith('ja'):
+                self._lang = 'ja'
+            else:
+                self._lang = 'en'
+        except Exception:
+            self._lang = 'en'
+
+    def _load_messages(self) -> None:
+        self._messages = {
+            'ja': {
+                # download.py
+                'retry_problem': '再試行します。{}',
+                'redirect_occurred': 'リダイレクトが発生しました。{}',
+                'problem_not_found': '問題が見つかりません。{}',
+                'server_error': 'サーバーエラーが発生しました。{}',
+                'html_fetch_failed': '{}に対応するHTMLファイルを取得できませんでした。',
+                'save_failed': '{}の保存に失敗しました',
+                'file_saved': 'ファイルを保存しました: {}',
+                'solve_contest_problems': 'コンテストの問題を解きたい',
+                'download_one_problem': '1問だけダウンロードする',
+                'exit': '終了する',
+                'download_atcoder_html': 'AtCoderの問題のHTMLファイルをダウンロードします',
+                'navigate_with_arrows': '十字キーで移動,[enter]で実行',
+                'input_contest_name': 'コンテスト名を入力してください (例: abc012, abs, typical90)',
+                'which_problem_download': 'どの問題をダウンロードしますか?',
+                'exiting': '終了します',
+                'invalid_selection': '無効な選択です',
+                'specify_contest_name': 'コンテスト名を指定してください',
+                'invalid_download_args': 'ダウンロードの引数が正しくありません',
+                # test.py
+                'runner_not_found': '{}の適切な言語のランナーが見つかりませんでした.',
+                'test_of': '{}のテスト \n',
+                'compile_time': 'コンパイルにかかった時間: [not italic cyan]{}[/] ms[/]',
+                'compiler_message': 'コンパイラーのメッセージ',
+                'status': 'ステータス ',
+                'execution_time': '実行時間   [cyan]{}[/cyan] ms',
+                'input': '入力',
+                'output': '出力',
+                'expected_output': '正解の出力',
+                'test_in_progress': 'テスト進行中',
+                'test_completed': 'テスト完了',
+                'problem_file_not_found': '問題のファイルが見つかりません。\n問題のファイルが存在するディレクトリーに移動してから実行してください。',
+                # markdown.py
+                'markdown_created': 'Markdownファイルを作成しました.',
+                # login.py
+                'already_logged_in': 'すでにログインしています. ',
+                'username': 'ユーザー名: ',
+                'password': 'パスワード: ',
+                'solve_captcha': 'キャプチャー認証を解決してください',
+                'logging_in': 'ログインします',
+                'waiting_login_result': 'ログインの結果の待機中...',
+                'login_success': 'ログイン成功!',
+                'error': 'エラー: {}',
+                # submit.py
+                'select_implementation': '以下の一覧から{}の実装/コンパイラーを選択してください',
+                'submission_failed': '提出に失敗しました',
+                'submission_id_not_found': '提出IDが取得できませんでした',
+                'submission_success': '提出に成功しました！',
+                'waiting_judge': 'ジャッジ待機中',
+                'judge_timeout': '15秒待ってもジャッジが開始されませんでした',
+                'judging': 'ジャッジ中',
+                'judge_completed': 'ジャッジ完了',
+                'not_logged_in': 'ログインしていません.',
+                'login_failed': 'ログインに失敗しました.',
+                'sample_not_ac': 'サンプルケースが AC していないので提出できません',
+                'logout_success': 'ログアウトしました.',
+                'submission_details': '提出ID: {}, URL: {}',
+                # open.py
+                'not_found': 'が見つかりません',
+                'url_opened': 'URLを開きました',
+                'url_not_found_in': 'にURLが見つかりませんでした',
+                # generate.py
+                'generating_code': 'コード生成中 (by {})',
+                'code_generation_success': 'コードの生成に成功しました. ',
+                'code_by_model': '{}による{}コード',
+                'code_saved': '{} の出力したコードを保存しました：{}',
+                'generating_template': '{}のテンプレートを生成しています...',
+                'template_created': 'テンプレートファイルを作成 :{}',
+                'nth_code_generation': '{}回目のコード生成中 (by {})',
+                'regenerating_with_prompt': '次のプロンプトを{}に与え,再生成します',
+                'code_generation_success_file': 'コードの生成に成功しました！：{}',
+                'testing_generated_code': '{}が生成したコードをテスト中',
+                'test_success': 'コードのテストに成功!',
+                'test_failed': 'コードのテストに失敗!',
+                'log_saved': '{}の出力のログを保存しました：{}',
+                # util/gpt.py
+                'api_key_validation_failed': '環境変数に設定されているAPIキーの検証に失敗しました ',
+                'get_api_key_prompt': 'https://platform.openai.com/api-keys からchatGPTのAPIキーを入手しましょう。\nAPIキー入力してください: ',
+                'api_key_test_success': 'APIキーのテストに成功しました。',
+                'save_api_key_prompt': '以下, ~/.zshrcにAPIキーを保存しますか? [y/n]',
+                'api_key_saved': 'APIキーを {} に保存しました。次回シェル起動時に読み込まれます。',
+                'api_key_required': 'コード生成にはAPIキーが必要です。',
+                'api_key_validation_error': 'APIキーの検証に失敗しました。',
+                'response_format_error': 'Error:レスポンスの形式が正しくありません. \n',
+                # util/fileops.py
+                'multiple_files_found': '複数のファイルが見つかりました.ファイルを選択してください:',
+                'target_file_not_found': '対象ファイルが見つかりません。',
+                'file_not_selected': 'ファイルが選択されませんでした。',
+                # util/problem.py and util/parse.py
+                'name_required': 'nameは必須です',
+                'language_not_supported': '言語は {} に対応していません',
+                'form_not_found': '問題ページにフォームが存在しません',
+                'problem_table_not_found': '問題のテーブルが見つかりませんでした.',
+                'tbody_not_found': 'tbodyが見つかりませんでした.',
+                # util/session.py
+                'response_info': 'レスポンス情報',
+                'item': '項目',
+                'content': '内容',
+                'status_code': 'ステータスコード',
+                'reason': '理由',
+                'response_headers': 'レスポンスヘッダー',
+                'key': 'キー',
+                'value': '値',
+                'redirect_history': 'リダイレクト履歴',
+                'step': 'ステップ',
+                'response_body': 'レスポンスボディ',
+                'hello_user': 'こんにちは！[cyan]{}[/] さん',
+                'session_check_error': 'セッションチェック中にエラーが発生しました: {}',
+                # Click command descriptions
+                'cmd_download': 'AtCoder の問題をダウンロード',
+                'cmd_test': 'テストを実行',
+                'cmd_submit': 'ソースを提出',
+                'cmd_generate': 'コードを生成',
+                'cmd_login': 'AtCoderへログイン',
+                'cmd_logout': 'AtCoderへログアウト',
+                'cmd_markdown': 'Markdown形式で問題を表示します',
+                'cmd_open': 'HTMLファイルを開く',
+                # Click option descriptions
+                'opt_no_test': 'テストをスキップ',
+                'opt_no_feedback': 'フィードバックをスキップ',
+                'opt_lang': '出力する言語を指定',
+                'opt_save': '変換結果をファイルに保存',
+                'opt_output_lang': '出力するプログラミング言語',
+                'opt_model': '使用するGPTモデル',
+                'opt_without_test': 'テストケースを省略して生成',
+                'opt_template': 'テンプレートを生成',
+            },
+            'en': {
+                # download.py
+                'retry_problem': 'Retrying... {}',
+                'redirect_occurred': 'Redirect occurred. {}',
+                'problem_not_found': 'Problem not found. {}',
+                'server_error': 'Server error occurred. {}',
+                'html_fetch_failed': 'Failed to fetch HTML file for {}.',
+                'save_failed': 'Failed to save {}',
+                'file_saved': 'File saved: {}',
+                'solve_contest_problems': 'Solve contest problems',
+                'download_one_problem': 'Download one problem',
+                'exit': 'Exit',
+                'download_atcoder_html': 'Download AtCoder problem HTML files',
+                'navigate_with_arrows': 'Use arrow keys to navigate, [enter] to execute',
+                'input_contest_name': 'Enter contest name (e.g., abc012, abs, typical90)',
+                'which_problem_download': 'Which problem to download?',
+                'exiting': 'Exiting',
+                'invalid_selection': 'Invalid selection',
+                'specify_contest_name': 'Please specify contest name',
+                'invalid_download_args': 'Invalid download arguments',
+                # test.py
+                'runner_not_found': 'Appropriate language runner for {} not found.',
+                'test_of': 'Testing {} \n',
+                'compile_time': 'Compile time: [not italic cyan]{}[/] ms[/]',
+                'compiler_message': 'Compiler message',
+                'status': 'Status ',
+                'execution_time': 'Execution time   [cyan]{}[/cyan] ms',
+                'input': 'Input',
+                'output': 'Output',
+                'expected_output': 'Expected output',
+                'test_in_progress': 'Testing in progress',
+                'test_completed': 'Test completed',
+                'problem_file_not_found': 'Problem file not found.\nPlease navigate to the directory containing the problem file.',
+                # markdown.py
+                'markdown_created': 'Markdown file created.',
+                # login.py
+                'already_logged_in': 'Already logged in. ',
+                'username': 'Username: ',
+                'password': 'Password: ',
+                'solve_captcha': 'Please solve the captcha',
+                'logging_in': 'Logging in',
+                'waiting_login_result': 'Waiting for login result...',
+                'login_success': 'Login successful!',
+                'error': 'Error: {}',
+                # submit.py
+                'select_implementation': 'Select {} implementation/compiler from the list below',
+                'submission_failed': 'Submission failed',
+                'submission_id_not_found': 'Submission ID not found',
+                'submission_success': 'Submission successful!',
+                'waiting_judge': 'Waiting for judge',
+                'judge_timeout': 'Judge did not start after 15 seconds',
+                'judging': 'Judging',
+                'judge_completed': 'Judge completed',
+                'not_logged_in': 'Not logged in.',
+                'login_failed': 'Login failed.',
+                'sample_not_ac': 'Cannot submit because sample cases are not AC',
+                'logout_success': 'Logged out successfully.',
+                'submission_details': 'Submission ID: {}, URL: {}',
+                # open.py
+                'not_found': ' not found',
+                'url_opened': 'URL opened',
+                'url_not_found_in': 'URL not found in ',
+                # generate.py
+                'generating_code': 'Generating code (by {})',
+                'code_generation_success': 'Code generation successful. ',
+                'code_by_model': '{} code by {}',
+                'code_saved': 'Code generated by {} saved: {}',
+                'generating_template': 'Generating {} template...',
+                'template_created': 'Template file created: {}',
+                'nth_code_generation': 'Code generation attempt {} (by {})',
+                'regenerating_with_prompt': 'Regenerating with the following prompt for {}',
+                'code_generation_success_file': 'Code generation successful: {}',
+                'testing_generated_code': 'Testing code generated by {}',
+                'test_success': 'Code test successful!',
+                'test_failed': 'Code test failed!',
+                'log_saved': 'Log of {} output saved: {}',
+                # util/gpt.py
+                'api_key_validation_failed': 'API key validation failed ',
+                'get_api_key_prompt': 'Get your ChatGPT API key from https://platform.openai.com/api-keys\nEnter API key: ',
+                'api_key_test_success': 'API key test successful.',
+                'save_api_key_prompt': 'Save API key to ~/.zshrc? [y/n]',
+                'api_key_saved': 'API key saved to {}. Will be loaded on next shell startup.',
+                'api_key_required': 'API key required for code generation.',
+                'api_key_validation_error': 'API key validation failed.',
+                'response_format_error': 'Error: Response format is invalid. \n',
+                # util/fileops.py
+                'multiple_files_found': 'Multiple files found. Please select a file:',
+                'target_file_not_found': 'Target file not found.',
+                'file_not_selected': 'No file selected.',
+                # util/problem.py and util/parse.py
+                'name_required': 'Name is required',
+                'language_not_supported': 'Language {} is not supported',
+                'form_not_found': 'Form not found on problem page',
+                'problem_table_not_found': 'Problem table not found.',
+                'tbody_not_found': 'tbody not found.',
+                # util/session.py
+                'response_info': 'Response Information',
+                'item': 'Item',
+                'content': 'Content',
+                'status_code': 'Status Code',
+                'reason': 'Reason',
+                'response_headers': 'Response Headers',
+                'key': 'Key',
+                'value': 'Value',
+                'redirect_history': 'Redirect History',
+                'step': 'Step',
+                'response_body': 'Response Body',
+                'hello_user': 'Hello [cyan]{}[/]!',
+                'session_check_error': 'Error occurred during session check: {}',
+                # Click command descriptions
+                'cmd_download': 'Download AtCoder problems',
+                'cmd_test': 'Run tests',
+                'cmd_submit': 'Submit source code',
+                'cmd_generate': 'Generate code',
+                'cmd_login': 'Login to AtCoder',
+                'cmd_logout': 'Logout from AtCoder',
+                'cmd_markdown': 'Display problem in Markdown format',
+                'cmd_open': 'Open HTML file',
+                # Click option descriptions
+                'opt_no_test': 'Skip testing',
+                'opt_no_feedback': 'Skip feedback',
+                'opt_lang': 'Specify output language',
+                'opt_save': 'Save conversion result to file',
+                'opt_output_lang': 'Specify output programming language',
+                'opt_model': 'GPT model to use',
+                'opt_without_test': 'Generate without test cases',
+                'opt_template': 'Generate template',
+            },
+        }
+
+    def get(self, key: str, *args) -> str:
+        """
+        メッセージを取得する
+        Args:
+            key: メッセージキー
+            *args: フォーマット引数
+        Returns:
+            フォーマット済みのメッセージ
+        """
+        message = self._messages.get(self._lang or 'en', {}).get(key, key)
+        if args:
+            try:
+                return message.format(*args)
+            except Exception:
+                return message
+        return message
+
+    def set_language(self, lang: str) -> None:
+        """言語を明示的に設定する"""
+        if lang in self._messages:
+            self._lang = lang
+
+    @property
+    def language(self) -> str:
+        """現在の言語を取得する"""
+        return self._lang or 'en'
+
+
+# シングルトンインスタンス
+i18n = I18n()
+
+
+# 便利な関数
+def _(key: str, *args) -> str:
+    """i18n.get()のショートカット"""
+    return i18n.get(key, *args)

atcdr/util/parse.py

@@ -5,6 +5,8 @@ from bs4 import BeautifulSoup as bs
 from bs4 import Tag
 from markdownify import MarkdownConverter
 
+from atcdr.util.i18n import _, i18n
+
 
 class HTML:
     def __init__(self, html: str) -> None:
@@ -68,13 +70,16 @@ class ProblemForm(Tag):
 
 
 class ProblemHTML(HTML):
-    def repair_me(self) -> None:
+    def repair_me(self, lang: Optional[str] = None) -> None:
         html = self.html.replace('//img.atcoder.jp', 'https://img.atcoder.jp')
-        html = html.replace(
-            '<meta http-equiv="Content-Language" content="en">',
-            '<meta http-equiv="Content-Language" content="ja">',
-        )
-        html = html.replace('LANG = "en"', 'LANG="ja"')
+        # 言語パラメータまたはi18nの設定に基づいて言語を決定
+        target_lang = lang or i18n.language
+        if target_lang == 'ja':
+            html = html.replace(
+                '<meta http-equiv="Content-Language" content="en">',
+                '<meta http-equiv="Content-Language" content="ja">',
+            )
+            html = html.replace('LANG = "en"', 'LANG="ja"')
         self.soup = bs(html, 'html.parser')
 
     def abstract_problem_part(self, lang: str) -> Optional[Tag]:
@@ -87,7 +92,7 @@ class ProblemHTML(HTML):
         elif lang == 'en':
             lang_class = 'lang-en'
         else:
-            raise ValueError(f'言語は {lang} に対応していません')
+            raise ValueError(_('language_not_supported', lang))
         span = task_statement.find('span', {'class': lang_class})
         return span
 
@@ -105,7 +110,7 @@ class ProblemHTML(HTML):
     def load_labeled_testcase(self) -> List:
         from atcdr.test import LabeledTestCase, TestCase
 
-        problem_part = self.abstract_problem_part('en')
+        problem_part = self.abstract_problem_part(i18n.language)
         if problem_part is None:
             return []
 
@@ -146,7 +151,7 @@ class ProblemHTML(HTML):
     def form(self) -> ProblemForm:
         form = self.soup.find('form', class_='form-code-submit')
         if not isinstance(form, Tag):
-            raise ValueError('問題ページにフォームが存在しません')
+            raise ValueError(_('form_not_found'))
         form.__class__ = ProblemForm
         return form
 
@@ -178,12 +183,12 @@ def get_problem_urls_from_tasks(html_content: str) -> list[tuple[str, str]]:
     soup = bs(html_content, 'html.parser')
     table = soup.find('table')
     if not table:
-        raise ValueError('問題のテーブルが見つかりませんでした.')
+        raise ValueError(_('problem_table_not_found'))
 
     # tbodyタグを見つける
     tbody = table.find('tbody')
     if not tbody:
-        raise ValueError('tbodyが見つかりませんでした.')
+        raise ValueError(_('tbody_not_found'))
 
     # tbody内の1列目のaタグのリンクと中身を取得
     links = []

atcdr/util/problem.py

@@ -3,19 +3,20 @@ from dataclasses import dataclass
 
 import requests
 
+from atcdr.util.i18n import _
 from atcdr.util.parse import get_problem_urls_from_tasks
 
 
 class Contest:
     def __init__(self, name: str, session: requests.Session):
         if not name:
-            raise ValueError('nameは必須です')
+            raise ValueError(_('name_required'))
         self.name = name
 
         self.url = f'https://atcoder.jp/contests/{name}/tasks'
         retry_attempts = 2
         retry_wait = 0.30
-        for _ in range(retry_attempts):
+        for attempt in range(retry_attempts):
             response = session.get(self.url)
             if response.ok:
                 break