#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""修复 crawl_liyang_xxgk.py 的 --pages 参数 bug"""
import shutil

path = "/root/gov_crawler/crawl_liyang_xxgk.py"
shutil.copy(path, path + ".bak_0820")

src = open(path, encoding="utf-8").read()
old = """            if page_no < maxp:
                page_no += 1
                time.sleep(random.uniform(0.8, 1.5))
                continue"""
new = """            if page_no < maxp and (args.pages == 0 or page_no < args.pages):
                page_no += 1
                time.sleep(random.uniform(0.8, 1.5))
                continue"""
assert old in src, "未找到分页代码"
src = src.replace(old, new)
open(path, "w", encoding="utf-8").write(src)
print("✅ --pages bug 已修复")