smartrack-sdk 1.0.0__py3-none-any.whl

smartrack_sdk/callback/flask_handler.py

@@ -0,0 +1,137 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Flask WebAPI回调处理器
+提供Flask框架的WebAPI回调处理功能
+"""
+
+import logging
+from typing import Dict, Any, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from flask import Flask
+
+try:
+    from flask import Flask, request, jsonify
+    FLASK_AVAILABLE = True
+except ImportError:
+    FLASK_AVAILABLE = False
+
+from .handler import CellEventHandler, CellEventCallbackService
+from ..models.callback import CellEventHandleRequest, CellEventHandleResponse
+
+logger = logging.getLogger(__name__)
+
+
+class FlaskCellEventHandler:
+    """Flask储位事件回调处理器"""
+
+    def __init__(self, app: 'Flask' = None, event_handler: CellEventHandler = None):
+        """
+        初始化Flask回调处理器
+
+        Args:
+            app: Flask应用实例
+            event_handler: 事件处理器
+        """
+        if not FLASK_AVAILABLE:
+            raise ImportError("Flask not available. Install with: pip install Flask")
+
+        self.event_handler = event_handler or CellEventCallbackService()
+
+        if app:
+            self.init_app(app)
+
+    def init_app(self, app: 'Flask') -> None:
+        """
+        初始化Flask应用
+
+        Args:
+            app: Flask应用实例
+        """
+        app.add_url_rule(
+            '/api/SmartRack/CellEventHandle',
+            endpoint='cell_event_handle',
+            view_func=self.handle_cell_event,
+            methods=['POST']
+        )
+
+        app.add_url_rule(
+            '/api/SmartRack/health',
+            endpoint='health',
+            view_func=self.health,
+            methods=['GET']
+        )
+
+        logger.info("Flask SmartRack回调路由注册完成")
+
+    def handle_cell_event(self):
+        """处理储位事件回调请求"""
+        try:
+            # 获取请求数据
+            request_data = request.get_json()
+            if not request_data:
+                return jsonify({
+                    'code': 100100,
+                    'message': '请求数据为空',
+                    'sessionId': '',
+                    'timestamp': self._get_timestamp()
+                }), 400
+
+            # 创建回调请求对象
+            callback_request = CellEventHandleRequest.from_dict(request_data)
+            logger.info(f"收到储位事件回调: {callback_request}")
+
+            # 处理事件
+            response = self.event_handler.handle(callback_request)
+
+            # 返回响应
+            return jsonify(response.to_dict())
+
+        except Exception as e:
+            logger.error("处理储位事件回调时发生错误", exc_info=True)
+            return jsonify({
+                'code': 100101,
+                'message': f'处理回调时发生错误: {str(e)}',
+                'sessionId': '',
+                'timestamp': self._get_timestamp()
+            }), 500
+
+    def health(self):
+        """健康检查接口"""
+        return jsonify({
+            'status': 'ok',
+            'message': 'SmartRack Callback Service is running',
+            'timestamp': self._get_timestamp()
+        })
+
+    def _get_timestamp(self) -> str:
+        """获取当前时间戳"""
+        from datetime import datetime
+        return datetime.now().isoformat()
+
+
+def create_flask_app(event_handler: CellEventHandler = None) -> 'Flask':
+    """
+    创建Flask应用
+
+    Args:
+        event_handler: 事件处理器
+
+    Returns:
+        Flask应用实例
+    """
+    if not FLASK_AVAILABLE:
+        raise ImportError("Flask not available. Install with: pip install Flask")
+
+    app = Flask(__name__)
+
+    # 配置
+    app.config['JSON_AS_ASCII'] = False
+
+    # 初始化回调处理器
+    flask_handler = FlaskCellEventHandler(app, event_handler)
+
+    logger.info("Flask应用创建完成")
+    return app

smartrack_sdk/callback/handler.py

@@ -0,0 +1,220 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+回调处理器接口和默认实现
+"""
+
+import logging
+from abc import ABC, abstractmethod
+from typing import List
+
+from ..models.callback import CellEventHandleRequest, CellEventHandleResponse, CellEventData
+from ..models.enums import OperationResultCode, TriggerState
+
+logger = logging.getLogger(__name__)
+
+
+class CellEventHandler(ABC):
+    """储位事件回调处理器接口"""
+
+    @abstractmethod
+    def handle(self, request: CellEventHandleRequest) -> CellEventHandleResponse:
+        """
+        处理储位事件回调
+
+        Args:
+            request: 回调请求
+
+        Returns:
+            回调响应
+        """
+        pass
+
+
+class CellEventCallbackService(CellEventHandler):
+    """默认储位事件回调处理服务"""
+
+    def handle(self, request: CellEventHandleRequest) -> CellEventHandleResponse:
+        """
+        处理储位事件回调
+
+        Args:
+            request: 回调请求
+
+        Returns:
+            回调响应
+        """
+        logger.info(f"处理储位事件回调 - 用户: {request.user_id}, 客户端: {request.client_id}")
+
+        event_data_list = request.data
+        if not event_data_list:
+            logger.warning("没有事件数据")
+            return CellEventHandleResponse.error(100102, "没有事件数据", request.session_id)
+
+        try:
+            # 处理每个事件数据
+            for event_data in event_data_list:
+                self._process_single_event(event_data)
+
+            # 所有事件处理成功
+            return CellEventHandleResponse.success(request.session_id)
+
+        except Exception as e:
+            logger.error("处理事件失败", exc_info=True)
+            return CellEventHandleResponse.error(100103, f"处理事件失败: {str(e)}", request.session_id)
+
+    def _process_single_event(self, event_data: CellEventData) -> None:
+        """
+        处理单个事件数据
+
+        Args:
+            event_data: 事件数据
+        """
+        logger.info(f"处理事件 - 储位: {event_data.cell_id}, GRN: {event_data.grn}, "
+                   f"状态: {event_data.cell_state.name_en}, 结果: {event_data.code}")
+
+        # 检查操作结果
+        if event_data.code != OperationResultCode.SUCCESS.value:
+            self._handle_operation_error(event_data)
+            return
+
+        # 根据触发状态处理不同操作
+        if event_data.cell_state == TriggerState.PUT_IN:
+            self._handle_put_in_event(event_data)
+        elif event_data.cell_state == TriggerState.TAKE_OUT:
+            self._handle_take_out_event(event_data)
+        else:
+            logger.warning(f"未知的储位状态: {event_data.cell_state}")
+
+    def _handle_put_in_event(self, event_data: CellEventData) -> None:
+        """
+        处理放入事件
+
+        Args:
+            event_data: 事件数据
+        """
+        logger.info(f"处理入库事件 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+
+        # 这里可以添加具体的入库业务逻辑
+        try:
+            # 验证数据
+            if not event_data.grn or not event_data.grn.strip():
+                raise ValueError("GRN不能为空")
+
+            # 更新库存
+            self._update_inventory_for_put_in(event_data)
+
+            # 发送通知
+            self._send_put_in_notification(event_data)
+
+            logger.info(f"入库事件处理完成 - 储位: {event_data.cell_id}")
+
+        except Exception as e:
+            logger.error(f"入库事件处理失败 - 储位: {event_data.cell_id}", exc_info=True)
+            raise RuntimeError(f"入库处理失败: {str(e)}")
+
+    def _handle_take_out_event(self, event_data: CellEventData) -> None:
+        """
+        处理取出事件
+
+        Args:
+            event_data: 事件数据
+        """
+        logger.info(f"处理出库事件 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+
+        # 这里可以添加具体的出库业务逻辑
+        try:
+            # 验证数据
+            if not event_data.grn or not event_data.grn.strip():
+                raise ValueError("GRN不能为空")
+
+            # 更新库存
+            self._update_inventory_for_take_out(event_data)
+
+            # 发送通知
+            self._send_take_out_notification(event_data)
+
+            logger.info(f"出库事件处理完成 - 储位: {event_data.cell_id}")
+
+        except Exception as e:
+            logger.error(f"出库事件处理失败 - 储位: {event_data.cell_id}", exc_info=True)
+            raise RuntimeError(f"出库处理失败: {str(e)}")
+
+    def _handle_operation_error(self, event_data: CellEventData) -> None:
+        """
+        处理操作错误
+
+        Args:
+            event_data: 事件数据
+        """
+        logger.error(f"操作错误 - 储位: {event_data.cell_id}, GRN: {event_data.grn}, "
+                    f"代码: {event_data.code}, 消息: {event_data.message}")
+
+        # 处理不同类型的错误
+        try:
+            if event_data.code == OperationResultCode.ILLEGAL_STOCK_IN.value:
+                self._handle_illegal_stock_in(event_data)
+            elif event_data.code == OperationResultCode.ILLEGAL_STOCK_OUT.value:
+                self._handle_illegal_stock_out(event_data)
+            elif event_data.code == OperationResultCode.ILLEGAL_STOCK_IN_CLEAR_ERROR.value:
+                self._handle_illegal_stock_in_clear_error(event_data)
+            elif event_data.code == OperationResultCode.ILLEGAL_STOCK_OUT_CLEAR_ERROR.value:
+                self._handle_illegal_stock_out_clear_error(event_data)
+            else:
+                self._handle_unknown_error(event_data)
+        except Exception as e:
+            logger.error(f"处理操作错误时发生异常", exc_info=True)
+
+    # 以下方法为示例实现，实际业务中需要根据具体需求实现
+
+    def _update_inventory_for_put_in(self, event_data: CellEventData) -> None:
+        """更新库存信息 - 入库"""
+        logger.debug(f"更新库存 - 入库: 储位={event_data.cell_id}, GRN={event_data.grn}")
+        # 实际实现中，这里会更新数据库中的库存记录
+
+    def _update_inventory_for_take_out(self, event_data: CellEventData) -> None:
+        """更新库存信息 - 出库"""
+        logger.debug(f"更新库存 - 出库: 储位={event_data.cell_id}, GRN={event_data.grn}")
+        # 实际实现中，这里会更新数据库中的库存记录
+
+    def _send_put_in_notification(self, event_data: CellEventData) -> None:
+        """发送入库通知"""
+        logger.debug(f"发送入库通知 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        # 实际实现中，这里会发送邮件、短信或其他通知
+
+    def _send_take_out_notification(self, event_data: CellEventData) -> None:
+        """发送出库通知"""
+        logger.debug(f"发送出库通知 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        # 实际实现中，这里会发送邮件、短信或其他通知
+
+    def _handle_illegal_stock_in(self, event_data: CellEventData) -> None:
+        """处理非法上架"""
+        logger.warning(f"非法上架 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        self._send_alert("非法上架警告", event_data)
+
+    def _handle_illegal_stock_out(self, event_data: CellEventData) -> None:
+        """处理非法下架"""
+        logger.warning(f"非法下架 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        self._send_alert("非法下架警告", event_data)
+
+    def _handle_illegal_stock_in_clear_error(self, event_data: CellEventData) -> None:
+        """处理非法上架清错"""
+        logger.warning(f"非法上架清错 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        self._send_alert("非法上架清错警告", event_data)
+
+    def _handle_illegal_stock_out_clear_error(self, event_data: CellEventData) -> None:
+        """处理非法下架清错"""
+        logger.warning(f"非法下架清错 - 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        self._send_alert("非法下架清错警告", event_data)
+
+    def _handle_unknown_error(self, event_data: CellEventData) -> None:
+        """处理未知错误"""
+        logger.error(f"未知错误 - 储位: {event_data.cell_id}, GRN: {event_data.grn}, "
+                    f"错误代码: {event_data.code}")
+        self._send_alert("未知错误警告", event_data)
+
+    def _send_alert(self, alert_type: str, event_data: CellEventData) -> None:
+        """发送告警通知"""
+        logger.debug(f"发送告警 - 类型: {alert_type}, 储位: {event_data.cell_id}, GRN: {event_data.grn}")
+        # 实际实现中，这里会发送告警邮件、短信或其他通知

smartrack_sdk/callback/service.py

@@ -0,0 +1,14 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+回调服务模块
+导出回调处理器和服务
+"""
+
+from .handler import CellEventHandler, CellEventCallbackService
+
+__all__ = [
+    "CellEventHandler",
+    "CellEventCallbackService",
+]

smartrack_sdk/client/__init__.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+客户端包
+包含SmartRack客户端和相关功能
+"""
+
+from .smart_rack_client import SmartRackClient
+
+__all__ = [
+    "SmartRackClient",
+]