#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
检查ads表的结构
"""

import sqlite3


def check_ads_table():
    """
    检查ads表的结构
    """
    conn = sqlite3.connect('srtsegjoin/dpd.db')
    cur = conn.cursor()
    
    # 查询ads表的结构
    cur.execute("PRAGMA table_info(ads)")
    columns = cur.fetchall()
    print(f"ads表字段:")
    for col in columns:
        print(f"  {col[1]} ({col[2]}), primary key: {col[5]}")
    
    # 查询一些有srts数据的ads数据
    cur.execute("SELECT oid, name FROM ads WHERE oid IS NOT NULL LIMIT 10")
    ads_rows = cur.fetchall()
    print(f"\n有oid的ads数据示例（前10条）:")
    for ad in ads_rows:
        print(f"  oid={ad[0]}, name={ad[1][:50]}...")
        
        # 查询srts数量
        cur.execute("SELECT COUNT(*) FROM srts WHERE adid=?", (ad[0],))
        srts_count = cur.fetchone()[0]
        print(f"    srts数量: {srts_count}")
    
    conn.close()


if __name__ == "__main__":
    check_ads_table()