#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
查询artemis_II_十天绕月的数据
"""

import sqlite3


def search_artemis_data():
    """
    搜索artemis相关数据
    """
    conn = sqlite3.connect('srtsegjoin/dpd.db')
    cur = conn.cursor()
    
    # 查询ads表
    cur.execute("SELECT oid, name, author, signature FROM ads WHERE name LIKE '%artemis%' OR name LIKE '%十天绕月%'")
    ads_rows = cur.fetchall()
    print(f"找到 {len(ads_rows)} 条ads数据:")
    for row in ads_rows:
        print(f"  oid={row[0]}, name={row[1]}, author={row[2]}, signature={row[3]}")
    
    # 查询srts表
    if ads_rows:
        adid = ads_rows[0][0]
        print(f"\n查询 adid={adid} 的srts数据:")
        cur.execute("SELECT idx, cn, en, start, end FROM srts WHERE adid=? ORDER BY idx", (adid,))
        srts_rows = cur.fetchall()
        print(f"找到 {len(srts_rows)} 条srts数据")
        print(f"前5条:")
        for row in srts_rows[:5]:
            print(f"  idx={row[0]}, cn={row[1][:30]}, en={row[2][:30]}")
    
    conn.close()


if __name__ == "__main__":
    search_artemis_data()