Create wms_client.py

services/wms_client.py

@@ -0,0 +1,42 @@
+
+from typing import Dict, Any
+from public_api.public_api_auth import login
+from public_api.field_mapping import TOOL_REGISTRY
+from public_api.utils import format_bot_response, flatten_json
+
+class WmsClient:
+    def __init__(self):
+        self.session_data = {"wms_auth": None, "user_warehouse_id": None}
+
+    def connect(self) -> bool:
+        session, wh_id, ok = login()
+        if not ok:
+            return False
+        self.session_data["wms_auth"] = session
+        self.session_data["user_warehouse_id"] = wh_id
+        return True
+
+    def call_tool(self, tool_name: str, **kwargs) -> Dict[str, Any]:
+        cfg = TOOL_REGISTRY[tool_name]
+        func = cfg["function"]
+        mapping = cfg["mapping"]
+        formatters = cfg.get("formatters", {})
+        response_key = cfg.get("response_key", "data")
+
+        warehouse_id = kwargs.get("warehouse_id") or self.session_data["user_warehouse_id"]
+
+        raw = func(self.session_data, warehouse_id, **kwargs)
+        payload = raw.get(response_key, raw)
+        flat = flatten_json(payload)
+
+        requested_order = kwargs.get("fields")
+        from public_api.utils import extract_fields as extract_fields_ordered
+        extracted = extract_fields_ordered(flat, mapping, formatters, requested_order=requested_order)
+
+        return format_bot_response(
+            text="OK" if not raw.get("error") else "ERROR",
+            source=tool_name,
+            data=extracted,
+            state={"warehouse_id": warehouse_id},
+            debug=raw.get("error") or ""
+        )