#!/usr/bin/env python3
"""清理河津相关旧数据后重爬"""
import sqlite3, subprocess, sys

# 清理
conn = sqlite3.connect("/root/search.db")
sites = ["河津市环境保护通知公示", "河津开发区公示公告"]
for site in sites:
    c = conn.execute("DELETE FROM gov_raw WHERE site_name=?", (site,))
    rc = c.rowcount
    conn.execute("DELETE FROM gov_search WHERE site_name=?", (site,))
    print(f"清理 {site}: {rc}条")
conn.commit()
conn.close()

# 重爬
for script in ["crawl_sxhj", "crawl_sxhj_hjkfq"]:
    print(f"\n--- 重爬 {script} 3页 ---")
    r = subprocess.run(
        ["python3", "-c", f"import {script}; {script}.MAX_PAGES=3; {script}.crawl()"],
        capture_output=True, text=True, timeout=120
    )
    print(r.stdout)
    if r.stderr:
        print("STDERR:", r.stderr[:500])
