From 5bed29566c2009b6246047bc30fce404f3fb51a8 Mon Sep 17 00:00:00 2001
From: gene-2012 <gene201248@126.com>
Date: Thu, 1 May 2025 14:33:06 +0800
Subject: [PATCH] first edition

---
 .env        |   4 ++
 .gitignore  |   4 ++
 contests.py |  42 +++++++++++++
 luogu.ico   | Bin 0 -> 4286 bytes
 main.py     | 114 +++++++++++++++++++++++++++++++++++
 main.spec   |  39 ++++++++++++
 models.py   | 168 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 7 files changed, 371 insertions(+)
 create mode 100644 .env
 create mode 100644 .gitignore
 create mode 100644 contests.py
 create mode 100644 luogu.ico
 create mode 100644 main.py
 create mode 100644 main.spec
 create mode 100644 models.py

diff --git a/.env b/.env
new file mode 100644
index 0000000..d59f8eb
--- /dev/null
+++ b/.env
@@ -0,0 +1,4 @@
+CLIENT_ID=
+UID=
+CF_CLEARANCE=
+C3VK=
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..81bb05f
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,4 @@
+__pycache__/
+build/
+dist/
+env/
\ No newline at end of file
diff --git a/contests.py b/contests.py
new file mode 100644
index 0000000..83cf44b
--- /dev/null
+++ b/contests.py
@@ -0,0 +1,42 @@
+class Contest:
+    def __init__(self, contest_id, unrated=False):
+        self.contest_id = contest_id
+        self.unrated = unrated
+        self.contest_url = f"https://luogu.com.cn/contest/{contest_id}"
+        self.join_url = f"https://www.luogu.com.cn/fe/api/contest/join/{contest_id}"
+
+        # 动态设置 Referer
+        self.headers = BASE_HEADERS.copy()
+        self.headers['Referer'] = self.contest_url
+
+        self.csrf_token = self._get_csrf_token()
+        self._join_contest()
+
+    def _get_csrf_token(self):
+        resp = requests.get(
+            self.contest_url,
+            headers=self.headers,
+            cookies=cookies
+        )
+        soup = BeautifulSoup(resp.text, 'html.parser')
+        meta_tag = soup.find('meta', {'name': 'csrf-token'})
+        if meta_tag:
+            return meta_tag.get('content')
+        else:
+            raise ValueError("无法找到 csrf-token")
+
+    def _join_contest(self):
+        headers = self.headers.copy()
+        headers['X-CSRF-Token'] = self.csrf_token
+
+        data = {"unrated": str(self.unrated).lower()}
+
+        resp = requests.post(
+            self.join_url,
+            headers=headers,
+            cookies=cookies,
+            json=data
+        )
+
+        print(f"Status Code: {resp.status_code}")
+        print(f"Response Text: {resp.text}")
\ No newline at end of file
diff --git a/luogu.ico b/luogu.ico
new file mode 100644
index 0000000000000000000000000000000000000000..80c5505a63d4f19d2dd54bdd620926fb086d4e7b
GIT binary patch
literal 4286
zcmc&&Sx-|@5WYr#fWGJ}kNynbjQFk~Mq|V^`eb5!&`8v{8)M>vC`8tP(EzfDfGklF
z&_L9-wzjmTR%lsUXo0TdeCPDuVxXZ5#UVq>x#!F`XJ*bf=b|W^_$Mbv!EdE@tD<aI
z6lEtX4=KA?F)oCkZQCJL4k!u{LAS8xvNo|!v1UtV*l-K$URGcNdoybx>m=(&39yc_
z0vr2wvu?CZFd*XaCJ}IOev)QQy)<L!pjl%#1?SgJO?b&i-jUBVqwf&w7N=~9%wxQn
z6@IKL?k{zuE6FE)$pO}Ua&|qUNN8!*xfsN=u@9xBe|}J`(LLQyuKwzT^Z5Zb6iqb2
z=lns&iW3t5SpgYJ5A&Wn6E5+XKa!#BFzKI3f6aC0XklU`k@>a9qfwe1Zj}3Be8j<6
zd6K-={uOJNrX6H?bza0toi{%^v|?VW1L1(5{`3@UF+d)E)tn{7MwQ^)1lgLdNNo7Y
z@t^#Lqnt3b>e8e>gT869x88``XCKTjIks*yg_q{Z(Q#jUhI+9zU!(BiTvE1^ofPo8
z$lh{Yivw^*>MzpR$1={hT5ow|sya=+Ke}YirZf-d?|0_|Szm}V>Wr}srH8mqkCV0G
z63yAaa@f}=?yET$TJVsw`!VAJU)H3l`V4t2p9r<G5rUjMd!EF7U_VeF0dIC`6x`->
zeixkRvGkGS-QBnkTo0pfZzaW%>BI<U0J)d^Qb0dn6w<Wrova7&V=Nb=ZFks*^=)OS
zyV17WTJ5X3_n1FW$UiIgEjT*ngT3`8ffLq4P;Vm69AAlhfOJpu$j)aR*btme)PCHD
zd~&qk%bWxF9JvSU;&yp|h&yU8;2GDpCpfaVgy*R~N)Gf|2ediqd<iXh<D4Fg?Zi+$
zMI&Kpj{O4<+nTNl4p8%=@;~@76N&J7wl-dtJvfFx-dE0jHXzK-{CqF^k?7HQ9-AZh
zf4G5pD3S@hSUat6kM%!U<KQ|Ge?IHrV_oq+nIH6gR%7`wat+i_B)FI{HVj093xqub
zSMn`5kIy=vvsqJ*xDVh?;4yy^_a48?#M=(73$|(m_Zs(L?TQ>byB>1?^QqIa`l;Qm
zRIZ8T8-8W8-z?u0c*ypq=%0EtRh{BHmSy?0tbRvtH&&j~I03ak-dB+cW{S-^DtKS3
zN5;3jzCfWxZ@NCDOw_0aSAUJ@ML2)ph&1aEjjAyn$H76kbD;kG<HmGMU=7}V_ONF6
z;sB(T0?b@iyaOI#-AEr0_oFdv^&NOO|BYPWS&B8uE7lg+gl#qE;{Ce_{~q{H0_?v)
C75Rq%

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..37fbe21
--- /dev/null
+++ b/main.py
@@ -0,0 +1,114 @@
+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]")
\ No newline at end of file
diff --git a/main.spec b/main.spec
new file mode 100644
index 0000000..e505279
--- /dev/null
+++ b/main.spec
@@ -0,0 +1,39 @@
+# -*- mode: python ; coding: utf-8 -*-
+
+
+a = Analysis(
+    ['main.py'],
+    pathex=[],
+    binaries=[],
+    datas=[],
+    hiddenimports=[],
+    hookspath=[],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=[],
+    noarchive=False,
+    optimize=0,
+)
+pyz = PYZ(a.pure)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    a.binaries,
+    a.datas,
+    [],
+    name='main',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    runtime_tmpdir=None,
+    console=True,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+    icon=['luogu.ico'],
+)
diff --git a/models.py b/models.py
new file mode 100644
index 0000000..c5fd67b
--- /dev/null
+++ b/models.py
@@ -0,0 +1,168 @@
+import os
+import json
+import requests
+from bs4 import BeautifulSoup
+from dotenv import load_dotenv
+from requests.exceptions import RequestException
+
+# 加载 .env 文件
+load_dotenv()
+
+# 请求头与 Cookie
+headers = {
+    'User-Agent': (
+        'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) '
+        'Gecko/20100101 Firefox/115.0'
+    )
+}
+
+cookies = {
+    "__client_id": os.getenv("CLIENT_ID"),
+    "_uid": os.getenv("UID"),
+    "cf_clearance": os.getenv("CF_CLEARANCE"),
+    "C3VK": os.getenv("C3VK")
+}
+
+BASE_HEADERS = {
+    'User-Agent': (
+        'Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) '
+        'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Mobile Safari/537.36'
+    ),
+    'Origin': 'https://www.luogu.com.cn',
+    'Content-Type': 'application/json',
+    'X-Requested-With': 'XMLHttpRequest'
+}
+
+
+class Problem:
+    def __init__(self, problem_id):
+        self.problem_id = problem_id
+        self.problem_url = f"https://luogu.com.cn/problem/{problem_id}"
+        self.html = None
+        self.json = None
+        self.title = ""
+        self.background = ""
+        self.description = ""
+        self.formatI = ""
+        self.formatO = ""
+        self.hint = ""
+        self.samples = []
+
+        try:
+            resp = requests.get(self.problem_url, headers=headers, cookies=cookies, timeout=10)
+            resp.raise_for_status()
+            self.html = resp.text
+        except RequestException as e:
+            raise ConnectionError(f"无法访问题目页面：{self.problem_url}，错误：{e}")
+
+        try:
+            self.json = self._extract_json_from_html(self.html)['data']['problem']
+        except (KeyError, ValueError) as e:
+            raise RuntimeError(f"解析题目数据失败：{e}")
+
+        self.title = self.json.get('title', '')
+        self.background = self.json['contenu'].get('background', '')
+        self.description = self.json['contenu'].get('description', '')
+        self.formatI = self.json['contenu'].get('formatI', '')
+        self.formatO = self.json['contenu'].get('formatO', '')
+        self.hint = self.json['contenu'].get('hint', '')
+        self.samples = self.json.get('samples', [])
+
+    def _extract_json_from_html(self, html_content):
+        try:
+            soup = BeautifulSoup(html_content, 'html.parser')
+            script_tag = soup.find('script', {'id': 'lentille-context', 'type': 'application/json'})
+            if not script_tag:
+                raise ValueError("无法找到指定的JSON脚本标签")
+            return json.loads(script_tag.string)
+        except json.JSONDecodeError as e:
+            raise ValueError(f"JSON 解析失败：{e}")
+        except Exception as e:
+            raise RuntimeError(f"HTML 解析失败：{e}")
+
+    def gen_problem_md(self):
+        try:
+            md = f"# {self.title}"
+            if self.background:
+                md += f"\n\n## 【题目背景】\n{self.background}"
+            if self.description:
+                md += f"\n\n## 【题目描述】\n{self.description}"
+            if self.formatI:
+                md += f"\n\n## 【输入格式】\n{self.formatI}"
+            if self.formatO:
+                md += f"\n\n## 【输出格式】\n{self.formatO}"
+            if self.hint:
+                md += f"\n\n## 【提示】\n{self.hint}"
+
+            with open(f"{self.problem_id}.md", "w", encoding="utf8") as f:
+                f.write(md)
+        except IOError as e:
+            raise RuntimeError(f"写入题面Markdown文件失败：{e}")
+
+    def gen_samples(self):
+        try:
+            os.makedirs("samples", exist_ok=True)
+            cnt = 1
+            for sample in self.samples:
+                with open(f"samples/sample_{cnt}.in", "w", encoding="utf8") as f:
+                    f.write(sample[0])
+                with open(f"samples/sample_{cnt}.out", "w", encoding="utf8") as f:
+                    f.write(sample[1])
+                cnt += 1
+        except (IOError, IndexError) as e:
+            raise RuntimeError(f"写入样例文件失败：{e}")
+
+    def gen_solutions(self):
+        solution_url = f"https://luogu.com.cn/problem/solution/{self.problem_id}"
+        try:
+            resp = requests.get(solution_url, headers=headers, cookies=cookies, timeout=10)
+            resp.raise_for_status()
+        except RequestException as e:
+            raise ConnectionError(f"无法访问题解页面：{solution_url}，错误：{e}")
+
+        try:
+            solutions_data = self._extract_json_from_html(resp.text).get("data", {})
+            solutions = solutions_data.get("solutions", {}).get("result", [])
+        except (KeyError, ValueError) as e:
+            raise RuntimeError(f"解析题解数据失败：{e}")
+
+        try:
+            os.makedirs("solutions", exist_ok=True)
+            for solution in solutions:
+                author = solution.get('author', {})
+                title = solution.get('title', '未命名题解')
+                lid = solution.get('lid', 'unknown')
+                content = solution.get('content', '')
+
+                markdown_content = (
+                    f"# {title}\n"
+                    f"> By ![{author.get('name', '')}]({author.get('avatar', '')}?"
+                    "x-oss-process=image/resize,m_fixed,h_30,w_30,image/circle,r_100/format,png)"
+                    f"[{author.get('name', '')}](https://luogu.com.cn/user/{author.get('uid', '')})"
+                    "\n\n"
+                    f"{content}"
+                )
+
+                filename = f"solutions/{title}_{lid}.md"
+                with open(filename, "w", encoding="utf8") as f:
+                    f.write(markdown_content)
+        except IOError as e:
+            raise RuntimeError(f"写入题解文件失败：{e}")
+
+    def gen_all(self):
+        try:
+            os.makedirs(self.problem_id, exist_ok=True)
+            os.chdir(self.problem_id)
+            self.gen_problem_md()
+            self.gen_samples()
+            self.gen_solutions()
+        except Exception as e:
+            raise RuntimeError(f"生成全部内容失败：{e}")
+
+
+if __name__ == "__main__":
+    try:
+        p = Problem("P1145")
+        p.gen_all()
+    except Exception as e:
+        print(f"[ERROR] 发生错误：{e}")
\ No newline at end of file