from database import TradeDB
db = TradeDB()
try:
    val = db.get_env("LS_WS_MAX_SUBSCRIPTIONS")
    print("현재 값(get_env):", val)
    
    # set_env 함수가 있는지 확인 및 호출
    if hasattr(db, 'set_env'):
        db.set_env("LS_WS_MAX_SUBSCRIPTIONS", "0")  # 0: 무제한 로컬 한도 없음
        print("set_env 로 '0' (무제한) 세팅 완료!")
        
    val_after = db.get_env("LS_WS_MAX_SUBSCRIPTIONS")
    print("변경 후 값(get_env):", val_after)
except Exception as e:
    print("Error:", e)
finally:
    db.close()