first commit

config.py

@@ -0,0 +1,61 @@
+import os
+from typing import Dict, Any
+from dotenv import load_dotenv
+
+# 加载环境变量文件
+load_dotenv()
+
+class DatabaseConfig:
+    """数据库配置类"""
+    
+    # MySQL配置
+    MYSQL_CONFIG = {
+        "host": os.getenv("MYSQL_HOST", "localhost"),
+        "port": int(os.getenv("MYSQL_PORT", "3306")),
+        "username": os.getenv("MYSQL_USERNAME", "root"),
+        "password": os.getenv("MYSQL_PASSWORD", "password"),
+        "database": os.getenv("MYSQL_DATABASE", "test_db")
+    }
+    
+    # Oracle配置
+    ORACLE_CONFIG = {
+        "host": os.getenv("ORACLE_HOST", "192.168.13.27"),
+        "port": int(os.getenv("ORACLE_PORT", "1521")),
+        "username": os.getenv("ORACLE_USERNAME", "bizuser"),
+        "password": os.getenv("ORACLE_PASSWORD", "MySecurePass123"),
+        "service_name": os.getenv("ORACLE_SERVICE_NAME", "ORCLPDB1")
+    }
+    
+    # SQL Server配置
+    SQLSERVER_CONFIG = {
+        "host": os.getenv("SQLSERVER_HOST", "localhost"),
+        "port": int(os.getenv("SQLSERVER_PORT", "1433")),
+        "username": os.getenv("SQLSERVER_USERNAME", "sa"),
+        "password": os.getenv("SQLSERVER_PASSWORD", "password"),
+        "database": os.getenv("SQLSERVER_DATABASE", "master")
+    }
+    
+    # PostgreSQL配置
+    POSTGRESQL_CONFIG = {
+        "host": os.getenv("POSTGRESQL_HOST", "localhost"),
+        "port": int(os.getenv("POSTGRESQL_PORT", "5432")),
+        "username": os.getenv("POSTGRESQL_USERNAME", "postgres"),
+        "password": os.getenv("POSTGRESQL_PASSWORD", "password"),
+        "database": os.getenv("POSTGRESQL_DATABASE", "postgres")
+    }
+    
+    @classmethod
+    def get_config(cls, db_type: str) -> Dict[str, Any]:
+        """根据数据库类型获取配置"""
+        config_map = {
+            "mysql": cls.MYSQL_CONFIG,
+            "oracle": cls.ORACLE_CONFIG,
+            "sqlserver": cls.SQLSERVER_CONFIG,
+            "postgresql": cls.POSTGRESQL_CONFIG
+        }
+        return config_map.get(db_type.lower(), {})
+    
+    @classmethod
+    def is_sample_data_enabled(cls) -> bool:
+        """检查是否启用示例数据初始化"""
+        return os.getenv("ENABLE_SAMPLE_DATA", "true").lower() == "true"