#!/usr/bin/env python3
"""
稷山县-环境保护 爬虫
站点: www.jishan.gov.cn/zfxxgk/fdzdgknr/sthj/
列表: 4页, 每页20条
正文: <div id="Zoom">
"""
import sys, os, re, time, json, sqlite3, urllib.request, urllib.parse
from datetime import datetime, timedelta
import os

SEARCH_DB = os.getenv("SEARCH_DB", "/root/search.db")
BASE_URL = "http://www.jishan.gov.cn/zfxxgk/fdzdgknr/sthj"
SITE_NAME = "稷山环境保护"
THREE_YEARS_AGO = datetime.now() - timedelta(days=3*365)
THREE_YEAR_LIMIT = True

HEADERS = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"}

def save_to_db(records):
    db = sqlite3.connect(SEARCH_DB)
    try:
        for r in records:
            db.execute("INSERT OR IGNORE INTO gov_raw(title, content, publish_date, source_url, page_url, site_name) VALUES (?,?,?,?,?,?)",
                       (r['title'], r['content'], r['publish_date'], r['url'], r['url'], SITE_NAME))
        db.commit()
    finally:
        db.close()

def fetch_list_page(page):
    """获取列表页"""
    if page == 1:
        url = f"{BASE_URL}/index.shtml"
    else:
        url = f"{BASE_URL}/index_{page}.shtml"
    req = urllib.request.Request(url, headers=HEADERS)
    try:
        resp = urllib.request.urlopen(req, timeout=30)
        html = resp.read().decode('utf-8', errors='replace')
        return html
    except Exception as e:
        print(f"  ⚠ 第{page}页失败: {e}")
        return ""

def parse_list(html):
    """解析列表页"""
    items = []
    # 找 box_list 区域
    m = re.search(r'class="box_list mgb40"[^>]*>(.*?)<div class="list_page', html, re.DOTALL)
    if not m:
        return items
    content = m.group(1)
    # 提取每一条
    lis = re.findall(r'<li[^>]*>(.*?)</li>', content, re.DOTALL)
    for li in lis:
        a = re.search(r'href=["\']([^"\']+)["\'][^>]*title=["\']([^"\']*)["\']', li)
        if not a:
            continue
        url = a.group(1)
        title = a.group(2).strip()
        date_span = re.search(r'<span[^>]*>([^<]+)</span>', li)
        date_str = date_span.group(1).strip() if date_span else ""
        items.append((url, title, date_str))
    return items

def fetch_detail(url):
    """获取详情页正文 - <div id='Zoom'> 区域"""
    full_url = url if url.startswith('http') else f"http://www.jishan.gov.cn{url}"
    req = urllib.request.Request(full_url, headers=HEADERS)
    try:
        resp = urllib.request.urlopen(req, timeout=30)
        html = resp.read().decode('utf-8', errors='replace')
        # 提取 Zoom 区域内容
        content = ""
        m = re.search(r'<div id="Zoom"[^>]*>(.*?)</div>', html, re.DOTALL)
        if m:
            content = m.group(1).strip()
        if not content:
            m = re.search(r'<div id=["\']zoom["\'][^>]*>(.*?)</div>', html, re.DOTALL)
            if m:
                content = m.group(1).strip()
        return content
    except Exception as e:
        print(f"  ⚠ 详情失败: {url}: {e}")
        return ""

def main():
    print(f"{'='*50}")
    print(f"{SITE_NAME}")
    print(f"{'='*50}")
    
    all_articles = []
    all_urls = set()
    
    for page in range(1, 5):  # 4页
        html = fetch_list_page(page)
        if not html:
            continue
        items = parse_list(html)
        print(f"  📄 第{page}页: {len(items)}条", end="")
        for url, title, date_str in items:
            if url in all_urls:
                continue
            all_urls.add(url)
            # 日期过滤
            if THREE_YEAR_LIMIT and date_str:
                try:
                    pub_date = datetime.strptime(date_str, '%Y-%m-%d')
                    if pub_date < THREE_YEARS_AGO:
                        continue
                except:
                    pass
            all_articles.append({'url': url, 'title': title, 'publish_date': date_str})
        print(f" → 累计{len(all_articles)}条")
    
    print(f"\n📊 列表总计: {len(all_articles)} 条")
    print(f"\n{'='*50}")
    print("抓取详情页...")
    
    batch = []
    total_saved = 0
    total_skipped = 0
    
    for i, art in enumerate(all_articles):
        title = art['title']
        print(f"  [{i+1}/{len(all_articles)}] {title[:50]}...", end=" ")
        
        content = fetch_detail(art['url'])
        if content:
            batch.append({
                'title': title,
                'content': content,
                'publish_date': art['publish_date'],
                'url': f"http://www.jishan.gov.cn{art['url']}"
            })
            print(f"✅ ({len(content)}字)")
            
            if len(batch) >= 10:
                save_to_db(batch)
                total_saved += len(batch)
                batch = []
        else:
            print("⚠ 无正文")
    
    if batch:
        save_to_db(batch)
        total_saved += len(batch)
    
    print(f"\n{'='*50}")
    print(f"✅ 完成! 入库: {total_saved}条")
    
    if total_saved > 0:
        db = sqlite3.connect(SEARCH_DB)
        try:
            db.execute("INSERT INTO gov_search(rowid, title, site_name, summary) SELECT rowid, title, site_name, substr(content,1,200) FROM gov_raw WHERE site_name=? AND rowid NOT IN (SELECT rowid FROM gov_search WHERE site_name=?)", (SITE_NAME, SITE_NAME))
            db.commit()
            print("   FTS已更新")
        except Exception as e:
            print(f"   FTS更新: {e}")
        finally:
            db.close()

if __name__ == '__main__':
    main()