AtCoderStudyBooster 0.3__py3-none-any.whl → 0.3.2__py3-none-any.whl

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

atcdr/util/filetype.py

@@ -1,6 +1,6 @@
 import os
 from enum import Enum
-from typing import Dict, List, Optional, TypeAlias
+from typing import Dict, List, TypeAlias
 
 # ファイル名と拡張子の型エイリアスを定義
 Filename: TypeAlias = str
@@ -8,98 +8,98 @@ Extension: TypeAlias = str
 
 
 class Lang(Enum):
-	PYTHON = 'Python'
-	JAVASCRIPT = 'JavaScript'
-	JAVA = 'Java'
-	C = 'C'
-	CPP = 'C++'
-	CSHARP = 'C#'
-	RUBY = 'Ruby'
-	PHP = 'php'
-	GO = 'Go'
-	RUST = 'Rust'
-	HTML = 'HTML'
-	MARKDOWN = 'markdown'
-	JSON = 'json'
+    PYTHON = 'Python'
+    JAVASCRIPT = 'JavaScript'
+    JAVA = 'Java'
+    C = 'C'
+    CPP = 'C++'
+    CSHARP = 'C#'
+    RUBY = 'Ruby'
+    PHP = 'php'
+    GO = 'Go'
+    RUST = 'Rust'
+    HTML = 'HTML'
+    MARKDOWN = 'markdown'
+    JSON = 'json'
 
 
 # ファイル拡張子と対応する言語の辞書
 FILE_EXTENSIONS: Dict[Lang, str] = {
-	Lang.PYTHON: '.py',
-	Lang.JAVASCRIPT: '.js',
-	Lang.JAVA: '.java',
-	Lang.C: '.c',
-	Lang.CPP: '.cpp',
-	Lang.CSHARP: '.cs',
-	Lang.RUBY: '.rb',
-	Lang.PHP: '.php',
-	Lang.GO: '.go',
-	Lang.RUST: '.rs',
-	Lang.HTML: '.html',
-	Lang.MARKDOWN: '.md',
-	Lang.JSON: '.json',
+    Lang.PYTHON: '.py',
+    Lang.JAVASCRIPT: '.js',
+    Lang.JAVA: '.java',
+    Lang.C: '.c',
+    Lang.CPP: '.cpp',
+    Lang.CSHARP: '.cs',
+    Lang.RUBY: '.rb',
+    Lang.PHP: '.php',
+    Lang.GO: '.go',
+    Lang.RUST: '.rs',
+    Lang.HTML: '.html',
+    Lang.MARKDOWN: '.md',
+    Lang.JSON: '.json',
 }
 
 # ドキュメント言語のリスト
 DOCUMENT_LANGUAGES: List[Lang] = [
-	Lang.HTML,
-	Lang.MARKDOWN,
-	Lang.JSON,
+    Lang.HTML,
+    Lang.MARKDOWN,
+    Lang.JSON,
 ]
 
 # コンパイル型言語のリスト
 COMPILED_LANGUAGES: List[Lang] = [
-	Lang.JAVA,
-	Lang.C,
-	Lang.CPP,
-	Lang.CSHARP,
-	Lang.GO,
-	Lang.RUST,
+    Lang.JAVA,
+    Lang.C,
+    Lang.CPP,
+    Lang.CSHARP,
+    Lang.GO,
+    Lang.RUST,
 ]
 
 # インタプリター型言語のリスト
 INTERPRETED_LANGUAGES: List[Lang] = [
-	Lang.PYTHON,
-	Lang.JAVASCRIPT,
-	Lang.RUBY,
-	Lang.PHP,
+    Lang.PYTHON,
+    Lang.JAVASCRIPT,
+    Lang.RUBY,
+    Lang.PHP,
 ]
 
 
 def str2lang(lang: str) -> Lang:
-	lang_map = {
-		'py': Lang.PYTHON,
-		'python': Lang.PYTHON,
-		'js': Lang.JAVASCRIPT,
-		'javascript': Lang.JAVASCRIPT,
-		'java': Lang.JAVA,
-		'c': Lang.C,
-		'cpp': Lang.CPP,
-		'c++': Lang.CPP,
-		'csharp': Lang.CSHARP,
-		'cs': Lang.CSHARP,
-		'c#': Lang.CSHARP,
-		'rb': Lang.RUBY,
-		'ruby': Lang.RUBY,
-		'php': Lang.PHP,
-		'go': Lang.GO,
-		'rs': Lang.RUST,
-		'rust': Lang.RUST,
-		'html': Lang.HTML,
-		'md': Lang.MARKDOWN,
-		'markdown': Lang.MARKDOWN,
-		'json': Lang.JSON,
-	}
-	return lang_map[lang.lower()]
+    lang_map = {
+        'py': Lang.PYTHON,
+        'python': Lang.PYTHON,
+        'js': Lang.JAVASCRIPT,
+        'javascript': Lang.JAVASCRIPT,
+        'java': Lang.JAVA,
+        'c': Lang.C,
+        'cpp': Lang.CPP,
+        'c++': Lang.CPP,
+        'csharp': Lang.CSHARP,
+        'cs': Lang.CSHARP,
+        'c#': Lang.CSHARP,
+        'rb': Lang.RUBY,
+        'ruby': Lang.RUBY,
+        'php': Lang.PHP,
+        'go': Lang.GO,
+        'rs': Lang.RUST,
+        'rust': Lang.RUST,
+        'html': Lang.HTML,
+        'md': Lang.MARKDOWN,
+        'markdown': Lang.MARKDOWN,
+        'json': Lang.JSON,
+    }
+    return lang_map[lang.lower()]
 
 
 def lang2str(lang: Lang) -> str:
-	return lang.value
+    return lang.value
 
 
-def detect_language(path: str) -> Optional[Lang]:
-	ext = os.path.splitext(path)[1]  # ファイルの拡張子を取得
-	lang = next(
-		(lang for lang, extension in FILE_EXTENSIONS.items() if extension == ext), None
-	)
-	return lang
+def detect_language(path: str) -> Lang:
+    ext = os.path.splitext(path)[1]  # ファイルの拡張子を取得
+    lang = next(
+        (lang for lang, extension in FILE_EXTENSIONS.items() if extension == ext)
+    )
+    return lang

atcdr/util/gpt.py

@@ -1,112 +1,118 @@
 import os
+from enum import Enum
 from typing import Dict, List, Optional
 
 import requests
 
-from atcdr.util.cost import CostType, Currency, Model, Rate
+
+class Model(Enum):
+    GPT4O = 'gpt-4o'
+    GPT41 = 'gpt-4.1'
+    GPT41_MINI = 'gpt-4.1-mini'
+    GPT41_NANO = 'gpt-4.1-nano'
+    GPT4O_MINI = 'gpt-4o-mini'
+    O1_PREVIEW = 'o1-preview'
+    O1 = 'o1'
+    O3 = 'o3'
+    O1_MINI = 'o1-mini'
+    O3_MINI = 'o3-mini'
+    O4_MINI = 'o4-mini'
 
 
 def set_api_key() -> Optional[str]:
-	api_key = os.getenv('OPENAI_API_KEY')
-	if api_key and validate_api_key(api_key):
-		return api_key
-	elif api_key:
-		print('環境変数に設定されているAPIキーの検証に失敗しました ')
-	else:
-		pass
-
-	api_key = input(
-		'https://platform.openai.com/api-keys からchatGPTのAPIキーを入手しましょう。\nAPIキー入力してください: '
-	)
-	if validate_api_key(api_key):
-		print('APIキーのテストに成功しました。')
-		print('以下, ~/.zshrcにAPIキーを保存しますか? [y/n]')
-		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} に保存しました。次回シェル起動時に読み込まれます。'
-			)
-		os.environ['OPENAI_API_KEY'] = api_key
-		return api_key
-	else:
-		print('コード生成にはAPIキーが必要です。')
-		return None
+    api_key = os.getenv('OPENAI_API_KEY')
+    if api_key and validate_api_key(api_key):
+        return api_key
+    elif api_key:
+        print('環境変数に設定されているAPIキーの検証に失敗しました ')
+    else:
+        pass
+
+    api_key = input(
+        'https://platform.openai.com/api-keys からchatGPTのAPIキーを入手しましょう。\nAPIキー入力してください: '
+    )
+    if validate_api_key(api_key):
+        print('APIキーのテストに成功しました。')
+        print('以下, ~/.zshrcにAPIキーを保存しますか? [y/n]')
+        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} に保存しました。次回シェル起動時に読み込まれます。'
+            )
+        os.environ['OPENAI_API_KEY'] = api_key
+        return api_key
+    else:
+        print('コード生成にはAPIキーが必要です。')
+        return None
 
 
 def validate_api_key(api_key: str) -> bool:
-	headers = {
-		'Content-Type': 'application/json',
-		'Authorization': f'Bearer {api_key}',
-	}
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {api_key}',
+    }
 
-	response = requests.get('https://api.openai.com/v1/models', headers=headers)
+    response = requests.get('https://api.openai.com/v1/models', headers=headers)
 
-	if response.status_code == 200:
-		return True
-	else:
-		print('APIキーの検証に失敗しました。')
-		return False
+    if response.status_code == 200:
+        return True
+    else:
+        print('APIキーの検証に失敗しました。')
+        return False
 
 
 class ChatGPT:
-	API_URL = 'https://api.openai.com/v1/chat/completions'
-
-	# APIの使い方 https://platform.openai.com/docs/api-reference/making-requests
-	def __init__(
-		self,
-		api_key: Optional[str] = None,
-		model: Model = Model.GPT4O_MINI,
-		max_tokens: int = 3000,
-		temperature: float = 0.7,
-		messages: Optional[List[Dict[str, str]]] = None,
-		system_prompt: str = 'You are a helpful assistant.',
-	) -> None:
-		self.api_key = api_key or os.getenv('OPENAI_API_KEY')
-		self.model = model
-		self.max_tokens = max_tokens
-		self.temperature = temperature
-		self.messages = (
-			messages
-			if messages is not None
-			else [{'role': 'system', 'content': system_prompt}]
-		)
-
-		self.sum_cost: Currency = Currency(usd=0)
-		self.__headers = {
-			'Content-Type': 'application/json',
-			'Authorization': f'Bearer {self.api_key}',
-		}
-
-	def tell(self, message: str) -> str:
-		self.messages.append({'role': 'user', 'content': message})
-
-		settings = {
-			'model': self.model.value,
-			'messages': self.messages,
-			'max_tokens': self.max_tokens,
-			'temperature': self.temperature,
-		}
-
-		response = requests.post(self.API_URL, headers=self.__headers, json=settings)
-		responsej = response.json()
-		try:
-			reply = responsej['choices'][0]['message']['content']
-		except KeyError:
-			print('Error:レスポンスの形式が正しくありません. \n' + str(responsej))
-			return 'Error: Unable to retrieve response.'
-
-		self.messages.append({'role': 'assistant', 'content': reply})
-
-		usage = responsej['usage']
-		input_tokens = usage.get('prompt_tokens', 0)
-		output_tokens = usage.get('completion_tokens', 0)
-		self.sum_cost += Rate.calc_cost(
-			model=self.model, cost_type=CostType.INPUT, token_count=input_tokens
-		)
-		self.sum_cost += Rate.calc_cost(
-			model=self.model, cost_type=CostType.OUTPUT, token_count=output_tokens
-		)
-
-		return reply
+    API_URL = 'https://api.openai.com/v1/chat/completions'
+
+    # APIの使い方 https://platform.openai.com/docs/api-reference/making-requests
+    def __init__(
+        self,
+        api_key: Optional[str] = None,
+        model: Model = Model.GPT41_MINI,
+        max_tokens: int = 3000,
+        temperature: float = 0.7,
+        messages: Optional[List[Dict[str, str]]] = None,
+        system_prompt: str = 'You are a helpful assistant.',
+    ) -> None:
+        self.api_key = api_key or os.getenv('OPENAI_API_KEY')
+        self.model = model
+        self.max_tokens = max_tokens
+        self.temperature = temperature
+        self.messages = (
+            messages
+            if messages is not None
+            else [{'role': 'system', 'content': system_prompt}]
+        )
+
+        self.__headers = {
+            'Content-Type': 'application/json',
+            'Authorization': f'Bearer {self.api_key}',
+        }
+
+    def tell(self, message: str) -> str:
+        self.messages.append({'role': 'user', 'content': message})
+
+        settings = {
+            'model': self.model.value,
+            'messages': self.messages,
+            'max_tokens': self.max_tokens,
+            'temperature': self.temperature,
+        }
+
+        response = requests.post(self.API_URL, headers=self.__headers, json=settings)
+        responsej = response.json()
+        try:
+            reply = responsej['choices'][0]['message']['content']
+        except KeyError:
+            print('Error:レスポンスの形式が正しくありません. \n' + str(responsej))
+            return 'Error: Unable to retrieve response.'
+
+        self.messages.append({'role': 'assistant', 'content': reply})
+
+        # usage = responsej['usage']
+        # input_tokens = usage.get('prompt_tokens', 0)
+        # output_tokens = usage.get('completion_tokens', 0)
+
+        return reply