smartrack-sdk 1.0.0__py3-none-any.whl

examples/test_client.py

@@ -0,0 +1,303 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+SmartRack SDK 测试客户端
+根据提供的测试参数进行API调用测试
+"""
+
+import logging
+from datetime import datetime
+from smartrack_sdk import SmartRackClient
+from smartrack_sdk.models.requests import (
+    StockInRequest, StockOutRequest, ClearErrorRequest,
+    ResetCellRequest, RackTestRequest, StockInFinishRequest
+)
+from smartrack_sdk.models.enums import LedColors, RackTestMode
+from smartrack_sdk.exceptions import SmartRackException
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+
+class SmartRackTestClient:
+    """SmartRack测试客户端"""
+
+    def __init__(self, base_url: str, user_id: str = "test", client_id: str = "MES"):
+        """
+        初始化测试客户端
+
+        Args:
+            base_url: API基础URL
+            user_id: 用户ID
+            client_id: 客户端ID
+        """
+        self.base_url = base_url
+        self.user_id = user_id
+        self.client_id = client_id
+        self.client = None
+
+    def test_all_apis(self):
+        """测试所有API"""
+        logger.info("开始SmartRack API测试")
+        logger.info(f"API地址: {self.base_url}")
+        logger.info(f"用户ID: {self.user_id}")
+        logger.info(f"客户端ID: {self.client_id}")
+        logger.info("=" * 60)
+
+        try:
+            # 创建客户端
+            self._create_client()
+
+            # 测试各个API
+            self._test_stock_in_request()
+            self._test_rack_test()
+            self._test_clear_error()
+            self._test_reset_cell()
+
+            logger.info("=" * 60)
+            logger.info("所有API测试完成!")
+
+        except Exception as e:
+            logger.error(f"测试过程中出错: {e}", exc_info=True)
+
+        finally:
+            self._close_client()
+
+    def _create_client(self):
+        """创建客户端"""
+        logger.info("创建SmartRack客户端...")
+        self.client = SmartRackClient(
+            base_url=self.base_url,
+            timeout=30
+        )
+        logger.info("客户端创建完成")
+
+    def _close_client(self):
+        """关闭客户端"""
+        if self.client:
+            self.client.close()
+            logger.info("客户端已关闭")
+
+    def _generate_sheet_id(self, prefix: str = "SI") -> str:
+        """生成单号"""
+        return f"{prefix}-{datetime.now().strftime('%Y%m%d%H%M%S')}"
+
+    def _print_result(self, api_name: str, response, additional_info: str = ""):
+        """打印测试结果"""
+        logger.info(f"=== {api_name} 测试结果 ===")
+        logger.info(f"Code: {response.code}")
+        logger.info(f"Message: {response.message}")
+        if response.session_id:
+            logger.info(f"SessionId: {response.session_id}")
+        if additional_info:
+            logger.info(f"Additional Info: {additional_info}")
+        logger.info(f"Success: {response.is_success}")
+        logger.info("-" * 40)
+
+    def _test_stock_in_request(self):
+        """测试入库请求"""
+        logger.info("测试: 入库请求")
+
+        try:
+            # 生成入库单号
+            sheet_id = self._generate_sheet_id()
+
+            # 创建入库请求
+            request = StockInRequest(self.user_id, self.client_id)
+            request.set_sheet_id(sheet_id)
+            request.set_rack_id("A001")
+            request.set_grn("TEST-PRODUCT-001")
+            request.set_led_color(LedColors.GREEN)
+            request.set_blink(True)
+
+            logger.info(f"发送入库请求 - 单号: {sheet_id}")
+            logger.info(f"料架: A001, GRN: TEST-PRODUCT-001")
+
+            response = self.client.stock_in_request(request)
+            self._print_result("入库请求", response, f"单号: {sheet_id}")
+
+            if response.is_success:
+                logger.info("入库请求成功，等待3秒后测试入库完成...")
+                import time
+                time.sleep(3)
+
+                # 测试入库完成
+                self._test_stock_in_finish(sheet_id)
+            else:
+                logger.error("入库请求失败，跳过入库完成测试")
+
+        except SmartRackException as e:
+            logger.error(f"入库请求异常: {e}")
+
+    def _test_stock_in_finish(self, sheet_id: str):
+        """测试入库完成"""
+        logger.info("测试: 入库完成")
+
+        try:
+            # 创建入库完成请求
+            request = StockInFinishRequest(self.user_id, self.client_id)
+            request.set_sheet_id(sheet_id)
+
+            logger.info(f"发送入库完成请求 - 单号: {sheet_id}")
+
+            response = self.client.stock_in_finish(request)
+            self._print_result("入库完成", response, f"单号: {sheet_id}")
+
+        except SmartRackException as e:
+            logger.error(f"入库完成异常: {e}")
+
+    def _test_rack_test(self):
+        """测试料架测试"""
+        logger.info("测试: 料架测试")
+
+        try:
+            # RGB测试
+            request = RackTestRequest(self.user_id, self.client_id)
+            request.set_rack_id("A001")
+            request.set_test_mode(RackTestMode.RGB_TEST)
+            request.set_led_color(LedColors.WHITE)
+            request.set_blink(False)
+
+            logger.info("发送RGB测试请求 - 料架: A001")
+
+            response = self.client.rack_test(request)
+            self._print_result("RGB测试", response, "测试模式: RGB_TEST")
+
+            if response.is_success:
+                logger.info("RGB测试成功，等待3秒后测试全部灭灯...")
+                import time
+                time.sleep(3)
+
+                # 测试全部灭灯
+                self._test_rack_all_off()
+            else:
+                logger.error("RGB测试失败，跳过全部灭灯测试")
+
+        except SmartRackException as e:
+            logger.error(f"料架测试异常: {e}")
+
+    def _test_rack_all_off(self):
+        """测试全部灭灯"""
+        logger.info("测试: 全部灭灯")
+
+        try:
+            request = RackTestRequest(self.user_id, self.client_id)
+            request.set_rack_id("A001")
+            request.set_test_mode(RackTestMode.ALL_OFF)
+            request.set_led_color(LedColors.OFF)
+            request.set_blink(False)
+
+            logger.info("发送全部灭灯请求 - 料架: A001")
+
+            response = self.client.rack_test(request)
+            self._print_result("全部灭灯", response, "测试模式: ALL_OFF")
+
+        except SmartRackException as e:
+            logger.error(f"全部灭灯异常: {e}")
+
+    def _test_clear_error(self):
+        """测试清除错误"""
+        logger.info("测试: 清除错误")
+
+        try:
+            # 清除指定储位的错误
+            request = ClearErrorRequest(self.user_id, self.client_id)
+            request.set_cell_id("A0010001")
+
+            logger.info("发送清除错误请求 - 储位: A0010001")
+
+            response = self.client.clear_error(request)
+            self._print_result("清除错误", response)
+
+            if response.is_success and response.result_data:
+                logger.info("清错结果:")
+                for item in response.result_data:
+                    status_text = "空" if item.is_empty() else "非空"
+                    logger.info(f"  储位: {item.cell}, GRN: {item.grn}, 状态: {status_text}")
+            else:
+                logger.error("清除错误失败或无返回数据")
+
+        except SmartRackException as e:
+            logger.error(f"清除错误异常: {e}")
+
+    def _test_reset_cell(self):
+        """测试重置储位状态"""
+        logger.info("测试: 重置储位状态")
+
+        try:
+            # 重置多个储位状态
+            request = ResetCellRequest(self.user_id, self.client_id)
+            request.add_cell("A0010001")
+            request.add_cell("A0010002")
+            request.add_cell("A0010003")
+
+            logger.info(f"发送重置储位请求 - 储位列表: {request.cell_list}")
+
+            response = self.client.reset_cell(request)
+            self._print_result("重置储位状态", response, f"储位数量: {len(request.cell_list)}")
+
+        except SmartRackException as e:
+            logger.error(f"重置储位状态异常: {e}")
+
+    def test_single_api(self, api_name: str):
+        """测试单个API"""
+        logger.info(f"测试单个API: {api_name}")
+
+        try:
+            self._create_client()
+
+            if api_name == "stock_in":
+                self._test_stock_in_request()
+            elif api_name == "stock_in_finish":
+                sheet_id = self._generate_sheet_id()
+                self._test_stock_in_finish(sheet_id)
+            elif api_name == "rack_test":
+                self._test_rack_test()
+            elif api_name == "clear_error":
+                self._test_clear_error()
+            elif api_name == "reset_cell":
+                self._test_reset_cell()
+            else:
+                logger.error(f"未知的API名称: {api_name}")
+
+        except Exception as e:
+            logger.error(f"测试API {api_name} 时出错: {e}", exc_info=True)
+
+        finally:
+            self._close_client()
+
+
+def main():
+    """主函数"""
+    # 测试参数
+    base_url = "http://192.168.8.3:7800"
+    user_id = "test"
+    client_id = "MES"
+
+    logger.info("SmartRack SDK 测试客户端")
+    logger.info(f"API地址: {base_url}")
+    logger.info(f"用户ID: {user_id}")
+    logger.info(f"客户端ID: {client_id}")
+    logger.info("=" * 60)
+
+    # 检查命令行参数
+    import sys
+    if len(sys.argv) > 1:
+        api_name = sys.argv[1]
+        logger.info(f"测试单个API: {api_name}")
+
+        client = SmartRackTestClient(base_url, user_id, client_id)
+        client.test_single_api(api_name)
+    else:
+        logger.info("测试所有API")
+        client = SmartRackTestClient(base_url, user_id, client_id)
+        client.test_all_apis()
+
+
+if __name__ == "__main__":
+    main()

smartrack_sdk/__init__.py

@@ -0,0 +1,129 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+SmartRack SDK for Python
+智能料架系统Python SDK
+
+主要功能:
+- 完整的API接口支持
+- WebAPI回调处理
+- 强类型模型
+- 异常处理
+- 日志记录
+"""
+
+__version__ = "1.0.0"
+__author__ = "SmartRack SDK Team"
+__email__ = "support@smartrack.com"
+
+# 导入主要的类和函数
+from .client import SmartRackClient
+from .exceptions import (
+    SmartRackException,
+    SmartRackApiException,
+    SmartRackNetworkException,
+    SmartRackConfigurationException,
+    SmartRackValidationException,
+)
+
+# 导入模型类
+from .models import (
+    LedColors,
+    CellStatus,
+    RackTestMode,
+    OperationResultCode,
+    TriggerState,
+)
+
+# 导入请求和响应类
+from .models.requests import (
+    BaseRequest,
+    ClearErrorRequest,
+    RackTestRequest,
+    StockInRequest,
+    StockOutRequest,
+    StockOutSheetRequest,
+    StockOutSheetCancelRequest,
+    StockOutSheetFinishRequest,
+    StockInFinishRequest,
+    ResetCellRequest,
+    GetStockByGRNListRequest,
+)
+
+from .models.responses import (
+    BaseResponse,
+    ClearErrorResponse,
+    StockDataResponse,
+    StockDataDto,
+    StockOutResponse,
+    StockOutResponseDto,
+    StockOutSheetResponse,
+)
+
+from .models.callback import (
+    CellEventData,
+    CellEventHandleRequest,
+    CellEventHandleResponse,
+)
+
+# 导入回调处理类
+from .callback import (
+    CellEventHandler,
+    CellEventCallbackService,
+)
+
+__all__ = [
+    # 版本信息
+    "__version__",
+    "__author__",
+    "__email__",
+
+    # 核心客户端
+    "SmartRackClient",
+
+    # 异常类
+    "SmartRackException",
+    "SmartRackApiException",
+    "SmartRackNetworkException",
+    "SmartRackConfigurationException",
+    "SmartRackValidationException",
+
+    # 枚举类
+    "LedColors",
+    "CellStatus",
+    "RackTestMode",
+    "OperationResultCode",
+    "TriggerState",
+
+    # 请求类
+    "BaseRequest",
+    "ClearErrorRequest",
+    "RackTestRequest",
+    "StockInRequest",
+    "StockOutRequest",
+    "StockOutSheetRequest",
+    "StockOutSheetCancelRequest",
+    "StockOutSheetFinishRequest",
+    "StockInFinishRequest",
+    "ResetCellRequest",
+    "GetStockByGRNListRequest",
+
+    # 响应类
+    "BaseResponse",
+    "ClearErrorResponse",
+    "StockDataResponse",
+    "StockDataDto",
+    "StockOutResponse",
+    "StockOutResponseDto",
+    "StockOutSheetResponse",
+
+    # 回调模型
+    "CellEventData",
+    "CellEventHandleRequest",
+    "CellEventHandleResponse",
+
+    # 回调处理
+    "CellEventHandler",
+    "CellEventCallbackService",
+]

smartrack_sdk/callback/__init__.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+回调处理包
+包含WebAPI回调处理相关功能
+"""
+
+from .handler import CellEventHandler
+from .service import CellEventCallbackService
+from .flask_handler import FlaskCellEventHandler
+from .fastapi_handler import FastAPICellEventHandler
+
+__all__ = [
+    "CellEventHandler",
+    "CellEventCallbackService",
+    "FlaskCellEventHandler",
+    "FastAPICellEventHandler",
+]

smartrack_sdk/callback/fastapi_handler.py

@@ -0,0 +1,151 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+FastAPI WebAPI回调处理器
+提供FastAPI框架的WebAPI回调处理功能
+"""
+
+import logging
+from typing import Dict, Any, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from fastapi import FastAPI
+
+try:
+    from fastapi import FastAPI, HTTPException
+    from pydantic import BaseModel
+    FASTAPI_AVAILABLE = True
+except ImportError:
+    FASTAPI_AVAILABLE = False
+
+from .handler import CellEventHandler, CellEventCallbackService
+from ..models.callback import CellEventHandleRequest, CellEventHandleResponse
+
+logger = logging.getLogger(__name__)
+
+
+if FASTAPI_AVAILABLE:
+    class CellEventRequest(BaseModel):
+        """储位事件请求模型"""
+        UserId: str
+        ClientId: str
+        SessionId: str
+        Timestamp: str
+        Data: list
+
+    class CellEventResponse(BaseModel):
+        """储位事件响应模型"""
+        code: int
+        message: str
+        sessionId: str
+        timestamp: str
+
+    class HealthResponse(BaseModel):
+        """健康检查响应模型"""
+        status: str
+        message: str
+        timestamp: str
+
+
+class FastAPICellEventHandler:
+    """FastAPI储位事件回调处理器"""
+
+    def __init__(self, app: 'FastAPI' = None, event_handler: CellEventHandler = None):
+        """
+        初始化FastAPI回调处理器
+
+        Args:
+            app: FastAPI应用实例
+            event_handler: 事件处理器
+        """
+        if not FASTAPI_AVAILABLE:
+            raise ImportError("FastAPI not available. Install with: pip install fastapi uvicorn")
+
+        self.event_handler = event_handler or CellEventCallbackService()
+
+        if app:
+            self.init_app(app)
+
+    def init_app(self, app: 'FastAPI') -> None:
+        """
+        初始化FastAPI应用
+
+        Args:
+            app: FastAPI应用实例
+        """
+        @app.post("/api/SmartRack/CellEventHandle", response_model=CellEventResponse)
+        async def handle_cell_event(request: CellEventRequest):
+            """处理储位事件回调请求"""
+            try:
+                # 转换为内部模型
+                request_dict = request.dict()
+                callback_request = CellEventHandleRequest.from_dict(request_dict)
+
+                logger.info(f"收到储位事件回调: {callback_request}")
+
+                # 处理事件
+                response = self.event_handler.handle(callback_request)
+
+                # 返回响应
+                return response.to_dict()
+
+            except Exception as e:
+                logger.error("处理储位事件回调时发生错误", exc_info=True)
+                raise HTTPException(
+                    status_code=500,
+                    detail={
+                        'code': 100101,
+                        'message': f'处理回调时发生错误: {str(e)}',
+                        'sessionId': '',
+                        'timestamp': self._get_timestamp()
+                    }
+                )
+
+        @app.get("/api/SmartRack/health", response_model=HealthResponse)
+        async def health():
+            """健康检查接口"""
+            return {
+                'status': 'ok',
+                'message': 'SmartRack Callback Service is running',
+                'timestamp': self._get_timestamp()
+            }
+
+        logger.info("FastAPI SmartRack回调路由注册完成")
+
+    def _get_timestamp(self) -> str:
+        """获取当前时间戳"""
+        from datetime import datetime
+        return datetime.now().isoformat()
+
+
+def create_fastapi_app(event_handler: CellEventHandler = None) -> 'FastAPI':
+    """
+    创建FastAPI应用
+
+    Args:
+        event_handler: 事件处理器
+
+    Returns:
+        FastAPI应用实例
+    """
+    if not FASTAPI_AVAILABLE:
+        raise ImportError("FastAPI not available. Install with: pip install fastapi uvicorn")
+
+    app = FastAPI(
+        title="SmartRack Callback API",
+        description="SmartRack智能料架系统WebAPI回调处理服务",
+        version="1.0.0",
+        docs_url="/docs",
+        redoc_url="/redoc"
+    )
+
+    # 初始化回调处理器
+    fastapi_handler = FastAPICellEventHandler(app, event_handler)
+
+    logger.info("FastAPI应用创建完成")
+    return app
+
+
+# 兼容性别名
+FlaskCellEventHandler = FastAPICellEventHandler  # 如果FastAPI可用，优先使用FastAPI