#!/usr/bin/env python3
"""检查生产DB是否有交城县数据"""
import sqlite3

# 生产DB
conn = sqlite3.connect("/root/search.db")
cur = conn.cursor()
cur.execute("SELECT COUNT(*) FROM gov_raw WHERE url LIKE '%sx-jc.gov.cn%'")
raw_count = cur.fetchone()[0]
print(f"生产DB gov_raw中交城县数据: {raw_count}条")

cur.execute("SELECT COUNT(*) FROM gov_search WHERE site_name LIKE '%交城%'")
fts_count = cur.fetchone()[0]
print(f"生产DB gov_search FTS中交城县数据: {fts_count}条")

# 爬虫DB
conn2 = sqlite3.connect("/root/gov_crawler/search.db")
cur2 = conn2.cursor()
cur2.execute("SELECT COUNT(*) FROM gov_raw WHERE url LIKE '%sx-jc.gov.cn%'")
raw_count2 = cur2.fetchone()[0]
print(f"爬虫DB gov_raw中交城县数据: {raw_count2}条")

cur2.execute("SELECT COUNT(*) FROM gov_search WHERE site_name LIKE '%交城%'")
fts_count2 = cur2.fetchone()[0]
print(f"爬虫DB gov_search FTS中交城县数据: {fts_count2}条")

conn.close()
conn2.close()