AtCoderStudyBooster 0.1.0__py3-none-any.whl

atcdr/download.py

@@ -0,0 +1,266 @@
+import os
+import re
+import time
+from dataclasses import dataclass
+from enum import Enum
+from typing import Callable, List, Match, Optional, Union, cast
+
+import requests
+
+from atcdr.util.problem import make_problem_markdown
+
+
+class Diff(Enum):
+	A = 'A'
+	B = 'B'
+	C = 'C'
+	D = 'D'
+	E = 'E'
+	F = 'F'
+	G = 'G'
+
+
+@dataclass
+class Problem:
+	number: int
+	difficulty: Diff
+
+
+def get_problem_html(problem: Problem) -> Optional[str]:
+	url = f'https://atcoder.jp/contests/abc{problem.number}/tasks/abc{problem.number}_{problem.difficulty.value.lower()}'
+	response = requests.get(url)
+	retry_attempts = 3
+	retry_wait = 1  # 1 second
+
+	for _ in range(retry_attempts):
+		response = requests.get(url)
+		if response.status_code == 200:
+			return response.text
+		elif response.status_code == 429:
+			print(
+				f'[Error{response.status_code}] 再試行します. abc{problem.number} {problem.difficulty.value}'
+			)
+			time.sleep(retry_wait)
+		elif 300 <= response.status_code < 400:
+			print(
+				f'[Erroe{response.status_code}] リダイレクトが発生しました。abc{problem.number} {problem.difficulty.value}'
+			)
+		elif 400 <= response.status_code < 500:
+			print(
+				f'[Error{response.status_code}] 問題が見つかりません。abc{problem.number} {problem.difficulty.value}'
+			)
+			break
+		elif 500 <= response.status_code < 600:
+			print(
+				f'[Error{response.status_code}] サーバーエラーが発生しました。abc{problem.number} {problem.difficulty.value}'
+			)
+			break
+		else:
+			print(
+				f'[Error{response.status_code}] abc{problem.number} {problem.difficulty.value}に対応するHTMLファイルを取得できませんでした。'
+			)
+			break
+	return None
+
+
+def repair_html(html: str) -> str:
+	html = 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"')
+	return html
+
+
+def get_title_from_html(html: str) -> Optional[str]:
+	title_match: Optional[Match[str]] = re.search(
+		r'<title>(?:.*?-\s*)?([^<]*)</title>', html, re.IGNORECASE | re.DOTALL
+	)
+	if title_match:
+		title: str = title_match.group(1).replace(' ', '')
+		title = re.sub(r'[\\/*?:"<>| ]', '', title)
+		return title
+	return None
+
+
+def save_file(file_path: str, html: str) -> None:
+	with open(file_path, 'w', encoding='utf-8') as file:
+		file.write(html)
+	print(f'[+] ファイルを保存しました :{file_path}')
+
+
+def mkdir(path: str) -> None:
+	if not os.path.exists(path):
+		os.makedirs(path)
+		print(f'[+] フォルダー: {path} を作成しました')
+
+
+class GenerateMode:
+	@staticmethod
+	def gene_path_on_diff(base: str, number: int, diff: Diff) -> str:
+		return os.path.join(base, diff.name, str(number))
+
+	@staticmethod
+	def gene_path_on_num(base: str, number: int, diff: Diff) -> str:
+		return os.path.join(base, str(number), diff.name)
+
+
+def generate_problem_directory(
+	base_path: str, problems: List[Problem], gene_path: Callable[[str, int, Diff], str]
+) -> None:
+	for problem in problems:
+		dir_path = gene_path(base_path, problem.number, problem.difficulty)
+
+		html = get_problem_html(problem)
+		if html is None:
+			continue
+
+		title = get_title_from_html(html)
+		if title is None:
+			print('[Error] タイトルが取得できませんでした')
+			title = f'problem{problem.number}{problem.difficulty.value}'
+
+		mkdir(dir_path)
+		repaired_html = repair_html(html)
+
+		html_path = os.path.join(dir_path, f'{title}.html')
+		save_file(html_path, repaired_html)
+		md = make_problem_markdown(html, 'ja')
+		save_file(os.path.join(dir_path, f'{title}.md'), md)
+
+
+def parse_range(range_str: str) -> List[int]:
+	match = re.match(r'^(\d+)\.\.(\d+)$', range_str)
+	if match:
+		start, end = map(int, match.groups())
+		return list(range(start, end + 1))
+	else:
+		raise ValueError('Invalid range format')
+
+
+def parse_diff_range(range_str: str) -> List[Diff]:
+	match = re.match(r'^([A-F])\.\.([A-F])$', range_str)
+	if match:
+		start, end = match.groups()
+		start_index = ord(start) - ord('A')
+		end_index = ord(end) - ord('A')
+		if start_index <= end_index:
+			return [Diff(chr(i + ord('A'))) for i in range(start_index, end_index + 1)]
+	raise ValueError('A..C の形式になっていません')
+
+
+def convert_arg(arg: Union[str, int]) -> Union[List[int], List[Diff]]:
+	if isinstance(arg, int):
+		return [arg]
+	elif isinstance(arg, str):
+		if arg.isdigit():
+			return [int(arg)]
+		elif arg in Diff.__members__:
+			return [Diff[arg]]
+		elif re.match(r'^\d+\.\.\d+$', arg):
+			return parse_range(arg)
+		elif re.match(r'^[A-F]\.\.[A-F]$', arg):
+			return parse_diff_range(arg)
+	raise ValueError(f'{arg}は認識できません')
+
+
+def are_all_integers(args: Union[List[int], List[Diff]]) -> bool:
+	return all(isinstance(arg, int) for arg in args)
+
+
+def are_all_diffs(args: Union[List[int], List[Diff]]) -> bool:
+	return all(isinstance(arg, Diff) for arg in args)
+
+
+def download(
+	first: Union[str, int, None] = None,
+	second: Union[str, int, None] = None,
+	base_path: str = '.',
+) -> None:
+	if first is None:
+		main()
+		return
+
+	first_args = convert_arg(str(first))
+	if second is None:
+		if isinstance(first, Diff):
+			raise ValueError(
+				"""難易度だけでなく, 問題番号も指定してコマンドを実行してください.
+                             例 atcdr -d A 120  : A問題の120をダウンロードます
+                             例 atcdr -d A 120..130  : A問題の120から130をダウンロードます
+                             """
+			)
+		second_args: Union[List[int], List[Diff]] = list(Diff)
+	else:
+		second_args = convert_arg(str(second))
+
+	if are_all_integers(first_args) and are_all_diffs(second_args):
+		first_args_int = cast(List[int], first_args)
+		second_args_diff = cast(List[Diff], second_args)
+		problems = [
+			Problem(number, diff)
+			for number in first_args_int
+			for diff in second_args_diff
+		]
+		generate_problem_directory(base_path, problems, GenerateMode.gene_path_on_num)
+	elif are_all_diffs(first_args) and are_all_integers(second_args):
+		first_args_diff = cast(List[Diff], first_args)
+		second_args_int = cast(List[int], second_args)
+		problems = [
+			Problem(number, diff)
+			for diff in first_args_diff
+			for number in second_args_int
+		]
+		generate_problem_directory(base_path, problems, GenerateMode.gene_path_on_diff)
+	else:
+		raise ValueError(
+			"""次のような形式で問題を指定してください
+                            例 atcdr -d A 120..130  : A問題の120から130をダウンロードします
+                            例 atcdr -d 120         : ABCのコンテストの問題をダウンロードします
+                         """
+		)
+
+
+def main() -> None:
+	print('AtCoderの問題のHTMLファイルをダウンロードします')
+	print(
+		"""
+    1. 番号の範囲を指定してダウンロードする
+    2. 1ファイルだけダウンロードする
+    q: 終了
+    """
+	)
+
+	choice = input('選択してください: ')
+
+	if choice == '1':
+		start_end = input(
+			'開始と終了のコンテストの番号をスペースで区切って指定してください (例: 223 230): '
+		)
+		start, end = map(int, start_end.split(' '))
+		difficulty = Diff[
+			input(
+				'ダウンロードする問題の難易度を指定してください (例: A, B, C): '
+			).upper()
+		]
+		problem_list = [Problem(number, difficulty) for number in range(start, end + 1)]
+		generate_problem_directory('.', problem_list, GenerateMode.gene_path_on_diff)
+	elif choice == '2':
+		number = int(input('コンテストの番号を指定してください: '))
+		difficulty = Diff[
+			input(
+				'ダウンロードする問題の難易度を指定してください (例: A, B, C): '
+			).upper()
+		]
+		generate_problem_directory(
+			'.', [Problem(number, difficulty)], GenerateMode.gene_path_on_diff
+		)
+	elif choice == 'q':
+		print('終了します')
+	else:
+		print('無効な選択です')
+
+
+if __name__ == '__main__':
+	main()

atcdr/generate.py

@@ -0,0 +1,167 @@
+import json
+import os
+import re
+
+from atcdr.test import (
+	ResultStatus,
+	create_testcases_from_html,
+	judge_code_from,
+	render_result,
+)
+from atcdr.util.filename import (
+	FILE_EXTENSIONS,
+	Filename,
+	Lang,
+	execute_files,
+	lang2str,
+	str2lang,
+)
+from atcdr.util.gpt import ChatGPT, set_api_key
+from atcdr.util.problem import make_problem_markdown
+
+
+def get_code_from_gpt_output(output: str) -> str:
+	pattern = re.compile(r'```(?:\w+)?\s*(.*?)\s*```', re.DOTALL)
+	match = pattern.search(output)
+	return match.group(1) if match else ''
+
+
+def generate_code(file: Filename, lang: Lang) -> None:
+	with open(file, 'r') as f:
+		html_content = f.read()
+	md = make_problem_markdown(html_content, 'en')
+
+	if set_api_key() is None:
+		return
+	gpt = ChatGPT(
+		system_prompt=f"""You are an excellent programmer. You solve problems in competitive programming.When a user provides you with a problem from a programming contest called AtCoder, including the Problem,Constraints, Input, Output, Input Example, and Output Example, please carefully consider these and solve the problem.Make sure that your output code block contains no more than two blocks. Pay close attention to the Input, Input Example, Output, and Output Example.Create the solution in {lang2str(lang)}.""",
+	)
+
+	reply = gpt.tell(md)
+	code = get_code_from_gpt_output(reply)
+	print(f'AI利用にかかったAPIコスト: {gpt.sum_cost}')
+
+	saved_filename = (
+		os.path.splitext(file)[0] + f'_by_{gpt.model.value}' + FILE_EXTENSIONS[lang]
+	)
+	with open(saved_filename, 'w') as f:
+		print(f'[+]:{gpt.model.value}の出力したコードを保存しました：{f.name}')
+		f.write(code)
+
+
+def generate_template(file: Filename, lang: Lang) -> None:
+	with open(file, 'r') as f:
+		html_content = f.read()
+	md = make_problem_markdown(html_content, 'en')
+
+	if set_api_key() is None:
+		return
+	gpt = ChatGPT(
+		system_prompt='You are a highly skilled programmer. Your role is to create a template code for competitive programming.',
+		temperature=0.0,
+	)
+
+	propmpt = f"""
+The user will provide a problem from a programming contest called AtCoder. This problem will include the Problem Statement, Constraints, Input, Output, Input Example, and Output Example. You should focus on the Constraints and Input sections to create the template in {lang2str(lang)}.
+
+- First, create the part of the code that handles input. Then, you should read ###Input Block and ###Constraints Block.
+- After receiving the input, define variables in the program by reading ###Constraints Block and explain how to use the variables in the comment of your code block with example.
+- Last, define variables needed for output. Then you should read ###Output Block and ###Constraints Block.
+
+You must not solve the problem. Please faithfully reproduce the variable names defined in the problem.
+    """
+	reply = gpt.tell(md + propmpt)
+	code = get_code_from_gpt_output(reply)
+	print(f'AI利用にかかったAPIコスト:{gpt.sum_cost}')
+
+	savaed_filename = os.path.splitext(file)[0] + FILE_EXTENSIONS[lang]
+	with open(savaed_filename, 'w') as f:
+		print(f'[+]:テンプレートファイル{savaed_filename}を作成しました.')
+		f.write(code)
+
+
+def solve_problem(file: Filename, lang: Lang) -> None:
+	with open(file, 'r') as f:
+		html_content = f.read()
+	md = make_problem_markdown(html_content, 'en')
+	labeled_cases = create_testcases_from_html(html_content)
+
+	if set_api_key() is None:
+		return
+	gpt = ChatGPT(
+		system_prompt=f"""You are a brilliant programmer. Your task is to solve an AtCoder problem. AtCoder is a platform that hosts programming competitions where participants write programs to solve algorithmic challenges.Please solve the problem in {lang2str(lang)}.""",
+	)
+
+	file_without_ext = os.path.splitext(file)[0]
+
+	reply = gpt.tell(md)
+
+	for i in range(1, 4):
+		code = get_code_from_gpt_output(reply)
+
+		saved_filename = (
+			f'{i}_'
+			+ file_without_ext
+			+ f'_by_{gpt.model.value}'
+			+ FILE_EXTENSIONS[lang]
+		)
+		with open(saved_filename, 'w') as f:
+			print(f'[+]:{gpt.model.value}の出力したコードを保存しました：{f.name}')
+			f.write(code)
+
+		labeled_results = judge_code_from(labeled_cases, saved_filename)
+		test_report = '\n'.join(render_result(lresult) for lresult in labeled_results)
+
+		print(f'{i}回目のコード生成でのテスト結果:---')
+		print(test_report)
+
+		if all(
+			labeled_result.result.passed == ResultStatus.AC
+			for labeled_result in labeled_results
+		):
+			print('コードのテストに成功!')
+			break
+		else:
+			reply = gpt.tell(f"""The following is the test report for the code you provided:
+{test_report}
+Please provide an updated version of the code in {lang2str(lang)}.""")
+
+	with open(
+		'log_'
+		+ file_without_ext
+		+ f'_by_{gpt.model.value}'
+		+ FILE_EXTENSIONS[Lang.JSON],
+		'w',
+	) as f:
+		print(f'[+]:{gpt.model.value}の出力のログを保存しました：{f.name}')
+		f.write(json.dumps(gpt.messages, indent=2))
+	print(f'AI利用にかかったAPIコスト:{gpt.sum_cost}')
+	return
+
+
+def generate(
+	*source: str,
+	lang: str = 'Python',
+	without_test: bool = False,
+	template: bool = False,
+) -> None:
+	la = str2lang(lang)
+
+	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),
+			target_filetypes=[Lang.HTML],
+		)
+	else:
+		execute_files(
+			*source,
+			func=lambda file: solve_problem(file, la),
+			target_filetypes=[Lang.HTML],
+		)

atcdr/main.py

@@ -0,0 +1,35 @@
+from importlib.metadata import metadata
+
+import fire  # type: ignore
+
+from atcdr.download import download
+from atcdr.generate import generate
+from atcdr.open import open_files
+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,
+	'--version': get_version,
+	'-v': get_version,
+}
+
+
+def main():
+	fire.Fire(MAP_COMMANDS)
+
+
+if __name__ == '__main__':
+	main()

atcdr/open.py

@@ -0,0 +1,36 @@
+import webbrowser
+
+from bs4 import BeautifulSoup as bs
+from bs4.element import Tag
+
+from atcdr.util.filename import Lang, execute_files
+
+
+def find_link_from(html: str) -> str | None:
+	soup = bs(html, 'html.parser')
+	meta_tag = soup.find('meta', property='og:url')
+	if isinstance(meta_tag, Tag) and 'content' in meta_tag.attrs:
+		content = meta_tag['content']
+		if isinstance(content, list):
+			return content[0]  # 必要に応じて、最初の要素を返す
+		return content
+	return None
+
+
+def open_html(file: str) -> None:
+	try:
+		with open(file, 'r') as f:
+			html_content = f.read()
+	except FileNotFoundError:
+		print(f"HTMLファイル '{file}' が見つかりません。")
+		return
+
+	url = find_link_from(html_content)
+	if url:
+		webbrowser.open(url)
+	else:
+		print('URLが見つかりませんでした。')
+
+
+def open_files(*args: str) -> None:
+	execute_files(*args, func=open_html, target_filetypes=[Lang.HTML])

atcdr/test.py

@@ -0,0 +1,276 @@
+import os
+import subprocess
+import tempfile
+import time
+from dataclasses import dataclass
+from enum import Enum
+from typing import Callable, Dict, List, Optional, Union
+
+import colorama
+from bs4 import BeautifulSoup as bs
+from colorama import Fore
+
+from atcdr.util.filename import FILE_EXTENSIONS, SOURCE_LANGUAGES, Lang, execute_files
+
+colorama.init(autoreset=True)
+
+
+@dataclass
+class TestCase:
+	input: str
+	output: str
+
+
+@dataclass
+class LabeledTestCase:
+	label: str
+	case: TestCase
+
+
+class ResultStatus(Enum):
+	CE = 'Compilation Error'
+	MLE = 'Memory Limit Exceeded'
+	TLE = 'Time Limit Exceeded'
+	RE = 'Runtime Error'
+	WA = 'Wrong Answer'
+	AC = 'Accepted'
+
+
+@dataclass
+class TestCaseResult:
+	output: str
+	executed_time: Union[int, None]
+	# memory_usage: Union[int, None]
+	passed: ResultStatus
+
+
+@dataclass
+class LabeledTestCaseResult:
+	label: str
+	testcase: TestCase
+	# TODO : 実はラベル自体を使わない方がいいかもしれない.ラベルという概念が削除してプリントするときに適当にTest1, Test2と適当に名前をつけてもいいかも.
+	result: TestCaseResult
+
+
+def create_testcases_from_html(html: str) -> List[LabeledTestCase]:
+	soup = bs(html, 'html.parser')
+	test_cases = []
+
+	for i in range(1, 20):
+		sample_input_section = soup.find('h3', text=f'Sample Input {i}')
+		sample_output_section = soup.find('h3', text=f'Sample Output {i}')
+		if not sample_input_section or not sample_output_section:
+			break
+
+		sample_input_pre = sample_input_section.find_next('pre')
+		sample_output_pre = sample_output_section.find_next('pre')
+
+		sample_input = (
+			sample_input_pre.get_text(strip=True)
+			if sample_input_pre is not None
+			else ''
+		)
+		sample_output = (
+			sample_output_pre.get_text(strip=True)
+			if sample_output_pre is not None
+			else ''
+		)
+
+		test_case = TestCase(input=sample_input, output=sample_output)
+		labeled_test_case = LabeledTestCase(label=f'Sample {i}', case=test_case)
+		test_cases.append(labeled_test_case)
+
+	return test_cases
+
+
+def run_code(cmd: list, case: TestCase) -> TestCaseResult:
+	try:
+		start_time = time.time()
+		proc = subprocess.run(
+			cmd, input=case.input, text=True, capture_output=True, timeout=4
+		)
+		end_time = time.time()
+
+		execution_time = int((end_time - start_time) * 1000)
+
+		if proc.returncode != 0:
+			return TestCaseResult(
+				output=proc.stderr, executed_time=None, passed=ResultStatus.RE
+			)
+
+		actual_output = proc.stdout.strip()
+		expected_output = case.output.strip()
+
+		if actual_output != expected_output:
+			return TestCaseResult(
+				output=actual_output,
+				executed_time=execution_time,
+				passed=ResultStatus.WA,
+			)
+
+		return TestCaseResult(
+			output=actual_output, executed_time=execution_time, passed=ResultStatus.AC
+		)
+	except subprocess.TimeoutExpired:
+		return TestCaseResult(
+			output='Time Limit Exceeded', executed_time=None, passed=ResultStatus.TLE
+		)
+	except Exception as e:
+		return TestCaseResult(output=str(e), executed_time=None, passed=ResultStatus.RE)
+
+
+def run_python(path: str, case: TestCase) -> TestCaseResult:
+	return run_code(['python3', path], case)
+
+
+def run_javascript(path: str, case: TestCase) -> TestCaseResult:
+	return run_code(['node', path], case)
+
+
+def run_c(path: str, case: TestCase) -> TestCaseResult:
+	with tempfile.NamedTemporaryFile(delete=True) as tmp:
+		exec_path = tmp.name
+		compile_result = subprocess.run(
+			['gcc', path, '-o', exec_path], capture_output=True, text=True
+		)
+		if compile_result.returncode != 0:
+			return TestCaseResult(
+				output=compile_result.stderr, executed_time=None, passed=ResultStatus.CE
+			)
+		if compile_result.stderr:
+			print(f'コンパイラーからのメッセージ\n{compile_result.stderr}')
+		return run_code([exec_path], case)
+
+
+def run_cpp(path: str, case: TestCase) -> TestCaseResult:
+	with tempfile.NamedTemporaryFile(delete=True) as tmp:
+		exec_path = tmp.name
+		compile_result = subprocess.run(
+			['g++', path, '-o', exec_path], capture_output=True, text=True
+		)
+		if compile_result.returncode != 0:
+			return TestCaseResult(
+				output=compile_result.stderr, executed_time=None, passed=ResultStatus.CE
+			)
+		if compile_result.stderr:
+			print(f'コンパイラーからのメッセージ\n{compile_result.stderr}')
+		return run_code([exec_path], case)
+
+
+def run_rust(path: str, case: TestCase) -> TestCaseResult:
+	with tempfile.NamedTemporaryFile(delete=True) as tmp:
+		exec_path = tmp.name
+		compile_result = subprocess.run(
+			['rustc', path, '-o', exec_path], capture_output=True, text=True
+		)
+		if compile_result.returncode != 0:
+			return TestCaseResult(
+				output=compile_result.stderr, executed_time=None, passed=ResultStatus.CE
+			)
+		if compile_result.stderr:
+			print(f'コンパイラーからのメッセージ\n{compile_result.stderr}')
+		return run_code([exec_path], case)
+
+
+def run_java(path: str, case: TestCase) -> TestCaseResult:
+	compile_result = subprocess.run(['javac', path], capture_output=True, text=True)
+	if compile_result.returncode != 0:
+		return TestCaseResult(
+			output=compile_result.stderr, executed_time=None, passed=ResultStatus.CE
+		)
+	class_file = os.path.splitext(path)[0]
+	try:
+		return run_code(['java', class_file], case)
+	finally:
+		class_path = class_file + '.class'
+		if os.path.exists(class_path):
+			os.remove(class_path)
+
+
+LANGUAGE_RUNNERS: Dict[Lang, Callable[[str, TestCase], TestCaseResult]] = {
+	Lang.PYTHON: run_python,
+	Lang.JAVASCRIPT: run_javascript,
+	Lang.C: run_c,
+	Lang.CPP: run_cpp,
+	Lang.RUST: run_rust,
+	Lang.JAVA: run_java,
+}
+
+
+def choose_lang(path: str) -> Optional[Callable[[str, TestCase], TestCaseResult]]:
+	ext = os.path.splitext(path)[1]
+	lang = next(
+		(lang for lang, extension in FILE_EXTENSIONS.items() if extension == ext), None
+	)
+	# lang が None でない場合のみ get を呼び出す
+	if lang is not None:
+		return LANGUAGE_RUNNERS.get(lang)
+	return None
+
+
+def judge_code_from(
+	lcases: List[LabeledTestCase], path: str
+) -> List[LabeledTestCaseResult]:
+	runner = choose_lang(path)
+	if runner is None:
+		raise ValueError(f'ランナーが見つかりませんでした。指定されたパス: {path}')
+
+	return [
+		LabeledTestCaseResult(lcase.label, lcase.case, runner(path, lcase.case))
+		for lcase in lcases
+	]
+
+
+CHECK_MARK = '\u2713'
+CROSS_MARK = '\u00d7'
+
+
+def render_result(lresult: LabeledTestCaseResult) -> str:
+	output = f'{Fore.CYAN}{lresult.label} of Test:\n'
+	result = lresult.result
+	testcase = lresult.testcase
+
+	if result.passed == ResultStatus.AC:
+		output += (
+			Fore.GREEN + f'{CHECK_MARK} Accepted !! Time: {result.executed_time} ms\n'
+		)
+	elif result.passed == ResultStatus.WA:
+		output += (
+			Fore.RED
+			+ f'{CROSS_MARK} Wrong Answer ! Time: {result.executed_time} ms\nOutput:\n{result.output}\nExpected Output:\n{testcase.output}\n'
+		)
+	elif result.passed == ResultStatus.RE:
+		output += Fore.YELLOW + f'[RE] Runtime Error\n  Output:\n{result.output}'
+	elif result.passed == ResultStatus.TLE:
+		output += Fore.YELLOW + '[TLE] Time Limit Exceeded\n'
+	elif result.passed == ResultStatus.CE:
+		output += Fore.YELLOW + f'[CE] Compile Error\n  Output:\n{result.output}'
+	elif result.passed == ResultStatus.MLE:
+		output += Fore.YELLOW + '[ME] Memory Limit Exceeded\n'
+
+	output += Fore.RESET
+
+	return output
+
+
+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問題のファイルが存在するディレクトリーに移動してから実行してください。'
+		)
+		return
+
+	with open(html_paths[0], 'r') as file:
+		html = file.read()
+
+	test_cases = create_testcases_from_html(html)
+	print(f'{path_of_code}をテストします。\n' + '-' * 20 + '\n')
+	test_results = judge_code_from(test_cases, path_of_code)
+	output = '\n'.join(render_result(lresult) for lresult in test_results)
+
+	print(output)
+
+
+def test(*args: str) -> None:
+	execute_files(*args, func=run_test, target_filetypes=SOURCE_LANGUAGES)

atcdr/util/cost.py

@@ -0,0 +1,120 @@
+from enum import Enum
+from typing import Optional
+
+import tiktoken
+import yfinance as yf  # type: ignore
+
+
+class Model(Enum):
+	GPT4O = 'gpt-4o'
+	GPT4O_MINI = 'gpt-4o-mini'
+
+
+class CostType(Enum):
+	INPUT = 'input'
+	OUTPUT = 'output'
+
+
+class Currency:
+	def __init__(
+		self, usd: Optional[float] = None, jpy: Optional[float] = None
+	) -> None:
+		self._exchange_rate = self.get_exchange_rate()
+		self._usd = usd if usd is not None else 0.0
+		self._jpy = jpy if jpy is not None else self.convert_usd_to_jpy(self._usd)
+
+	@staticmethod
+	def get_exchange_rate() -> float:
+		ticker = yf.Ticker('USDJPY=X')
+		todays_data = ticker.history(period='1d')
+		return todays_data['Close'].iloc[0]
+
+	def convert_usd_to_jpy(self, usd: float) -> float:
+		return usd * self._exchange_rate
+
+	def convert_jpy_to_usd(self, jpy: float) -> float:
+		return jpy / self._exchange_rate
+
+	@property
+	def usd(self) -> float:
+		return self._usd
+
+	@usd.setter
+	def usd(self, value: float) -> None:
+		self._usd = value
+		self._jpy = self.convert_usd_to_jpy(value)
+
+	@property
+	def jpy(self) -> float:
+		return self._jpy
+
+	@jpy.setter
+	def jpy(self, value: float) -> None:
+		self._jpy = value
+		self._usd = self.convert_jpy_to_usd(value)
+
+	def __add__(self, other: 'Currency') -> 'Currency':
+		return Currency(usd=self.usd + other.usd)
+
+	def __sub__(self, other: 'Currency') -> 'Currency':
+		return Currency(usd=self.usd - other.usd)
+
+	def __mul__(self, factor: float) -> 'Currency':
+		return Currency(usd=self.usd * factor)
+
+	def __truediv__(self, factor: float) -> 'Currency':
+		return Currency(usd=self.usd / factor)
+
+	def __eq__(self, other: object) -> bool:
+		if not isinstance(other, Currency):
+			return NotImplemented
+		epsilon = 1e-9  # 許容範囲
+		return abs(self.usd - other.usd) < epsilon
+
+	def __lt__(self, other: 'Currency') -> bool:
+		return self.usd < other.usd
+
+	def __repr__(self) -> str:
+		return f'Currency(usd={self.usd:.2f}, jpy={self.jpy:.2f})'
+
+	def __str__(self) -> str:
+		return f'USD: {self.usd:.2f}, JPY: {self.jpy:.2f}'
+
+
+class Rate:
+	_COST_RATES = {
+		Model.GPT4O: {CostType.INPUT: 5 / 1000**2, CostType.OUTPUT: 15 / 1000**2},
+		Model.GPT4O_MINI: {
+			CostType.INPUT: 0.15 / 1000**2,
+			CostType.OUTPUT: 0.60 / 1000**2,
+		},
+	}
+
+	@staticmethod
+	def calc_cost(model: Model, cost_type: CostType, token_count: int) -> Currency:
+		cost_in_usd = Rate._COST_RATES[model][cost_type] * token_count
+		return Currency(usd=cost_in_usd)
+
+
+class ApiCostCalculator:
+	def __init__(self, text: str, cost_type: CostType, model: Model) -> None:
+		self.text = text
+		self.cost_type = cost_type
+		self.model = model
+
+		# トークンモデルを取得
+		self.token_model = tiktoken.encoding_for_model(model.value)
+
+	@property
+	def token_count(self) -> int:
+		return len(self.token_model.encode(self.text))
+
+	@property
+	def cost(self) -> Currency:
+		return Rate.calc_cost(self.model, self.cost_type, self.token_count)
+
+	def __str__(self) -> str:
+		return f'Token count: {self.token_count}\nCost ({self.cost_type.value}): ${self.cost.usd:.2f} / ¥{self.cost.jpy:.2f}'
+
+	def __repr__(self) -> str:
+		return f'ApiCostCalculator(text={self.text}, cost_type={self.cost_type}, model={self.model})'

atcdr/util/filename.py

@@ -0,0 +1,137 @@
+import os
+from enum import Enum
+from typing import Callable, Dict, List, TypeAlias
+
+# ファイル名と拡張子の型エイリアスを定義
+Filename: TypeAlias = str
+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'
+
+
+# ファイル拡張子と対応する言語の辞書
+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',
+}
+
+DOCUMENT_LANGUAGES: List[Lang] = [
+	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,
+]
+
+
+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,
+		'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
+
+
+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:
+			print('複数のファイルが見つかりました。以下のファイルから選択してください:')
+			for i, file in enumerate(files):
+				print(f'{i + 1}. {file}')
+			choice = int(input('ファイル番号を入力してください: ')) - 1
+			if 0 <= choice < len(files):
+				func(files[choice])
+			else:
+				print('無効な選択です')
+	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/gpt.py

@@ -0,0 +1,112 @@
+import os
+from typing import Dict, List, Optional
+
+import requests
+
+from atcdr.util.cost import CostType, Currency, Model, Rate
+
+
+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
+
+
+def validate_api_key(api_key: str) -> bool:
+	headers = {
+		'Content-Type': 'application/json',
+		'Authorization': f'Bearer {api_key}',
+	}
+
+	response = requests.get('https://api.openai.com/v1/models', headers=headers)
+
+	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

atcdr/util/problem.py

@@ -0,0 +1,87 @@
+import re
+from enum import Enum
+from typing import Optional, Union
+
+from bs4 import BeautifulSoup as bs
+from bs4 import NavigableString, Tag
+from markdownify import MarkdownConverter  # type: ignore
+
+
+# TODO : そのうちgenerate.pyやtest.py, open.pyのHTMLのparse処理を全部まとめる
+class Lang(Enum):
+	JA = 'ja'
+	EN = 'en'
+
+
+class ProblemStruct:
+	def __init__(self) -> None:
+		self.problem_part: Optional[str] = None
+		self.condition_part: Optional[str] = None
+		self.io_part: Optional[str] = None
+		self.test_part: Optional[list[str]] = None
+
+	def divide_problem_part(self, task_statement: Union[Tag, NavigableString]) -> None:
+		if not isinstance(task_statement, Tag):
+			return
+
+		parts = task_statement.find_all('div', {'class': 'part'})
+
+		if len(parts) >= 2:
+			self.problem_part = str(parts[0])
+			self.condition_part = str(parts[1])
+
+		io_div = task_statement.find('div', {'class': 'io-style'})
+		if isinstance(io_div, Tag):
+			io_parts = io_div.find_all('div', {'class': 'part'})
+
+			if len(io_parts) > 0:
+				self.io_part = str(
+					io_parts[0]
+				)  # .find_all() はリストを返すので、str()でキャスト
+
+			# 2つ目以降のdivをtest_partに格納
+			self.test_part = [str(part) for part in io_parts[1:]]
+
+
+class CustomMarkdownConverter(MarkdownConverter):
+	def convert_var(self, el, text, convert_as_inline):
+		var_text = el.text.strip()
+		return f'\\({var_text}\\)'
+
+	def convert_pre(self, el, text, convert_as_inline):
+		pre_text = el.text.strip()
+		return f'```\n{pre_text}\n```'
+
+
+def custom_markdownify(html, **options):
+	return CustomMarkdownConverter(**options).convert(html)
+
+
+def remove_unnecessary_emptylines(md_text):
+	md_text = re.sub(r'\n\s*\n\s*\n+', '\n\n', md_text)
+	md_text = md_text.strip()
+	return md_text
+
+
+def abstract_problem_part(html_content: str, lang: str) -> str:
+	soup = bs(html_content, 'html.parser')
+	task_statement = soup.find('div', {'id': 'task-statement'})
+
+	if not isinstance(task_statement, Tag):
+		return ''
+
+	if lang == 'ja':
+		lang_class = 'lang-ja'
+	elif lang == 'en':
+		lang_class = 'lang-en'
+	else:
+		pass
+	span = task_statement.find('span', {'class': lang_class})
+	return str(span)
+
+
+def make_problem_markdown(html_content: str, lang: str) -> str:
+	problem_part = abstract_problem_part(html_content, lang)
+	problem_md = custom_markdownify(problem_part)
+	problem_md = remove_unnecessary_emptylines(problem_md)
+	return problem_md

atcoderstudybooster-0.1.0.dist-info/METADATA

@@ -0,0 +1,94 @@
+Metadata-Version: 2.3
+Name: AtCoderStudyBooster
+Version: 0.1.0
+Summary: A tool to download and manage AtCoder problems.
+Author-email: yuta6 <46110512+yuta6@users.noreply.github.com>
+Requires-Python: >=3.8
+Requires-Dist: beautifulsoup4
+Requires-Dist: colorama
+Requires-Dist: fire
+Requires-Dist: markdownify>=0.13.1
+Requires-Dist: requests
+Requires-Dist: tiktoken
+Requires-Dist: types-beautifulsoup4>=4.12.0.20240511
+Requires-Dist: types-colorama>=0.4.15.20240311
+Requires-Dist: types-requests>=2.32.0.20240712
+Requires-Dist: yfinance
+Description-Content-Type: text/markdown
+
+# AtCoderStudyBooster
+
+## 概要
+
+AtCoderStudyBoosterはAtCoderの学習を加速させるためのツールです。問題をローカルにダウンロードし、テスト、解答の作成をサポートするツールです。Pythonが入っていることが必須です。Pythonが入っている環境なら、`pip install AtCoderStudyBooster`でインストールできます。
+
+このツールは以下のプロジェクトに強く影響を受けています。
+[online-judge-tools](https://github.com/online-judge-tools)
+[atcoder-cli](https://github.com/Tatamo/atcoder-cli)
+これらとの違いですが、本ツールはAtCoderでのコンテストでの利用は想定しておらず、初心者の学習のサポートのみを意識しています。そのため、現時点で提出機能は備えていません。また, Chat GPT APIによる解答の作成サポート機能を備えています。
+
+## 利用ケース
+
+### B問題の練習したい場合
+
+ABCコンテストの223から226のB問題だけを集中的に練習したい場合、次のコマンドを実行します。
+
+```sh
+atcdr download B 223..226
+```
+
+コマンドを実行すると,次のようなフォルダーを作成して、各々のフォルダーに問題をダウンロードします。
+
+```css
+B
+├── 223
+│   ├── StringShifting.html
+│   └── StringShifting.md
+├── 224
+│   ├── Mongeness.html
+│   └── Mongeness.md
+├── 225
+│   ├── StarorNot.html
+│   └── StarorNot.md
+└── 226
+    ├── CountingArrays.html
+    └── CountingArrays.md
+```
+
+MarkdownファイルあるいはHTMLファイルをVS CodeのHTML Preview, Markdown Previewで開くと問題を確認できます。VS Codeで開くと左側にテキストエディターを表示して、右側で問題をみながら問題に取り組めます。
+
+![demo画像](./.images/demo1.png)
+
+### サンプルをローカルでテストする
+
+問題をダウンロードしたフォルダーに移動します。
+
+```sh
+cd B/224
+```
+
+移動したフォルダーで解答ファイルを作成後を実行すると, 自動的にテストします。
+
+```sh
+▷ ~/.../B/224
+atcdr t
+```
+
+```sh
+solution.pyをテストします。
+--------------------
+
+Sample 1 of Test:
+✓ Accepted !! Time: 24 ms
+
+Sample 2 of Test:
+✓ Accepted !! Time: 15 ms
+```
+
+と実行すると作成したソースコードをテストして、HTMLに書かれているテストケースを読み込んで実行し, Passするかを判定します。
+
+## 解答生成機能generateコマンドに関する注意点
+
+本ツールにはChatGPT APIを利用したコード生成機能があります。[AtCoder生成AI対策ルール](https://info.atcoder.jp/entry/llm-abc-rules-ja?_gl=1*1axgs02*_ga*ODc0NDAyNjA4LjE3MTk1ODEyNDA.*_ga_RC512FD18N*MTcyMzMxNDA1Ni43NC4xLjE3MjMzMTY1NjUuMC4wLjA.)によるとAtCoder Beginner Contestにおいてに問題文を生成AIに直接与えることは禁止されています。ただし、このルールは過去問を練習している際には適用されません。
+
+現時点で本ツールにはログイン機能がないため、コンテスト中の問題に対して`download`コマンドは利用して問題をダウンロードすることはできません。`generate`コマンドは`download`コマンドに依存しており、ダウンロードした問題のHTMLファイルをパースしてGPTに解釈しやすいmarkdownを与えることで実現しています。したがって、このコマンドがAtCoder Beginner Contest中に[AtCoder生成AI対策ルール](https://info.atcoder.jp/entry/llm-abc-rules-ja?_gl=1*1axgs02*_ga*ODc0NDAyNjA4LjE3MTk1ODEyNDA.*_ga_RC512FD18N*MTcyMzMxNDA1Ni43NC4xLjE3MjMzMTY1NjUuMC4wLjA.)に抵触することはありません。

atcoderstudybooster-0.1.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+atcdr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atcdr/download.py,sha256=UYOkKoFaN7Rj2p-13MOjvgq26QzdvoNmWKBaUW6cCKs,7995
+atcdr/generate.py,sha256=Bh0QHRUVeI8u5FvXbJssbs6gr55XtUkNT2p897FlUgs,5521
+atcdr/main.py,sha256=i-TFxFk7bFMtKZxtDgI7aPoZAF-dsXqNoz3O_ZsGvb4,605
+atcdr/open.py,sha256=vbOy3fthklhZ7_WIWNGyS2H3iK2FHLeClDt_tloJ_b0,924
+atcdr/test.py,sha256=it3QjFxdlR0GY6Hc0c2Qdke71Z4dj5eLhfuWVZU9cZA,7969
+atcdr/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atcdr/util/cost.py,sha256=0c9H8zLley7xZDLuYU4zJmB8m71qcO1WEIQOoEavD_4,3168
+atcdr/util/filename.py,sha256=taCgSwIpB5iCjWZrYWAGRncRyUUl9exoNfsP-KLF2bs,2984
+atcdr/util/gpt.py,sha256=Lto6SJHZGer8cC_Nq8lJVnaET2R7apFQteo6ZEFpjdM,3304
+atcdr/util/problem.py,sha256=iDfNGfoCk_sy9RQRZ4vVqd1ViyT8HSWe_ekKUb4PdKs,2412
+atcoderstudybooster-0.1.0.dist-info/METADATA,sha256=VO3J_117RJuCbT7euQAOyb375fMS65cfJMI_lORtquE,4397
+atcoderstudybooster-0.1.0.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+atcoderstudybooster-0.1.0.dist-info/entry_points.txt,sha256=_bhz0R7vp2VubKl_eIokDO8Wz9TdqvYA7Q59uWfy6Sk,42
+atcoderstudybooster-0.1.0.dist-info/RECORD,,

atcoderstudybooster-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.25.0
+Root-Is-Purelib: true
+Tag: py3-none-any

atcoderstudybooster-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+atcdr = atcdr.main:main