import argparse
import logging
import sys
from . import __version__
from .server import mcp
def run_command(args):
"""Run the MCP server."""
# Set logging to WARNING to quiet info logs
logging.basicConfig(level=logging.WARNING)
mcp.run()
def main():
"""Main entry point for the DX MCP Server."""
parser = argparse.ArgumentParser(description="DX MCP Server for database queries")
parser.add_argument(
"--version", action="version", version=f"%(prog)s {__version__}"
)
subparsers = parser.add_subparsers(dest="command", help="Commands")
run_parser = subparsers.add_parser("run", help="Run the MCP server")
args = parser.parse_args()
# If no command is provided, default to run
if args.command is None:
run_command(args)
elif args.command == "run":
run_command(args)
if __name__ == "__main__":
main()
