from database import TradeDB
db = TradeDB()
try:
    cols = [dict(r)['Field'] for r in db.conn.execute("SHOW COLUMNS FROM env_config").fetchall()]
    print(f"env_config cols: {cols}")
    
    # 올바른 컬럼명으로 400 덮어쓰기
    key_col = cols[0]
    val_col = cols[1]
    
    row = db.conn.execute(f"SELECT {val_col} FROM env_config WHERE {key_col}='LS_WS_MAX_SUBSCRIPTIONS'").fetchone()
    print("현재 DB 값:", dict(row) if row else "없음")
    
    db.conn.execute(
        f"INSERT INTO env_config ({key_col}, {val_col}) VALUES ('LS_WS_MAX_SUBSCRIPTIONS', '400') "
        f"ON DUPLICATE KEY UPDATE {val_col}='400'"
    )
    db.conn.commit()
    print("DB 업데이트 완료.")
finally:
    db.close()