Merge pull request #16 from abel9851/fix/add-package-entrypoint

feat: add how to set args

Author: abel9851

src/mcp_server_mariadb/server.py

@@ -1,6 +1,7 @@
+import argparse
 import os
 from contextlib import closing
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 
 import mariadb
 from dotenv import load_dotenv
@@ -16,19 +17,39 @@
 READ_ONLY_KEYWORD_NAMES = ", ".join(READ_ONLY_KEYWORDS)
 
 
+def get_arguments() -> dict:
+    """Parse command-line arguments and return as a dictionary."""
+    parser = argparse.ArgumentParser(description="MariaDB Configuration")
+    parser.add_argument("--host", help="MariaDB host")
+    parser.add_argument("--port", type=int, help="MariaDB port")
+    parser.add_argument("--user", help="MariaDB user")
+    parser.add_argument("--password", help="MariaDB password")
+    parser.add_argument("--database", help="MariaDB database")
+    args = parser.parse_args()
+
+    return {k: v for k, v in vars(args).items() if v is not None}
+
+
 @dataclass
 class DBconfig:
-    host: str = os.getenv("MARIADB_HOST", "localhost")
-    port: int = int(os.getenv("MARIADB_PORT", "3306"))
-    user: str = os.getenv("MARIADB_USER", "")
-    password: str = os.getenv("MARIADB_PASSWORD", "")
-    database: str = os.getenv("MARIADB_DATABASE", "")
+    """Database configuration"""
+
+    host: str = field(default_factory=lambda: os.getenv("MARIADB_HOST", "localhost"))
+    port: int = field(default_factory=lambda: int(os.getenv("MARIADB_PORT", "3306")))
+    user: str = field(default_factory=lambda: os.getenv("MARIADB_USER", ""))
+    password: str = field(default_factory=lambda: os.getenv("MARIADB_PASSWORD", ""))
+    database: str = field(default_factory=lambda: os.getenv("MARIADB_DATABASE", ""))
+
+    @classmethod
+    def from_args(cls) -> "DBconfig":
+        """Create a DBconfig instance from command-line arguments and environment variables."""
+        cli_args = get_arguments()
+        return cls(**{**cls().__dict__, **cli_args})
 
 
 def get_connection():
     """Create a connection to the database connection"""
-
-    config = DBconfig()
+    config = DBconfig.from_args()
 
     try:
         conn = mariadb.connect(