import sqlite3
site = '北京中环长青环境科技有限公司'
conn = sqlite3.connect('/root/search.db')
cur = conn.cursor()

cur.execute('SELECT COUNT(*) FROM gov_raw WHERE site_name=?', (site,))
print('Raw:', cur.fetchone()[0])

cur.execute('SELECT COUNT(*) FROM gov_search WHERE site_name=?', (site,))
print('Search:', cur.fetchone()[0])

cur.execute(
    'INSERT OR IGNORE INTO gov_search(rowid, title, site_name, summary) '
    'SELECT gr.id, gr.title, gr.site_name, gr.summary '
    'FROM gov_raw gr LEFT JOIN gov_search gs ON gs.rowid=gr.id '
    'WHERE gs.rowid IS NULL AND gr.site_name=?', (site,))
print('Synced search:', cur.rowcount)

cur.execute(
    'INSERT OR IGNORE INTO gov_search_v3(rowid, title, content, source_url, publish_date, site_name) '
    'SELECT gr.id, gr.title, gr.content, gr.page_url, gr.publish_date, gr.site_name '
    'FROM gov_raw gr LEFT JOIN gov_search_v3 gs ON gs.rowid=gr.id '
    'WHERE gs.rowid IS NULL AND gr.site_name=?', (site,))
print('Synced v3:', cur.rowcount)

conn.commit()

cur.execute('SELECT COUNT(*) FROM gov_search WHERE site_name=?', (site,))
print('Search after:', cur.fetchone()[0])
conn.close()
