import os
import pg8000.dbapi as pg
from lawrisk.utils.env_loader import load_env

def check():
    load_env()
    conn_params = {
        "host": os.getenv("LIC_PG_HOST", "172.24.240.1"),
        "port": int(os.getenv("LIC_PG_PORT", os.getenv("PG_PORT", "5432"))),
        "user": os.getenv("LIC_PG_USER", os.getenv("PG_USER", "postgres")),
        "password": os.getenv("LIC_PG_PASSWORD", ""),
        "database": os.getenv("LIC_PG_DATABASE", "licensing_risks"),
    }
    conn = pg.connect(**conn_params)
    cur = conn.cursor()
    
    cur.execute("SELECT column_name FROM information_schema.columns WHERE table_name = 'permit_theme_rules'")
    cols = [row[0] for row in cur.fetchall()]
    print(f"Columns in 'permit_theme_rules': {cols}")
    
    cur.execute("SELECT COUNT(*) FROM permit_theme_rules")
    print(f"Row count in 'permit_theme_rules': {cur.fetchone()[0]}")
    
    conn.close()

if __name__ == "__main__":
    check()