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

atcdr/util/filetype.py

@@ -0,0 +1,105 @@
+import os
+from enum import Enum
+from typing import Dict, List, Optional, 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, 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,
+]
+
+# コンパイル型言語のリスト
+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()]
+
+
+def lang2str(lang: Lang) -> str:
+	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

atcdr/util/problem.py

@@ -1,46 +1,48 @@
 import re
 from enum import Enum
-from typing import Optional, Union
+from typing import Match, Optional
 
 from bs4 import BeautifulSoup as bs
-from bs4 import NavigableString, Tag
+from bs4 import Tag
 from markdownify import MarkdownConverter  # type: ignore
 
 
-# TODO : そのうちgenerate.pyやtest.py, open.pyのHTMLのparse処理を全部まとめる
-class Lang(Enum):
-	JA = 'ja'
-	EN = 'en'
+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"')
+	html = remove_unnecessary_emptylines(html)
+	return html
 
 
-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 get_title_from_html(html: str) -> str:
+	title: Optional[Match[str]] = re.search(r'<title>([^<]*)</title>', html)
+	return title.group(1).strip() if title else ''
 
-	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'})
+def title_to_filename(title: str) -> str:
+	title = re.sub(r'[\\/*?:"<>| !@#$%^&()+=\[\]{};,\']', '', title)
+	title = re.sub(r'^[A-Z]-', '', title)
+	return title
 
-		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'})
+def find_link_from_html(html: str) -> str:
+	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 ''
 
-			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 Lang(Enum):
+	JA = 'ja'
+	EN = 'en'
 
 
 class CustomMarkdownConverter(MarkdownConverter):
@@ -57,10 +59,10 @@ 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 remove_unnecessary_emptylines(text):
+	text = re.sub(r'\n\s*\n\s*\n+', '\n\n', text)
+	text = text.strip()
+	return text
 
 
 def abstract_problem_part(html_content: str, lang: str) -> str:
@@ -81,7 +83,9 @@ def abstract_problem_part(html_content: str, lang: str) -> str:
 
 
 def make_problem_markdown(html_content: str, lang: str) -> str:
+	title = get_title_from_html(html_content)
 	problem_part = abstract_problem_part(html_content, lang)
 	problem_md = custom_markdownify(problem_part)
+	problem_md = f'# {title}\n{problem_md}'
 	problem_md = remove_unnecessary_emptylines(problem_md)
 	return problem_md

{atcoderstudybooster-0.1.1.dist-info → atcoderstudybooster-0.3.dist-info}/METADATA

@@ -1,19 +1,19 @@
 Metadata-Version: 2.3
 Name: AtCoderStudyBooster
-Version: 0.1.1
+Version: 0.3
 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: colorama
 Requires-Dist: fire
 Requires-Dist: markdownify>=0.13.1
+Requires-Dist: questionary>=2.0.1
 Requires-Dist: requests
+Requires-Dist: rich>=13.7.1
 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-0.3.dist-info/RECORD

@@ -0,0 +1,17 @@
+atcdr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atcdr/download.py,sha256=aqJEmrLop_Mj8GNJjoWRcSzWU0z9iNc9vtZOjhWXx3U,8945
+atcdr/generate.py,sha256=0yWX-5PS-FR6LTaP3muHq6a7rFB2a1Oek48mF45exoA,6972
+atcdr/main.py,sha256=y2IkXwcAyKZ_1y5PgU93GpXzo5lKak9oxo0XV_9d5Fo,727
+atcdr/markdown.py,sha256=jEktnYgrDYcgIuhxRpJImAzNpFmfSPkRikAesfMxAVk,1125
+atcdr/open.py,sha256=2UlmNWdieoMrPu1xSUWf-8sBB9Y19r0t6V9zDRBSPes,924
+atcdr/test.py,sha256=eNHEv_y2tNBW9BbfzbGxRjUjbZB7Skxp1UVZkUJyj24,12021
+atcdr/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+atcdr/util/cost.py,sha256=0c9H8zLley7xZDLuYU4zJmB8m71qcO1WEIQOoEavD_4,3168
+atcdr/util/execute.py,sha256=tcYflnVo_38LdaOGDUAuqfSfcA54bTrCaTRShH7kwUw,1750
+atcdr/util/filetype.py,sha256=n8alyOdrItoFZzLIssDoXg7i5LamHM9DaY1legUzOD0,2007
+atcdr/util/gpt.py,sha256=Lto6SJHZGer8cC_Nq8lJVnaET2R7apFQteo6ZEFpjdM,3304
+atcdr/util/problem.py,sha256=WprmpOZm6xpyvksIS3ou1uHqFnBO1FUZWadsLziG1bY,2484
+atcoderstudybooster-0.3.dist-info/METADATA,sha256=xnKGPQ0_6d592dSluNqZ17xodV23Rvi1h7CGFjsIv84,4467
+atcoderstudybooster-0.3.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+atcoderstudybooster-0.3.dist-info/entry_points.txt,sha256=_bhz0R7vp2VubKl_eIokDO8Wz9TdqvYA7Q59uWfy6Sk,42
+atcoderstudybooster-0.3.dist-info/RECORD,,

atcdr/util/filename.py

@@ -1,137 +0,0 @@
-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))

atcoderstudybooster-0.1.1.dist-info/RECORD

@@ -1,15 +0,0 @@
-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.1.dist-info/METADATA,sha256=6ncUVzTFelY1WHUoNP4NNLIqsjHiDAh_54Lgsg3OruI,4478
-atcoderstudybooster-0.1.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-atcoderstudybooster-0.1.1.dist-info/entry_points.txt,sha256=_bhz0R7vp2VubKl_eIokDO8Wz9TdqvYA7Q59uWfy6Sk,42
-atcoderstudybooster-0.1.1.dist-info/RECORD,,