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

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

@@ -1,94 +1,48 @@
-from typing import List, Optional
+import time
+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, number: Optional[int] = None):
+    def __init__(self, name: str, session: requests.Session):
         if not name:
-            raise ValueError('nameは必須です')
-        self._name = name
-        self._number = number
-        if number and number > 0:
-            self._contest = f'{name}{number:03}'
-        else:
-            self._contest = name
-
-    @property
-    def contest(self) -> str:
-        return self._contest
-
-    @property
-    def number(self) -> Optional[int]:
-        return self._number
-
-    @property
-    def url(self) -> str:
-        return f'https://atcoder.jp/contests/{self.contest}/tasks'
-
-    def __str__(self) -> str:
-        return f'{self.contest}'
-
-    def __repr__(self) -> str:
-        return f'Contest(name={self._name}, number={self._number})'
-
-    def problems(self, session: Optional[requests.Session] = None) -> List['Problem']:
-        session = session or requests.Session()
-        response = session.get(self.url)
-
-        if response.status_code != 200:
-            return []
-
-        return [
-            Problem(self, label=label, url=url)
+            raise ValueError(_('name_required'))
+        self.name = name
+
+        self.url = f'https://atcoder.jp/contests/{name}/tasks'
+        retry_attempts = 2
+        retry_wait = 0.30
+        for attempt in range(retry_attempts):
+            response = session.get(self.url)
+            if response.ok:
+                break
+            else:
+                time.sleep(retry_wait)
+
+        self.problems = [
+            Problem(url=url, contest=self, label=label)
             for label, url in get_problem_urls_from_tasks(response.text)
         ]
 
-
-class Diff(str):
-    def __new__(cls, diff: str) -> 'Diff':
-        if isinstance(diff, str) and len(diff) == 1 and diff.isalpha():
-            return super().__new__(cls, diff.upper())
-        raise ValueError('diffは英大文字または小文字の1文字である必要があります')
+    def __str__(self) -> str:
+        return f'{self.name}'
 
     def __repr__(self) -> str:
-        return f"Diff('{self}')"
+        return f'Contest(name={self.name})'
 
 
+@dataclass
 class Problem:
-    def __init__(
-        self,
-        contest: Contest,
-        difficulty: Optional[Diff] = None,
-        label: Optional[str] = None,
-        url: Optional[str] = None,
-    ):
-        self._contest = contest
-        if difficulty:
-            self._label = difficulty.upper()
-            self._url = contest.url + f'/{contest}_{difficulty.lower()}'
-        elif label and url:
-            self._label = label
-            self._url = url
-        else:
-            raise ValueError('labelとurlは両方必須かdifficultyが必要です')
-
-    @property
-    def contest(self) -> Contest:
-        return self._contest
+    url: str
+    contest: Contest
+    label: str
 
-    @property
-    def label(self) -> str:
-        return self._label
-
-    @property
-    def url(self) -> str:
-        return self._url
+    def __str__(self) -> str:
+        return f'{self.label} - {self.contest.name}'
 
     def __repr__(self) -> str:
-        return f'Problem(contest={self.contest}, label={self.label}, url={self.url})'
-
-    def __str__(self) -> str:
-        return f'{self.contest} {self.label}'
+        return f'Problem(url={self.url}, contest={self.contest}, label={self.label})'

atcdr/util/session.py

@@ -9,6 +9,7 @@ from rich.panel import Panel
 from rich.syntax import Syntax
 from rich.table import Table
 
+from atcdr.util.i18n import _
 from atcdr.util.parse import get_username_from_html
 
 COOKIE_PATH = os.path.join(os.path.expanduser('~'), '.cache', 'atcdr', 'session.json')
@@ -19,18 +20,18 @@ def print_rich_response(
     response: requests.Response, body_range: tuple = (0, 24)
 ) -> None:
     # レスポンス情報をテーブル形式で表示
-    info_table = Table(title='レスポンス情報')
-    info_table.add_column('項目', justify='left', style='cyan', no_wrap=True)
-    info_table.add_column('内容', justify='left', style='magenta')
+    info_table = Table(title=_('response_info'))
+    info_table.add_column(_('item'), justify='left', style='cyan', no_wrap=True)
+    info_table.add_column(_('content'), justify='left', style='magenta')
     info_table.add_row('URL', response.url)
-    info_table.add_row('ステータスコード', str(response.status_code))
-    info_table.add_row('理由', response.reason)
+    info_table.add_row(_('status_code'), str(response.status_code))
+    info_table.add_row(_('reason'), response.reason)
     info_table = Align.center(info_table)
 
     # ヘッダー情報をテーブル形式で表示
-    header_table = Table(title='レスポンスヘッダー')
-    header_table.add_column('キー', style='cyan', no_wrap=True)
-    header_table.add_column('値', style='magenta', overflow='fold')
+    header_table = Table(title=_('response_headers'))
+    header_table.add_column(_('key'), style='cyan', no_wrap=True)
+    header_table.add_column(_('value'), style='magenta', overflow='fold')
     for key, value in response.headers.items():
         value = unquote(value)
         header_table.add_row(key, value)
@@ -39,9 +40,9 @@ def print_rich_response(
     # リダイレクトの歴史
     redirect_table = None
     if response.history:
-        redirect_table = Table(title='リダイレクト履歴')
-        redirect_table.add_column('ステップ', style='cyan')
-        redirect_table.add_column('ステータスコード', style='magenta')
+        redirect_table = Table(title=_('redirect_history'))
+        redirect_table.add_column(_('step'), style='cyan')
+        redirect_table.add_column(_('status_code'), style='magenta')
         redirect_table.add_column('URL', style='green')
         for i, redirect_response in enumerate(response.history):
             redirect_table.add_row(
@@ -80,7 +81,7 @@ def print_rich_response(
             if response.text
             else None
         )
-    body_panel = Panel(body, title='レスポンスボディ') if body else None
+    body_panel = Panel(body, title=_('response_body')) if body else None
 
     print(info_table)
     print(header_table)
@@ -102,7 +103,7 @@ def load_session() -> requests.Session:
             response = session.get(ATCODER_URL)
             username = get_username_from_html(response.text)
             if username:
-                print(f'こんにちは！[cyan]{username}[/] さん')
+                print(_('hello_user', username))
             return session
         else:
             return requests.Session()
@@ -131,7 +132,7 @@ def validate_session(session: requests.Session) -> bool:
                 return False
         return False
     except requests.RequestException as e:
-        print(f'[red][-][/] セッションチェック中にエラーが発生しました: {e}')
+        print('[red][-][/] ' + _('session_check_error', e))
         return False