# -*- coding: utf-8 -*-
"""执行DWS配置数据导入"""

import os
from pathlib import Path
from dotenv import load_dotenv
import psycopg2

def main():
    # 加载.env配置
    env_path = Path(__file__).parent.parent / ".env"
    load_dotenv(env_path)
    
    dsn = os.getenv("PG_DSN")
    if not dsn:
        print("错误: 未找到 PG_DSN 配置")
        return
    
    # 读取SQL文件
    sql_file = Path(__file__).parent.parent / "database" / "seed_dws_config.sql"
    sql_content = sql_file.read_text(encoding="utf-8")
    
    print(f"连接数据库...")
    conn = psycopg2.connect(dsn)
    conn.autocommit = True
    
    with conn.cursor() as cur:
        print(f"执行SQL文件: {sql_file}")
        cur.execute(sql_content)
        print("DWS配置数据导入成功!")
    
    conn.close()

if __name__ == "__main__":
    main()