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

atcdr/logout.py

@@ -2,21 +2,22 @@ import rich_click as click
 import webview
 from rich import print
 
+from atcdr.util.i18n import _
 from atcdr.util.session import delete_session, load_session, validate_session
 
 ATCODER_LOGIN_URL = 'https://atcoder.jp/login'
 
 
-@click.command(short_help='AtCoderへログアウト')
+@click.command(short_help=_('cmd_logout'), help=_('cmd_logout'))
 def logout() -> None:
     """AtCoderからログアウトします."""
     session = load_session()
     if not validate_session(session):
-        print('[red][-][/] ログインしていません.')
+        print('[red][-][/] ' + _('not_logged_in'))
         return
 
     delete_session()
-    print('[green][+][/] ログアウトしました.')
+    print('[green][+][/] ' + _('logout_success'))
 
     window = webview.create_window('AtCoder Logout', ATCODER_LOGIN_URL, hidden=True)
 

atcdr/markdown.py

@@ -6,6 +6,7 @@ from rich.markdown import Markdown
 
 from atcdr.util.fileops import add_file_selector
 from atcdr.util.filetype import FILE_EXTENSIONS, Lang
+from atcdr.util.i18n import _, i18n
 from atcdr.util.parse import ProblemHTML
 
 
@@ -19,7 +20,7 @@ def save_markdown(html_path: str, lang: str) -> None:
 
     with open(md_path, 'w', encoding='utf-8') as f:
         f.write(md)
-        console.print('[green][+][/green] Markdownファイルを作成しました.')
+        console.print('[green][+][/green] ' + _('markdown_created'))
 
 
 def print_markdown(html_path: str, lang: str) -> None:
@@ -30,12 +31,15 @@ def print_markdown(html_path: str, lang: str) -> None:
     console.print(Markdown(md))
 
 
-@click.command(short_help='Markdown形式で問題を表示します')
+@click.command(short_help=_('cmd_markdown'), help=_('cmd_markdown'))
 @add_file_selector('files', filetypes=[Lang.HTML])
-@click.option('--lang', default='ja', help='出力する言語を指定')
-@click.option('--save', is_flag=True, help='変換結果をファイルに保存')
+@click.option('--lang', default=None, help=_('opt_lang'))
+@click.option('--save', is_flag=True, help=_('opt_save'))
 def markdown(files, lang, save):
-    """Markdown形式で問題を表示します"""
+    """問題をMarkdown形式で表示します"""
+    # langが指定されていない場合は現在のロケールを使用
+    if lang is None:
+        lang = i18n.language
     for path in files:
         if save:
             save_markdown(path, lang)

atcdr/open.py

@@ -2,6 +2,7 @@ import webbrowser  # noqa: I001
 from rich.panel import Panel
 from rich.console import Console
 
+from atcdr.util.i18n import _
 from atcdr.util.filetype import Lang
 from atcdr.util.fileops import add_file_selector
 import rich_click as click
@@ -16,7 +17,7 @@ def open_html(file: str) -> None:
     except FileNotFoundError:
         console.print(
             Panel(
-                f"{file}' [red]が見つかりません[/]",
+                f"{file}' [red]" + _('not_found') + '[/]',
                 border_style='red',
             )
         )
@@ -27,20 +28,20 @@ def open_html(file: str) -> None:
         webbrowser.open_new_tab(url)
         console.print(
             Panel(
-                f'[green]URLを開きました[/] {url}',
+                '[green]' + _('url_opened') + f'[/] {url}',
                 border_style='green',
             )
         )
     else:
         console.print(
             Panel(
-                f'{file} [yellow]にURLが見つかりませんでした[/]',
+                f'{file} [yellow]' + _('url_not_found_in') + '[/]',
                 border_style='yellow',
             )
         )
 
 
-@click.command(short_help='HTMLファイルを開く')
+@click.command(short_help=_('cmd_open'), help=_('cmd_open'))
 @add_file_selector('files', filetypes=[Lang.HTML])
 def open_files(files):
     """指定したHTMLファイルをブラウザで開きます。"""

atcdr/submit.py

@@ -29,6 +29,7 @@ from atcdr.util.filetype import (
     lang2str,
     str2lang,
 )
+from atcdr.util.i18n import _
 from atcdr.util.parse import ProblemHTML, get_submission_id
 from atcdr.util.session import load_session, validate_session
 
@@ -61,14 +62,14 @@ def choose_langid_interactively(lang_dict: dict, lang: Lang) -> int:
     options = [*filter(lambda option: option.lang == lang, options)]
 
     langid = q.select(
-        message=f'以下の一覧から{lang2str(lang)}の実装/コンパイラーを選択してください',
+        message=_('select_implementation', lang2str(lang)),
         qmark='',
         pointer='❯❯❯',
         choices=[
             q.Choice(title=f'{option.display_name}', value=option.id)
             for option in options
         ],
-        instruction='\n 十字キーで移動,[enter]で実行',
+        instruction='\n ' + _('navigate_with_arrows'),
         style=q.Style(
             [
                 ('question', 'fg:#2196F3 bold'),
@@ -136,24 +137,24 @@ def post_source(source_path: str, url: str, session: requests.Session) -> Option
 
     window.events.loaded += on_loaded
 
-    with Status('キャプチャー認証を解決してください', spinner='circleHalves'):
+    with Status(_('solve_captcha'), spinner='circleHalves'):
         webview.start(private_mode=False)
 
     if 'submit' in api.url:
-        print('[red][-][/red] 提出に失敗しました')
+        print('[red][-][/red] ' + _('submission_failed'))
         return None
     elif 'submissions' in api.url:
         submission_id = get_submission_id(api.html)
         if not submission_id:
-            print('[red][-][/red] 提出IDが取得できませんでした')
+            print('[red][-][/red] ' + _('submission_id_not_found'))
             return None
 
         url = api.url.replace('/me', f'/{submission_id}')
-        print('[green][+][/green] 提出に成功しました！')
-        print(f'提出ID: {submission_id}, URL: {url}')
+        print('[green][+][/green] ' + _('submission_success'))
+        print(_('submission_details', submission_id, url))
         return url + '/status/json'
     else:
-        print('[red][-][/red] 提出に失敗しました')
+        print('[red][-][/red] ' + _('submission_failed'))
         return None
 
 
@@ -211,19 +212,19 @@ def print_status_submission(
         BarColumn(),
     )
 
-    with Status('ジャッジ待機中', spinner='dots'):
-        for _ in range(15):
+    with Status(_('waiting_judge'), spinner='dots'):
+        for i in range(15):
             time.sleep(1)
             data = session.get(api_url).json()
             status = parse_submission_status_json(data)
             if status.total or status.current:
                 break
         else:
-            print('[red][-][/] 15秒待ってもジャッジが開始されませんでした')
+            print('[red][-][/] ' + _('judge_timeout'))
             return
 
     total = status.total or 0
-    task_id = progress.add_task(description='ジャッジ中', total=total)
+    task_id = progress.add_task(description=_('judging'), total=total)
 
     test_info = TestInformation(
         lang=detect_language(path),
@@ -248,24 +249,22 @@ def print_status_submission(
         test_info.summary = status.status
         test_info.results = [ResultStatus.AC] * total
 
-        progress.update(task_id, description='ジャッジ完了', completed=total)
+        progress.update(task_id, description=_('judge_completed'), completed=total)
         live.update(create_renderable_test_info(test_info, progress))
 
 
 def submit_source(path: str, no_test: bool, no_feedback: bool) -> None:
     session = load_session()
     if not validate_session(session):
-        print('[red][-][/] ログインしていません.')
+        print('[red][-][/] ' + _('not_logged_in'))
         login()
         if not validate_session(session):
-            print('[red][-][/] ログインに失敗しました.')
+            print('[red][-][/] ' + _('login_failed'))
             return
 
     html_files = [file for file in os.listdir('.') if file.endswith('.html')]
     if not html_files:
-        print(
-            '問題のファイルが見つかりません \n問題のファイルが存在するディレクトリーに移動してから実行してください'
-        )
+        print(_('problem_file_not_found'))
         return
 
     with open(html_files[0], 'r') as file:
@@ -279,7 +278,7 @@ def submit_source(path: str, no_test: bool, no_feedback: bool) -> None:
     print(create_renderable_test_info(test.info))
 
     if test.info.summary != ResultStatus.AC and not no_test:
-        print('[red][-][/] サンプルケースが AC していないので提出できません')
+        print('[red][-][/] ' + _('sample_not_ac'))
         return
 
     api_status_link = post_source(path, url, session)
@@ -290,12 +289,10 @@ def submit_source(path: str, no_test: bool, no_feedback: bool) -> None:
         print_status_submission(api_status_link, path, session)
 
 
-@click.command(short_help='ソースを提出')
+@click.command(short_help=_('cmd_submit'), help=_('cmd_submit'))
 @add_file_selector('files', filetypes=COMPILED_LANGUAGES + INTERPRETED_LANGUAGES)
-@click.option('--no-test', is_flag=True, default=False, help='テストをスキップ')
-@click.option(
-    '--no-feedback', is_flag=True, default=False, help='フィードバックをスキップ'
-)
+@click.option('--no-test', is_flag=True, default=False, help=_('opt_no_test'))
+@click.option('--no-feedback', is_flag=True, default=False, help=_('opt_no_feedback'))
 def submit(files, no_test, no_feedback):
     """指定したファイルをAtCoderへ提出します。"""
     for path in files:

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})