biszx-odoo-mcp 1.1.0__py3-none-any.whl → 1.1.2__py3-none-any.whl

biszx_odoo_mcp/server/tools.py

@@ -0,0 +1,432 @@
+"""
+MCP Tools for Odoo integration
+
+This module contains all the MCP tool functions for interacting with Odoo.
+"""
+
+from typing import Any, cast
+
+from biszx_odoo_mcp.exceptions import OdooMCPError, ToolError
+from biszx_odoo_mcp.server.context import AppContext
+from biszx_odoo_mcp.server.response import Response
+
+
+async def search_models(mcp: Any, query: str) -> dict[str, Any]:
+    """
+    Get a list of all available models in the Odoo system.
+    Args:
+        query: Search term to find models (searches in model name and display name)
+    Returns:
+        JSON string with matching models
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        data = app_context.odoo.search_models(query)
+        return Response(data=data).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+    except Exception as e:
+        tool_error = ToolError(
+            f"Unexpected error getting models: {str(e)}",
+            tool_name="search_models",
+            original_error=e,
+        )
+        return Response(error=tool_error.to_dict()).to_dict()
+
+
+async def get_model_info(mcp: Any, model_name: str) -> dict[str, Any]:
+    """
+    Get information about a specific Odoo model.
+    Args:
+        model_name: Name of the model (e.g., 'res.partner')
+    Returns:
+        Dictionary with model information
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        data = app_context.odoo.get_model_info(model_name)
+        return Response(data=data).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+    except Exception as e:
+        tool_error = ToolError(
+            f"Unexpected error getting model info: {str(e)}",
+            tool_name="get_model_info",
+            details={"model_name": model_name},
+            original_error=e,
+        )
+        return Response(error=tool_error.to_dict()).to_dict()
+
+
+async def get_model_fields(
+    mcp: Any, model_name: str, query_field: str
+) -> dict[str, Any]:
+    """
+    Get field definitions for a specific Odoo model.
+    Args:
+        model_name: Name of the model (e.g., 'res.partner')
+        query_field: Search term to find fields (searches in field name and string)
+    Returns:
+        Dictionary with field definitions
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        data = app_context.odoo.get_model_fields(model_name, query_field)
+        return Response(data=data).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+    except Exception as e:
+        tool_error = ToolError(
+            f"Unexpected error getting model fields: {str(e)}",
+            tool_name="get_model_fields",
+            details={"model_name": model_name},
+            original_error=e,
+        )
+        return Response(error=tool_error.to_dict()).to_dict()
+
+
+async def search_records(
+    mcp: Any,
+    model_name: str,
+    domain: list[Any],
+    fields: list[str] | None = None,
+    limit: int | None = None,
+    offset: int | None = None,
+    order: str | None = None,
+) -> dict[str, Any]:
+    """
+    Search for records in an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        domain: Search domain as list of tuples e.g., [['is_company', '=', true]]
+        fields: List of field names to return, None for all fields (default: None)
+        limit: Maximum number of records to return (default: None)
+        offset: Number of records to skip (default: None)
+        order: Sorting criteria e.g., 'name ASC, id DESC' (default: None)
+    Returns:
+        Dictionary with search results
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        data = app_context.odoo.search_read(
+            model_name, domain, fields=fields, limit=limit, offset=offset, order=order
+        )
+        return Response(data=data).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def read_records(
+    mcp: Any,
+    model_name: str,
+    ids: list[int],
+    fields: list[str] | None = None,
+) -> dict[str, Any]:
+    """
+    Read specific records by their IDs.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        ids: List of record IDs to read
+        fields: List of field names to return, None for all fields (default: None)
+    Returns:
+        Dictionary with record data
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        data = app_context.odoo.read_records(model_name, ids, fields=fields)
+        return Response(data=data).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def create_record(
+    mcp: Any,
+    model_name: str,
+    values: dict[str, Any],
+) -> dict[str, Any]:
+    """
+    Create a new record in an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        values: Dictionary with field values for the new record
+    Returns:
+        Dictionary with the created record ID
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        record_id = app_context.odoo.create_records(model_name, [values])
+        return Response(data={"id": record_id}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def create_records(
+    mcp: Any,
+    model_name: str,
+    values_list: list[dict[str, Any]],
+) -> dict[str, Any]:
+    """
+    Create multiple records in an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        values_list: List of dictionaries with field values for the new records
+    Returns:
+        Dictionary with the created record IDs
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        record_ids = app_context.odoo.create_records(model_name, values_list)
+        return Response(data={"ids": record_ids}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def write_record(
+    mcp: Any,
+    model_name: str,
+    record_id: int,
+    values: dict[str, Any],
+) -> dict[str, Any]:
+    """
+    Update a single record in an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        record_id: ID of the record to update
+        values: Dictionary with field values to update
+    Returns:
+        Dictionary with operation result
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        result = app_context.odoo.write_records(model_name, [record_id], values)
+        return Response(data={"success": result}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def write_records(
+    mcp: Any,
+    model_name: str,
+    record_ids: list[int],
+    values: dict[str, Any],
+) -> dict[str, Any]:
+    """
+    Update multiple records in an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        record_ids: List of record IDs to update
+        values: Dictionary with field values to update
+    Returns:
+        Dictionary with operation result
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        result = app_context.odoo.write_records(model_name, record_ids, values)
+        return Response(data={"success": result}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def unlink_record(
+    mcp: Any,
+    model_name: str,
+    record_id: int,
+) -> dict[str, Any]:
+    """
+    Delete a single record from an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        record_id: ID of the record to delete
+    Returns:
+        Dictionary with operation result
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        result = app_context.odoo.unlink_records(model_name, [record_id])
+        return Response(data={"success": result}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def unlink_records(
+    mcp: Any,
+    model_name: str,
+    record_ids: list[int],
+) -> dict[str, Any]:
+    """
+    Delete multiple records from an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        record_ids: List of record IDs to delete
+    Returns:
+        Dictionary with operation result
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        result = app_context.odoo.unlink_records(model_name, record_ids)
+        return Response(data={"success": result}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def search_count(
+    mcp: Any,
+    model_name: str,
+    domain: list[Any],
+) -> dict[str, Any]:
+    """
+    Count records that match a search domain.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        domain: Search domain as list of tuples e.g., [['is_company', '=', True]]
+    Returns:
+        Dictionary with the count of matching records
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        count = app_context.odoo.search_count(model_name, domain)
+        return Response(data={"count": count}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def search_ids(
+    mcp: Any,
+    model_name: str,
+    domain: list[Any],
+    offset: int | None = None,
+    limit: int | None = None,
+    order: str | None = None,
+) -> dict[str, Any]:
+    """
+    Search for record IDs that match a domain.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        domain: Search domain as list of tuples e.g., [['is_company', '=', True]]
+        offset: Number of records to skip (default: None)
+        limit: Maximum number of records to return (default: None)
+        order: Sorting criteria e.g., 'name ASC, id DESC' (default: None)
+    Returns:
+        Dictionary with list of matching record IDs
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        ids = app_context.odoo.search_ids(
+            model_name, domain, offset=offset, limit=limit, order=order
+        )
+        return Response(data={"ids": ids}).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def call_method(
+    mcp: Any,
+    model_name: str,
+    method_name: str,
+    args: list[Any] | None = None,
+    kwargs: dict[str, Any] | None = None,
+) -> dict[str, Any]:
+    """
+    Call a custom method on an Odoo model.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        method_name: Name of the method to call
+        args: Positional arguments to pass to the method (default: None)
+        kwargs: Keyword arguments to pass to the method (default: None)
+    Returns:
+        Dictionary with method result
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    if args is None:
+        args = []
+    if kwargs is None:
+        kwargs = {}
+
+    try:
+        result = app_context.odoo.call_method(model_name, method_name, args, kwargs)
+        return Response(data=result).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()
+
+
+async def search_and_update(
+    mcp: Any,
+    model_name: str,
+    domain: list[Any],
+    values: dict[str, Any],
+) -> dict[str, Any]:
+    """
+    Search for records and update them in one operation.
+    Args:
+        model_name: Name of the model e.g., 'res.partner'
+        domain: Search domain to find records to update
+        values: Dictionary with field values to update
+    Returns:
+        Dictionary with operation results including affected record count
+    """
+    # Access lifespan context to get the Odoo client
+    ctx = mcp.get_context()
+    app_context = cast(AppContext, ctx.request_context.lifespan_context)
+
+    try:
+        # First search for IDs
+        record_ids = app_context.odoo.search_ids(model_name, domain)
+        if not record_ids:
+            return Response(
+                data={"affected_records": 0, "message": "No records found"}
+            ).to_dict()
+
+        # Then update the found records
+        result = app_context.odoo.write_records(model_name, record_ids, values)
+        return Response(
+            data={
+                "affected_records": len(record_ids),
+                "record_ids": record_ids,
+                "updated": result,
+                "update_result": result,  # For backward compatibility
+            }
+        ).to_dict()
+    except OdooMCPError as e:
+        return Response(error=e.to_dict()).to_dict()

biszx_odoo_mcp/tools/config.py

@@ -0,0 +1,98 @@
+"""
+Odoo Configuration for MCP server integration
+"""
+
+import os
+import re
+from typing import Any
+
+from loguru import logger
+
+
+class Config:
+    """
+    Odoo client configuration
+    """
+
+    url: str
+    db: str
+    username: str
+    password: str
+    timeout: int
+    verify_ssl: bool
+
+    def __init__(self) -> None:
+        """
+        Initialize the Config object and load configuration
+        """
+        self.__dict__.update(self.load_config())
+        self.url = self._prepare_url(self.url)
+
+        logger.info("🔧 Odoo client configuration:")
+        logger.info(f"  URL: {self.url}")
+        logger.info(f"  Database: {self.db}")
+        logger.info(f"  Username: {self.username}")
+        logger.info(f"  Timeout: {self.timeout}s")
+        logger.info(f"  Verify SSL: {self.verify_ssl}")
+
+    def load_config(self) -> dict[str, Any]:
+        """
+        Load Odoo configuration from environment variables or config file
+
+        Raises:
+            OSError: If required environment variables are missing
+        Returns:
+            dict: Configuration parameters for Odoo client
+        """
+
+        self._validate_config()
+        return {
+            "url": os.environ["ODOO_URL"],
+            "db": os.environ["ODOO_DB"],
+            "username": os.environ["ODOO_USERNAME"],
+            "password": os.environ["ODOO_PASSWORD"],
+            "timeout": int(os.environ.get("ODOO_TIMEOUT", "30")),
+            "verify_ssl": os.environ.get("ODOO_VERIFY_SSL", "1").lower()
+            in ("1", "true", "yes"),
+        }
+
+    def _validate_config(self) -> None:
+        """
+        Validate the loaded configuration parameters
+        Raises:
+            ValueError: If any required parameter is missing or invalid
+        """
+        required_env = {"ODOO_URL", "ODOO_DB", "ODOO_USERNAME", "ODOO_PASSWORD"}
+        for var in required_env:
+            if var not in os.environ:
+                raise OSError(f"Missing required environment variable: {var}")
+        if "ODOO_TIMEOUT" in os.environ:
+            try:
+                int(os.environ["ODOO_TIMEOUT"])
+            except ValueError:
+                raise ValueError("ODOO_TIMEOUT must be an integer") from None
+        if "ODOO_VERIFY_SSL" in os.environ and os.environ[
+            "ODOO_VERIFY_SSL"
+        ].lower() not in (
+            "1",
+            "true",
+            "yes",
+            "0",
+            "false",
+            "no",
+        ):
+            raise ValueError("ODOO_VERIFY_SSL must be 1, true, yes, 0, false, or no")
+
+    def _prepare_url(self, url: str) -> str:
+        """
+        Prepare the URL by ensuring it has a protocol and no trailing slash
+
+        Args:
+            url: The URL to prepare
+
+        Returns:
+            str: The prepared URL
+        """
+        if not re.match(r"^https?://", url):
+            url = f"http://{url}"
+        return url.rstrip("/")