AtCoderStudyBooster 0.2__py3-none-any.whl → 0.3.1__py3-none-any.whl

atcdr/util/execute.py

@@ -8,56 +8,56 @@ from atcdr.util.filetype import FILE_EXTENSIONS, Filename, Lang
 
 
 def execute_files(
-	*args: str, func: Callable[[Filename], None], target_filetypes: List[Lang]
+    *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))
+    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))

atcdr/util/filetype.py

@@ -1,3 +1,4 @@
+import os
 from enum import Enum
 from typing import Dict, List, TypeAlias
 
@@ -7,85 +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, Extension] = {
-	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',
+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',
 }
 
+# ドキュメント言語のリスト
 DOCUMENT_LANGUAGES: List[Lang] = [
-	Lang.HTML,
-	Lang.MARKDOWN,
-	Lang.JSON,
+    Lang.HTML,
+    Lang.MARKDOWN,
+    Lang.JSON,
 ]
 
-# ソースコードファイルと言語のリスト
-SOURCE_LANGUAGES: List[Lang] = [
-	Lang.PYTHON,
-	Lang.JAVASCRIPT,
-	Lang.JAVA,
-	Lang.C,
-	Lang.CPP,
-	Lang.CSHARP,
-	Lang.RUBY,
-	Lang.PHP,
-	Lang.GO,
-	Lang.RUST,
+# コンパイル型言語のリスト
+COMPILED_LANGUAGES: List[Lang] = [
+    Lang.JAVA,
+    Lang.C,
+    Lang.CPP,
+    Lang.CSHARP,
+    Lang.GO,
+    Lang.RUST,
+]
+
+# インタプリター型言語のリスト
+INTERPRETED_LANGUAGES: List[Lang] = [
+    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) -> 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