#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
查询包含绕月的srts数据
"""

import sqlite3


def search_moon_srts():
    """
    搜索包含绕月的srts数据
    """
    conn = sqlite3.connect('srtsegjoin/dpd.db')
    cur = conn.cursor()
    
    # 查询adid=8397的数据（确认之前的数据）
    cur.execute("SELECT adid, adname, COUNT(*) FROM srts WHERE adid=8397")
    row = cur.fetchone()
    if row:
        print(f"adid=8397:")
        print(f"  adname={row[1]}")
        print(f"  srts数量={row[2]}")
    
    # 查询包含"绕月"的srts数据
    cur.execute("SELECT DISTINCT adid, adname FROM srts WHERE cn LIKE '%绕月%' OR en LIKE '%moon%' OR en LIKE '%lunar%'")
    rows = cur.fetchall()
    print(f"\n找到 {len(rows)} 条包含绕月/月球相关的srts数据:")
    for row in rows:
        print(f"\n  adid={row[0]}")
        print(f"  adname={row[1]}")
        
        # 查询srts数量
        cur.execute("SELECT COUNT(*) FROM srts WHERE adid=?", (row[0],))
        count = cur.fetchone()[0]
        print(f"  srts数量={count}")
        
        # 查询前几条
        cur.execute("SELECT idx, cn, en FROM srts WHERE adid=? ORDER BY idx LIMIT 3", (row[0],))
        snippets = cur.fetchall()
        print(f"  前3条:")
        for s in snippets:
            print(f"    idx={s[0]}: cn={s[1][:40]}... en={s[2][:40]}...")
    
    conn.close()


if __name__ == "__main__":
    search_moon_srts()