initial commit

한국투자증권(API)/MCP/Kis Trading MCP/module/plugin/environment.py

@@ -0,0 +1,43 @@
+import logging
+import os
+from collections import namedtuple
+
+from dotenv import load_dotenv
+
+# Environment 설정을 위한 namedtuple 정의
+EnvironmentConfig = namedtuple('EnvironmentConfig', [
+    'mcp_type', 'mcp_host', 'mcp_port', 'mcp_path'
+])
+
+
+def setup_environment(env: str) -> EnvironmentConfig:
+    # get api env
+    if not env:
+        logging.error("Environment variable ENV not defined")
+        exit(1)
+
+    # load .env
+    dotenv_path = os.path.join(os.getcwd(), f".env.{env}")
+    if not os.path.isfile(dotenv_path):
+        logging.error(f"Environment variable file .env.{env} not found")
+        exit(1)
+
+    load_dotenv(dotenv_path=dotenv_path)
+
+    # return environment
+    # MCP_TYPE 검증 및 기본값 설정
+    mcp_type = os.getenv("MCP_TYPE", "stdio")
+    if mcp_type not in ['stdio', 'sse', 'streamable-http']:
+        logging.warning(f"Invalid MCP_TYPE: {mcp_type}, using default: stdio")
+        mcp_type = "stdio"
+
+    # MCP_PORT가 빈 문자열이면 기본값 사용
+    mcp_port_str = os.getenv("MCP_PORT", "8000")
+    mcp_port = int(mcp_port_str) if mcp_port_str.strip() else 8000
+
+    return EnvironmentConfig(
+        mcp_type=mcp_type,
+        mcp_host=os.getenv("MCP_HOST", "localhost"),
+        mcp_port=mcp_port,
+        mcp_path=os.getenv("MCP_PATH", "/mcp")
+    )