#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
检查数据库中的idx是否连续
"""

import sqlite3


def check_idx_continuity():
    """
    检查数据库中的idx是否连续
    """
    conn = sqlite3.connect('../srtsegjoin/dpd.db')
    cur = conn.cursor()
    
    cur.execute('SELECT idx FROM srts WHERE adid=8397 ORDER BY idx')
    rows = cur.fetchall()
    
    print(f"Total rows: {len(rows)}")
    
    # 检查idx是否连续
    idx_list = [row[0] for row in rows]
    missing_idx = []
    
    for i in range(max(idx_list) + 1):
        if i not in idx_list:
            missing_idx.append(i)
    
    if missing_idx:
        print(f"Missing idx: {missing_idx}")
    else:
        print("All idx are continuous")
    
    conn.close()


if __name__ == "__main__":
    check_idx_continuity()