AtCoderStudyBooster 0.3.1__tar.gz → 0.3.2__tar.gz

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/PKG-INFO

@@ -1,17 +1,18 @@
 Metadata-Version: 2.3
 Name: AtCoderStudyBooster
-Version: 0.3.1
+Version: 0.3.2
 Summary: A tool to download and manage AtCoder problems.
 Project-URL: Homepage, https://github.com/yuta6/AtCoderStudyBooster
 Author-email: yuta6 <46110512+yuta6@users.noreply.github.com>
 License: MIT
 Requires-Python: >=3.8
 Requires-Dist: beautifulsoup4
-Requires-Dist: fire
+Requires-Dist: click-aliases>=1.0.5
 Requires-Dist: markdownify>=0.13.1
 Requires-Dist: pywebview>=5.4
 Requires-Dist: questionary>=2.0.1
 Requires-Dist: requests
+Requires-Dist: rich-click>=1.8.8
 Requires-Dist: rich>=13.7.1
 Requires-Dist: tiktoken
 Requires-Dist: types-beautifulsoup4>=4.12.0.20240511
@@ -22,14 +23,21 @@ Description-Content-Type: text/markdown
 
 ## 概要
 
-AtCoderStudyBoosterはAtCoderの学習を加速させるためのCLIツールです。問題をローカルにダウンロードし、テスト、提出、解答の作成をサポートするツールです。Pythonが入っていることが必須です。Pythonが入っている環境なら、`pip install AtCoderStudyBooster`でインストールできます。
+🚧 このプロジェクトはまだ実験段階です。日々のAtCoder学習に役立つ機能を順次追加しています。
+
+AtCoderStudyBoosterはAtCoderの学習を加速させるためのCLIツールです。問題をローカルにダウンロードし、テスト、提出、解答の作成をサポートするツールです。Pythonが入っていることが必須です。Pythonが入っている環境なら、`pip install AtCoderStudyBooster`でインストールできます。(Python3.8以上が必要です)
+
+キャプチャ認証が導入されたあとでも、CLIからほぼ自動でログイン＆提出できます。ただしキャプチャを解くために、GUI環境が必要になる場合があります。
 
 このツールは以下のプロジェクトに強く影響を受けています。
 [online-judge-tools](https://github.com/online-judge-tools)
 [atcoder-cli](https://github.com/Tatamo/atcoder-cli)
 
+
 ## 利用ケース
 
+まずは`download`コマンドを利用して問題をローカルにダウンロードしてみましょう。
+
 ### B問題の練習したい場合
 
 ABCコンテストの223から226のB問題だけを集中的に練習したい場合、次のコマンドを実行します。

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/README.md

@@ -2,14 +2,21 @@
 
 ## 概要
 
-AtCoderStudyBoosterはAtCoderの学習を加速させるためのCLIツールです。問題をローカルにダウンロードし、テスト、提出、解答の作成をサポートするツールです。Pythonが入っていることが必須です。Pythonが入っている環境なら、`pip install AtCoderStudyBooster`でインストールできます。
+🚧 このプロジェクトはまだ実験段階です。日々のAtCoder学習に役立つ機能を順次追加しています。
+
+AtCoderStudyBoosterはAtCoderの学習を加速させるためのCLIツールです。問題をローカルにダウンロードし、テスト、提出、解答の作成をサポートするツールです。Pythonが入っていることが必須です。Pythonが入っている環境なら、`pip install AtCoderStudyBooster`でインストールできます。(Python3.8以上が必要です)
+
+キャプチャ認証が導入されたあとでも、CLIからほぼ自動でログイン＆提出できます。ただしキャプチャを解くために、GUI環境が必要になる場合があります。
 
 このツールは以下のプロジェクトに強く影響を受けています。
 [online-judge-tools](https://github.com/online-judge-tools)
 [atcoder-cli](https://github.com/Tatamo/atcoder-cli)
 
+
 ## 利用ケース
 
+まずは`download`コマンドを利用して問題をローカルにダウンロードしてみましょう。
+
 ### B問題の練習したい場合
 
 ABCコンテストの223から226のB問題だけを集中的に練習したい場合、次のコマンドを実行します。

atcoderstudybooster-0.3.2/atcdr/cli.py

@@ -0,0 +1,86 @@
+from importlib.metadata import metadata
+
+import rich_click as click
+from click_aliases import ClickAliasedGroup
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.traceback import install
+from rich_click import RichGroup
+
+from atcdr.download import download
+from atcdr.generate import generate
+from atcdr.login import login
+from atcdr.logout import logout
+from atcdr.markdown import markdown
+from atcdr.open import open_files
+from atcdr.submit import submit
+from atcdr.test import test
+
+
+# ─── RichClick + ClickAliases 両対応の Group クラス ───
+class AliasedRichGroup(ClickAliasedGroup, RichGroup):
+    def format_commands(self, ctx, console, *args, **kwargs):
+        console = Console()
+        commands = self.list_commands(ctx)
+
+        table = Table(show_header=False, box=None, pad_edge=False)
+        table.add_column('command', style='bold cyan', no_wrap=True)
+        table.add_column('help', style='')
+
+        for name in commands:
+            cmd = self.get_command(ctx, name)
+            if not cmd or getattr(cmd, 'hidden', False):
+                continue
+
+            aliases = self._commands.get(name, [])
+            alias_part = f"[dim]({', '.join(aliases)})[/]" if aliases else ''
+
+            short = (
+                cmd.get_short_help_str()
+                if hasattr(cmd, 'get_short_help_str')
+                else cmd.short_help or ''
+            )
+            table.add_row(f'{name}{alias_part}', short)
+
+        panel = Panel(table, title='Commands', expand=False)
+        console.print(panel)
+
+
+# ─── CLI 定義 ──────────────────────────────────────────
+_meta = metadata('AtCoderStudyBooster')
+_NAME = _meta['Name']
+_VERSION = _meta['Version']
+
+click.rich_click.MAX_WIDTH = 100
+click.rich_click.SHOW_ARGUMENTS = True
+click.rich_click.STYLE_HELPTEXT_FIRST_LINE = 'bold cyan'
+click.rich_click.STYLE_HELPTEXT = 'dim'
+
+
+@click.group(
+    cls=AliasedRichGroup,
+    context_settings={'help_option_names': ['-h', '--help']},
+)
+@click.version_option(
+    _VERSION,
+    '-v',
+    '--version',
+    prog_name=_NAME,
+    message='%(prog)s %(version)s',
+)
+def cli():
+    install()
+
+
+cli.add_command(test, aliases=['t'])
+cli.add_command(download, aliases=['d'])
+cli.add_command(open_files, 'open', aliases=['o'])
+cli.add_command(generate, aliases=['g'])
+cli.add_command(markdown, aliases=['md'])
+cli.add_command(submit, aliases=['s'])
+cli.add_command(login)
+cli.add_command(logout)
+
+if __name__ == '__main__':
+    cli()

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/download.py

@@ -4,6 +4,7 @@ import time
 from typing import Callable, List, Union, cast
 
 import questionary as q
+import rich_click as click
 from rich import print
 from rich.prompt import Prompt
 
@@ -242,11 +243,32 @@ def interactive_download() -> None:
         print('[bold red]無効な選択です[/]')
 
 
+@click.command(short_help='AtCoderの問題をダウンロード')
+@click.argument('first', nargs=1, type=str, required=False)
+@click.argument('second', nargs=1, type=str, required=False)
 def download(
-    first: Union[str, int, None] = None,
-    second: Union[str, int, None] = None,
+    first: Union[str, None] = None,
+    second: Union[str, None] = None,
     base_path: str = '.',
 ) -> None:
+    """
+    AtCoderの問題をダウンロードします
+
+    download
+    対話形式でダウンロードを開始します。
+
+    download abc012
+        コンテスト abc012 の全問題をダウンロード
+
+    download A 120
+        難易度Aの120番問題をダウンロード
+
+    download 120..130 B
+        ABCの120～130番のB問題をダウンロード
+
+    download 120
+        ABCの120番問題をダウンロード
+    """
     if first is None:
         interactive_download()
         return

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/generate.py

@@ -2,12 +2,13 @@ import json
 import os
 import re
 
+import rich_click as click
 from rich.console import Console
 from rich.panel import Panel
 from rich.syntax import Syntax
 
 from atcdr.test import ResultStatus, TestRunner, create_renderable_test_info
-from atcdr.util.execute import execute_files
+from atcdr.util.fileops import add_file_selector
 from atcdr.util.filetype import (
     FILE_EXTENSIONS,
     Filename,
@@ -181,31 +182,21 @@ Please provide an updated version of the code in {lang2str(lang)}."""
     return
 
 
-def generate(
-    *source: str,
-    lang: str = 'Python',
-    model: str = Model.GPT41_MINI.value,
-    without_test: bool = False,
-    template: bool = False,
-) -> None:
+@click.command(short_help='コードを生成')
+@add_file_selector('files', filetypes=[Lang.HTML])
+@click.option('--lang', default='Python', help='出力するプログラミング言語')
+@click.option('--model', default=Model.GPT41_MINI.value, help='使用するGPTモデル')
+@click.option('--without-test', is_flag=True, help='テストケースを省略して生成')
+@click.option('--template', is_flag=True, help='テンプレートを生成')
+def generate(files, lang, model, without_test, template):
+    """HTMLファイルからコード生成またはテンプレート出力を行います。"""
     la = str2lang(lang)
     model_enum = Model(model)
 
-    if template:
-        execute_files(
-            *source,
-            func=lambda file: generate_template(file, la),
-            target_filetypes=[Lang.HTML],
-        )
-    elif without_test:
-        execute_files(
-            *source,
-            func=lambda file: generate_code(file, la, model_enum),
-            target_filetypes=[Lang.HTML],
-        )
-    else:
-        execute_files(
-            *source,
-            func=lambda file: solve_problem(file, la, model_enum),
-            target_filetypes=[Lang.HTML],
-        )
+    for path in files:
+        if template:
+            generate_template(path, la)
+        elif without_test:
+            generate_code(path, la, model_enum)
+        else:
+            solve_problem(path, la, model_enum)

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/login.py

@@ -1,6 +1,7 @@
 import threading
 import time
 
+import rich_click as click
 import webview
 from requests import Session
 from rich.console import Console
@@ -13,7 +14,9 @@ ATCODER_HOME_URL = 'https://atcoder.jp/home'
 console = Console()
 
 
+@click.command(short_help='AtCoderへログイン')
 def login() -> None:
+    """AtCoderへログインします."""
     session = load_session()
     if validate_session(session):
         console.print('[green][+][/] すでにログインしています. ')

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/logout.py

@@ -1,3 +1,4 @@
+import rich_click as click
 import webview
 from rich import print
 
@@ -6,7 +7,9 @@ from atcdr.util.session import delete_session, load_session, validate_session
 ATCODER_LOGIN_URL = 'https://atcoder.jp/login'
 
 
+@click.command(short_help='AtCoderへログアウト')
 def logout() -> None:
+    """AtCoderからログアウトします."""
     session = load_session()
     if not validate_session(session):
         print('[red][-][/] ログインしていません.')

atcoderstudybooster-0.3.2/atcdr/markdown.py

@@ -0,0 +1,43 @@
+import os
+
+import rich_click as click
+from rich.console import Console
+from rich.markdown import Markdown
+
+from atcdr.util.fileops import add_file_selector
+from atcdr.util.filetype import FILE_EXTENSIONS, Lang
+from atcdr.util.parse import ProblemHTML
+
+
+def save_markdown(html_path: str, lang: str) -> None:
+    console = Console()
+    with open(html_path, 'r', encoding='utf-8') as f:
+        html = ProblemHTML(f.read())
+    md = html.make_problem_markdown(lang)
+    file_without_ext = os.path.splitext(html_path)[0]
+    md_path = file_without_ext + FILE_EXTENSIONS[Lang.MARKDOWN]
+
+    with open(md_path, 'w', encoding='utf-8') as f:
+        f.write(md)
+        console.print('[green][+][/green] Markdownファイルを作成しました.')
+
+
+def print_markdown(html_path: str, lang: str) -> None:
+    console = Console()
+    with open(html_path, 'r', encoding='utf-8') as f:
+        html = ProblemHTML(f.read())
+    md = html.make_problem_markdown(lang)
+    console.print(Markdown(md))
+
+
+@click.command(short_help='Markdown形式で問題を表示します')
+@add_file_selector('files', filetypes=[Lang.HTML])
+@click.option('--lang', default='ja', help='出力する言語を指定')
+@click.option('--save', is_flag=True, help='変換結果をファイルに保存')
+def markdown(files, lang, save):
+    """Markdown形式で問題を表示します"""
+    for path in files:
+        if save:
+            save_markdown(path, lang)
+        else:
+            print_markdown(path, lang)

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/open.py

@@ -3,7 +3,8 @@ from rich.panel import Panel
 from rich.console import Console
 
 from atcdr.util.filetype import Lang
-from atcdr.util.execute import execute_files
+from atcdr.util.fileops import add_file_selector
+import rich_click as click
 from atcdr.util.parse import ProblemHTML
 
 
@@ -39,5 +40,9 @@ def open_html(file: str) -> None:
         )
 
 
-def open_files(*args: str) -> None:
-    execute_files(*args, func=open_html, target_filetypes=[Lang.HTML])
+@click.command(short_help='HTMLファイルを開く')
+@add_file_selector('files', filetypes=[Lang.HTML])
+def open_files(files):
+    """指定したHTMLファイルをブラウザで開きます。"""
+    for path in files:
+        open_html(path)

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/submit.py

@@ -5,6 +5,7 @@ from typing import Dict, List, NamedTuple, Optional
 
 import questionary as q
 import requests
+import rich_click as click
 import webview
 from bs4 import BeautifulSoup as bs
 from rich import print
@@ -19,7 +20,7 @@ from atcdr.test import (
     TestRunner,
     create_renderable_test_info,
 )
-from atcdr.util.execute import execute_files
+from atcdr.util.fileops import add_file_selector
 from atcdr.util.filetype import (
     COMPILED_LANGUAGES,
     INTERPRETED_LANGUAGES,
@@ -289,9 +290,13 @@ def submit_source(path: str, no_test: bool, no_feedback: bool) -> None:
         print_status_submission(api_status_link, path, session)
 
 
-def submit(*args: str, no_test: bool = False, no_feedback: bool = False) -> None:
-    execute_files(
-        *args,
-        func=lambda path: submit_source(path, no_test, no_feedback),
-        target_filetypes=COMPILED_LANGUAGES + INTERPRETED_LANGUAGES,
-    )
+@click.command(short_help='ソースを提出')
+@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='フィードバックをスキップ'
+)
+def submit(files, no_test, no_feedback):
+    """指定したファイルをAtCoderへ提出します。"""
+    for path in files:
+        submit_source(path, no_test, no_feedback)

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/atcdr/test.py

@@ -6,6 +6,7 @@ from dataclasses import dataclass, field
 from enum import Enum
 from typing import Dict, List, Optional, Tuple, Union
 
+import rich_click as click
 from rich.console import Group, RenderableType
 from rich.live import Live
 from rich.markup import escape
@@ -17,7 +18,7 @@ from rich.syntax import Syntax
 from rich.table import Table
 from rich.text import Text
 
-from atcdr.util.execute import execute_files
+from atcdr.util.fileops import add_file_selector
 from atcdr.util.filetype import (
     COMPILED_LANGUAGES,
     INTERPRETED_LANGUAGES,
@@ -424,9 +425,9 @@ def run_test(path_of_code: str) -> None:
     render_results(test)
 
 
-def test(*args: str) -> None:
-    execute_files(
-        *args,
-        func=run_test,
-        target_filetypes=INTERPRETED_LANGUAGES + COMPILED_LANGUAGES,
-    )
+@click.command(short_help='テストを実行')
+@add_file_selector('files', filetypes=COMPILED_LANGUAGES + INTERPRETED_LANGUAGES)
+def test(files):
+    """指定したソースコードをサンプルケースでテストします。"""
+    for path in files:
+        run_test(path)

atcoderstudybooster-0.3.2/atcdr/util/fileops.py

@@ -0,0 +1,102 @@
+import functools
+import glob
+import os
+from typing import List, Tuple
+
+import questionary as q
+import rich_click as click
+
+from atcdr.util.filetype import FILE_EXTENSIONS, Lang
+
+
+def collect_files(
+    patterns: Tuple[str, ...],
+    exts: Tuple[str, ...],
+    recursive: bool,
+) -> List[str]:
+    # 1) ベース候補
+    if recursive:
+        candidates = []
+        for root, _, files in os.walk('.'):
+            for f in files:
+                candidates.append(os.path.join(root, f))
+    else:
+        candidates = [f for f in os.listdir('.') if os.path.isfile(f)]
+
+    # 2) パターン＋glob 展開
+    matched = set()
+    pats = patterns or ['*']
+    for pat in pats:
+        for m in glob.glob(pat, recursive=recursive):
+            if os.path.isfile(m):
+                matched.add(m)
+        if '*' not in pat and os.path.isfile(pat):
+            matched.add(pat)
+
+    if exts:
+        matched = {f for f in matched if os.path.splitext(f)[1] in exts}
+
+    return sorted(matched)
+
+
+def select_files_interactively(files: List[str]) -> List[str]:
+    target_file = q.select(
+        message='複数のファイルが見つかりました.ファイルを選択してください:',
+        choices=[q.Choice(title=file, value=file) for file in files],
+        instruction='\n 十字キーで移動, [enter]で実行',
+        pointer='❯',
+        qmark='',
+        style=q.Style(
+            [
+                ('qmark', 'fg:#2196F3 bold'),
+                ('question', 'fg:#2196F3 bold'),
+                ('answer', 'fg:#FFB300 bold'),
+                ('pointer', 'fg:#FFB300 bold'),
+                ('highlighted', 'fg:#FFB300 bold'),
+                ('selected', 'fg:#FFB300 bold'),
+            ]
+        ),
+    ).ask()
+    return target_file
+
+
+def add_file_selector(
+    arg_name: str,
+    filetypes: list[Lang],
+):
+    def decorator(f):
+        @click.argument(arg_name, nargs=-1, type=click.STRING)
+        @click.pass_context
+        @functools.wraps(f)
+        def wrapper(ctx: click.Context, **kwargs):
+            # Click から渡される元のパターン一覧を取得
+            patterns: tuple[str, ...] = kwargs.pop(arg_name)
+
+            # 1) 拡張子リストを作成
+            exts = [FILE_EXTENSIONS[lang] for lang in filetypes]
+
+            # 2) ファイル収集 (非再帰固定)
+            files = collect_files(patterns, tuple(exts), recursive=False)
+            if not files:
+                click.echo('対象ファイルが見つかりません。')
+                ctx.exit(1)
+
+            # 3) 候補が1つなら即実行
+            if len(files) == 1:
+                return ctx.invoke(f, **{arg_name: files}, **kwargs)
+
+            # 4) 引数なしなら対話選択、それ以外はまとめて渡す
+            if not patterns:
+                selected = select_files_interactively(files)
+                if not selected:
+                    click.echo('ファイルが選択されませんでした。')
+                    ctx.exit(1)
+                selected_list = [selected]
+                return ctx.invoke(f, **{arg_name: selected_list}, **kwargs)
+
+            # 5) patterns 指定ありならマッチ全件を渡す
+            return ctx.invoke(f, **{arg_name: files}, **kwargs)
+
+        return wrapper
+
+    return decorator

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/pyproject.toml

@@ -1,13 +1,12 @@
 [project]
 name = "AtCoderStudyBooster"
-version = "0.3.1"
+version = "0.3.2"
 description = "A tool to download and manage AtCoder problems."
 authors = [
     { name = "yuta6", email = "46110512+yuta6@users.noreply.github.com" }
 ]
 dependencies = [
     "beautifulsoup4",
-    "fire",
     "requests",
     "tiktoken",
     "types-requests>=2.32.0.20240712",
@@ -16,6 +15,8 @@ dependencies = [
     "rich>=13.7.1",
     "questionary>=2.0.1",
     "pywebview>=5.4",
+    "rich-click>=1.8.8",
+    "click-aliases>=1.0.5",
 ]
 readme = "README.md"
 requires-python = ">= 3.8"
@@ -23,7 +24,7 @@ license = { text = "MIT" }
 urls = { Homepage = "https://github.com/yuta6/AtCoderStudyBooster" }
 
 [project.scripts]
-"atcdr" = "atcdr.main:main"
+"atcdr" = "atcdr.cli:cli"
 
 [build-system]
 requires = ["hatchling==1.26.3"]

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/requirements-dev.lock

@@ -21,12 +21,15 @@ cfgv==3.4.0
     # via pre-commit
 charset-normalizer==3.3.2
     # via requests
+click==8.1.8
+    # via click-aliases
+    # via rich-click
+click-aliases==1.0.5
+    # via atcoderstudybooster
 distlib==0.3.8
     # via virtualenv
 filelock==3.15.4
     # via virtualenv
-fire==0.6.0
-    # via atcoderstudybooster
 identify==2.6.0
     # via pre-commit
 idna==3.7
@@ -71,13 +74,13 @@ requests==2.32.3
     # via tiktoken
 rich==13.7.1
     # via atcoderstudybooster
+    # via rich-click
+rich-click==1.8.8
+    # via atcoderstudybooster
 six==1.16.0
-    # via fire
     # via markdownify
 soupsieve==2.5
     # via beautifulsoup4
-termcolor==2.4.0
-    # via fire
 tiktoken==0.7.0
     # via atcoderstudybooster
 types-beautifulsoup4==4.12.0.20240511
@@ -89,6 +92,7 @@ types-requests==2.32.0.20240712
 typing-extensions==4.12.2
     # via mypy
     # via pywebview
+    # via rich-click
 urllib3==2.2.2
     # via requests
     # via types-requests

{atcoderstudybooster-0.3.1 → atcoderstudybooster-0.3.2}/requirements.lock

@@ -19,7 +19,10 @@ certifi==2024.7.4
     # via requests
 charset-normalizer==3.3.2
     # via requests
-fire==0.6.0
+click==8.1.8
+    # via click-aliases
+    # via rich-click
+click-aliases==1.0.5
     # via atcoderstudybooster
 idna==3.7
     # via requests
@@ -46,13 +49,13 @@ requests==2.32.3
     # via tiktoken
 rich==13.7.1
     # via atcoderstudybooster
+    # via rich-click
+rich-click==1.8.8
+    # via atcoderstudybooster
 six==1.16.0
-    # via fire
     # via markdownify
 soupsieve==2.5
     # via beautifulsoup4
-termcolor==2.4.0
-    # via fire
 tiktoken==0.7.0
     # via atcoderstudybooster
 types-beautifulsoup4==4.12.0.20240511
@@ -63,6 +66,7 @@ types-requests==2.32.0.20240712
     # via atcoderstudybooster
 typing-extensions==4.13.2
     # via pywebview
+    # via rich-click
 urllib3==2.2.2
     # via requests
     # via types-requests

atcoderstudybooster-0.3.1/atcdr/main.py

@@ -1,47 +0,0 @@
-from importlib.metadata import metadata
-
-import fire  # type: ignore
-from rich.traceback import install
-
-from atcdr.download import download
-from atcdr.generate import generate
-from atcdr.login import login
-from atcdr.logout import logout
-from atcdr.markdown import markdown
-from atcdr.open import open_files
-from atcdr.submit import submit
-from atcdr.test import test
-
-
-def get_version() -> None:
-    meta = metadata('AtCoderStudyBooster')
-    print(meta['Name'], meta['Version'])
-
-
-MAP_COMMANDS: dict = {
-    'test': test,
-    't': test,
-    'download': download,
-    'd': download,
-    'open': open_files,
-    'o': open_files,
-    'generate': generate,
-    'g': generate,
-    'markdown': markdown,
-    'md': markdown,
-    'login': login,
-    'logout': logout,
-    'submit': submit,
-    's': submit,
-    '--version': get_version,
-    '-v': get_version,
-}
-
-
-def main():
-    install()
-    fire.Fire(MAP_COMMANDS)
-
-
-if __name__ == '__main__':
-    main()

atcoderstudybooster-0.3.1/atcdr/markdown.py

@@ -1,39 +0,0 @@
-import os
-
-from rich.console import Console
-from rich.markdown import Markdown
-
-from atcdr.util.execute import execute_files
-from atcdr.util.filetype import FILE_EXTENSIONS, Lang
-from atcdr.util.parse import ProblemHTML
-
-
-def save_markdown(html_path: str, lang: str) -> None:
-    console = Console()
-    with open(html_path, 'r', encoding='utf-8') as f:
-        html = ProblemHTML(f.read())
-    md = html.make_problem_markdown(lang)
-    file_without_ext = os.path.splitext(html_path)[0]
-    md_path = file_without_ext + FILE_EXTENSIONS[Lang.MARKDOWN]
-
-    with open(md_path, 'w', encoding='utf-8') as f:
-        f.write(md)
-        console.print('[green][+][/green] Markdownファイルを作成しました.')
-
-
-def print_markdown(md_path: str) -> None:
-    console = Console()
-    with open(md_path, 'r', encoding='utf-8') as f:
-        md = f.read()
-    console.print(Markdown(md))
-
-
-def markdown(*args: str, lang: str = 'ja', save: bool = False) -> None:
-    if save:
-        execute_files(
-            *args,
-            func=lambda html_path: save_markdown(html_path, lang),
-            target_filetypes=[Lang.HTML],
-        )
-    else:
-        execute_files(*args, func=print_markdown, target_filetypes=[Lang.MARKDOWN])

atcoderstudybooster-0.3.1/atcdr/util/execute.py

@@ -1,63 +0,0 @@
-import os
-from typing import Callable, List
-
-import questionary as q
-from rich import print
-
-from atcdr.util.filetype import FILE_EXTENSIONS, Filename, Lang
-
-
-def execute_files(
-    *args: str, func: Callable[[Filename], None], target_filetypes: List[Lang]
-) -> None:
-    target_extensions = [FILE_EXTENSIONS[lang] for lang in target_filetypes]
-
-    files = [
-        file
-        for file in os.listdir('.')
-        if os.path.isfile(file) and os.path.splitext(file)[1] in target_extensions
-    ]
-
-    if not files:
-        print(
-            '対象のファイルが見つかりません.\n対象ファイルが存在するディレクトリーに移動してから実行してください。'
-        )
-        return
-
-    if not args:
-        if len(files) == 1:
-            func(files[0])
-        else:
-            target_file = q.select(
-                message='複数のファイルが見つかりました.ファイルを選択してください:',
-                choices=[q.Choice(title=file, value=file) for file in files],
-                instruction='\n 十字キーで移動, [enter]で実行',
-                pointer='❯❯❯',
-                qmark='',
-                style=q.Style(
-                    [
-                        ('qmark', 'fg:#2196F3 bold'),
-                        ('question', 'fg:#2196F3 bold'),
-                        ('answer', 'fg:#FFB300 bold'),
-                        ('pointer', 'fg:#FFB300 bold'),
-                        ('highlighted', 'fg:#FFB300 bold'),
-                        ('selected', 'fg:#FFB300 bold'),
-                    ]
-                ),
-            ).ask()
-            list(map(func, [target_file]))
-    else:
-        target_files = set()
-        for arg in args:
-            if arg == '*':
-                target_files.update(files)
-            elif arg.startswith('*.'):
-                ext = arg[1:]  # ".py" のような拡張子を取得
-                target_files.update(file for file in files if file.endswith(ext))
-            else:
-                if arg in files:
-                    target_files.add(arg)
-                else:
-                    print(f'エラー: {arg} は存在しません。')
-
-        list(map(func, target_files))