114 lines
3.9 KiB
Python
114 lines
3.9 KiB
Python
|
import argparse
|
|||
|
|
import os
|
|||
|
|
from rich.console import Console
|
|||
|
|
from rich.markdown import Markdown
|
|||
|
|
from models import Problem
|
|||
|
|
|
|||
|
|
console = Console()
|
|||
|
|
|
|||
|
|
def parse_args():
|
|||
|
|
try:
|
|||
|
|
parser = argparse.ArgumentParser(description="Luogu CLI 工具")
|
|||
|
|
subparsers = parser.add_subparsers(dest="command", required=True)
|
|||
|
|
|
|||
|
|
# lgcli parse PID
|
|||
|
|
parse_parser = subparsers.add_parser("parse", help="下载指定题目的题面、样例和题解")
|
|||
|
|
parse_parser.add_argument("pid", help="题目编号，例如 P1145")
|
|||
|
|
|
|||
|
|
# lgcli view PID
|
|||
|
|
view_parser = subparsers.add_parser("view", help="查看题目描述")
|
|||
|
|
view_parser.add_argument("pid", help="题目编号，例如 P1145")
|
|||
|
|
|
|||
|
|
# lgcli solution PID
|
|||
|
|
solution_parser = subparsers.add_parser("solution", help="查看题解列表或具体题解内容")
|
|||
|
|
solution_parser.add_argument("pid", help="题目编号，例如 P1145")
|
|||
|
|
solution_parser.add_argument("--id", help="指定题解文件名（不带.md）", default=None)
|
|||
|
|
|
|||
|
|
return parser.parse_args()
|
|||
|
|
except argparse.ArgumentError as e:
|
|||
|
|
console.print(f"[red]参数解析错误：{e}[/red]")
|
|||
|
|
exit(1)
|
|||
|
|
except Exception as e:
|
|||
|
|
console.print(f"[red]未知错误：{e}[/red]")
|
|||
|
|
exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def cmd_view(pid):
|
|||
|
|
filename = f"{pid}/{pid}.md"
|
|||
|
|
if not os.path.exists(filename):
|
|||
|
|
console.print(f"[red]错误：{pid} 尚未解析，请先运行 lgcli parse {pid}[/red]")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
console.clear()
|
|||
|
|
with open(filename, "r", encoding="utf8") as f:
|
|||
|
|
content = f.read()
|
|||
|
|
console.print(Markdown(content))
|
|||
|
|
except IOError as e:
|
|||
|
|
console.print(f"[red]读取文件失败：{e}[/red]")
|
|||
|
|
except Exception as e:
|
|||
|
|
console.print(f"[red]渲染Markdown失败：{e}[/red]")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def cmd_solution(pid):
|
|||
|
|
folder = f"{pid}/solutions"
|
|||
|
|
if not os.path.exists(folder):
|
|||
|
|
console.print(f"[red]错误：{pid} 题解尚未下载，请先运行 lgcli parse {pid}[/red]")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
files = [f for f in os.listdir(folder) if f.endswith(".md")]
|
|||
|
|
if not files:
|
|||
|
|
console.print("[yellow]暂无题解。[/yellow]")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if args.id is None:
|
|||
|
|
console.clear()
|
|||
|
|
console.print("[bold blue]可用题解：[/bold blue]")
|
|||
|
|
for i, fname in enumerate(files):
|
|||
|
|
console.print(f"{i+1}. {fname[:-3]}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
selected_file = f"{args.id}.md"
|
|||
|
|
full_path = os.path.join(folder, selected_file)
|
|||
|
|
if not os.path.exists(full_path):
|
|||
|
|
console.print(f"[red]错误：题解 {selected_file} 不存在。请检查名称。[/red]")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
console.clear()
|
|||
|
|
with open(full_path, "r", encoding="utf8") as f:
|
|||
|
|
content = f.read()
|
|||
|
|
console.print(Markdown(content))
|
|||
|
|
except IOError as e:
|
|||
|
|
console.print(f"[red]读取题解文件失败：{e}[/red]")
|
|||
|
|
except Exception as e:
|
|||
|
|
console.print(f"[red]渲染Markdown失败：{e}[/red]")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def cmd_parse(pid):
|
|||
|
|
try:
|
|||
|
|
p = Problem(pid)
|
|||
|
|
p.gen_all()
|
|||
|
|
console.print(f"[green]✅ 题目 {pid} 已成功下载到目录 '{pid}'[/green]")
|
|||
|
|
except ConnectionError as e:
|
|||
|
|
console.print(f"[red]网络连接错误：{e}[/red]")
|
|||
|
|
except RuntimeError as e:
|
|||
|
|
console.print(f"[red]数据解析错误：{e}[/red]")
|
|||
|
|
except Exception as e:
|
|||
|
|
console.print(f"[red]未知错误：{e}[/red]")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
try:
|
|||
|
|
args = parse_args()
|
|||
|
|
|
|||
|
|
if args.command == "parse":
|
|||
|
|
cmd_parse(args.pid)
|
|||
|
|
elif args.command == "view":
|
|||
|
|
cmd_view(args.pid)
|
|||
|
|
elif args.command == "solution":
|
|||
|
|
cmd_solution(args.pid)
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
console.print("\n[yellow]用户中断操作。[/yellow]")
|
|||
|
|
except Exception as e:
|
|||
|
|
console.print(f"[red]发生致命错误：{e}[/red]")
|