#!/usr/bin/env python3
import os
import psycopg2

# 从环境变量获取 DATABASE_URL
db_url = os.environ.get('DATABASE_URL')
if not db_url:
    # 尝试从 docker-compose 生成的变量拼接
    db_url = "postgresql://wecom_user:wecom_pass@10.90.5.110:5432/wecom_it_desk"

conn = psycopg2.connect(db_url)
cur = conn.cursor()

# A表
cur.execute("SELECT COUNT(*) FROM config_change_logs")
a_count = cur.fetchone()[0]

# B表 audit_logs 中 action='config_change' 的数量
cur.execute("SELECT COUNT(*) FROM audit_logs WHERE action = 'config_change'")
b_config_change_count = cur.fetchone()[0]

# B表总数
cur.execute("SELECT COUNT(*) FROM audit_logs")
b_total = cur.fetchone()[0]

print(f"config_change_logs (A): {a_count}")
print(f"audit_logs (B) - config_change events: {b_config_change_count}")
print(f"audit_logs (B) - total: {b_total}")

conn.close()