31 lines
861 B
Python
31 lines
861 B
Python
|
#!/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()
|