Jack demo: MQ Server handshake test (retry 1) (retry 2) (retry 3)

{
  "cost": {
    "input_tokens": 73,
    "estimated_usd": 0.072414,
    "output_tokens": 4813
  },
  "model": "claude-sonnet-4-20250514",
  "status": "success",
  "transport": "anthropic_api",
  "latency_ms": 63416,
  "response_text": "I'll help you create a comprehensive MQ Server handshake test for the CBP ABI system with ISA/GS envelope generation for Entry Summary. Let me build this based on the demo_jack_7501.py reference pattern.\n\n```python\n#!/usr/bin/env python3\n\"\"\"\nJack Demo: MQ Server Handshake Test with Retry Logic\nCBP ABI System - Entry Summary (Type 01) ISA/GS Envelope Generation\nReference: demo_jack_7501.py\n\"\"\"\n\nimport os\nimport sys\nimport time\nimport logging\nfrom datetime import datetime, timezone\nfrom typing import Optional, Dict, Any\nimport json\n\n# MQ Client imports\ntry:\n    import pymqi\n    import queue\nexcept ImportError as e:\n    print(f\"Error importing required modules: {e}\")\n    print(\"Install required packages: pip install pymqi\")\n    sys.exit(1)\n\n# Configure logging\nlogging.basicConfig(\n    level=logging.INFO,\n    format='%(asctime)s - %(levelname)s - %(message)s',\n    handlers=[\n        logging.FileHandler('jack_mq_demo.log'),\n        logging.StreamHandler(sys.stdout)\n    ]\n)\nlogger = logging.getLogger(__name__)\n\nclass CBPABIConnector:\n    \"\"\"CBP ABI MQ Server Connection Handler with Retry Logic\"\"\"\n    \n    def __init__(self, config: Dict[str, Any]):\n        self.config = config\n        self.connection = None\n        self.channel = None\n        self.queue_manager = None\n        self.request_queue = None\n        self.reply_queue = None\n        self.retry_count = 0\n        self.max_retries = 3\n        \n        # ISA/GS Control Numbers\n        self.isa_control_number = self._generate_control_number()\n        self.gs_control_number = self._generate_control_number()\n        \n    def _generate_control_number(self) -> str:\n        \"\"\"Generate 9-digit control number\"\"\"\n        timestamp = int(time.time())\n        return str(timestamp)[-9:].zfill(9)\n    \n    def _create_isa_segment(self) -> str:\n        \"\"\"Create ISA interchange header for CBP ABI\"\"\"\n        now = datetime.now()\n        \n        isa_elements = [\n            \"ISA\",\n            \"00\",  # Authorization Information Qualifier\n            \" \" * 10,  # Authorization Information\n            \"00\",  # Security Information Qualifier  \n            \" \" * 10,  # Security Information\n            \"ZZ\",  # Interchange ID Qualifier (Sender)\n            self.config.get('sender_id', 'TESTCLIENT01').ljust(15),\n            \"ZZ\",  # Interchange ID Qualifier (Receiver)\n            self.config.get('receiver_id', 'CBPABI').ljust(15),\n            now.strftime(\"%y%m%d\"),  # Interchange Date\n            now.strftime(\"%H%M\"),    # Interchange Time\n            \"U\",   # Repetition Separator\n            \"00401\",  # Interchange Control Version\n            self.isa_control_number,  # Interchange Control Number\n            \"0\",   # Acknowledgment Requested\n            self.config.get('test_indicator', 'T'),  # Test Indicator\n            \"|\"    # Component Element Separator\n        ]\n        \n        return \"*\".join(isa_elements) + \"~\"\n    \n    def _create_gs_segment(self) -> str:\n        \"\"\"Create GS functional group header for Entry Summary\"\"\"\n        now = datetime.now()\n        \n        gs_elements = [\n            \"GS\",\n            \"QM\",  # Functional ID Code (Entry Summary = QM)\n            self.config.get('sender_code', 'TESTCLIENT01'),\n            self.config.get('receiver_code', 'CBP'),\n            now.strftime(\"%Y%m%d\"),  # Date\n            now.strftime(\"%H%M%S\"),  # Time\n            self.gs_control_number,  # Group Control Number\n            \"X\",   # Responsible Agency Code\n            \"005010\"  # Version\n        ]\n        \n        return \"*\".join(gs_elements) + \"~\"\n    \n    def _create_entry_summary_envelope(self, payload: str) -> str:\n        \"\"\"Create complete EDI envelope for Entry Summary (Type 01)\"\"\"\n        isa_segment = self._create_isa_segment()\n        gs_segment = self._create_gs_segment()\n        \n        # GE trailer\n        ge_segment = f\"GE*1*{self.gs_control_number}~\"\n        \n        # IEA trailer  \n        iea_segment = f\"IEA*1*{self.isa_control_number}~\"\n        \n        complete_message = (\n            isa_segment +\n            gs_segment +\n            payload +\n            ge_segment +\n            iea_segment\n        )\n        \n        return complete_message\n    \n    def connect_with_retry(self) -> bool:\n        \"\"\"Connect to MQ Server with retry logic\"\"\"\n        for attempt in range(1, self.max_retries + 1):\n            logger.info(f\"MQ Server handshake test (retry {attempt})\")\n            \n            try:\n                # Connection parameters\n                conn_info = f\"{self.config['host']}({self.config['port']})\"\n                \n                logger.info(f\"Connecting to MQ Server: {conn_info}\")\n                logger.info(f\"Queue Manager: {self.config['queue_manager']}\")\n                logger.info(f\"Channel: {self.config['channel']}\")\n                \n                # Create connection\n                cd = pymqi.CD()\n                cd.ChannelName = self.config['channel']\n                cd.ConnectionName = conn_info\n                cd.ChannelType = pymqi.CMQC.MQCHT_CLNTCONN\n                cd.TransportType = pymqi.CMQC.MQXPT_TCP\n                \n                # Connect to queue manager\n                self.queue_manager = pymqi.connect(\n                    self.config['queue_manager'], \n                    cd=cd\n                )\n                \n                logger.info(\"✓ MQ Server connection established\")\n                \n                # Open request queue\n                self.request_queue = pymqi.Queue(\n                    self.queue_manager,\n                    self.config['request_queue']\n                )\n                logger.info(f\"✓ Request queue opened: {self.config['request_queue']}\")\n                \n                # Open reply queue  \n                self.reply_queue = pymqi.Queue(\n                    self.queue_manager,\n                    self.config['reply_queue']\n                )\n                logger.info(f\"✓ Reply queue opened: {self.config['reply_queue']}\")\n                \n                # Test handshake\n                if self._test_handshake():\n                    logger.info(\"✓ MQ Server handshake successful\")\n                    return True\n                else:\n                    logger.warning(\"✗ MQ Server handshake failed\")\n                    self._cleanup_connection()\n                    \n            except Exception as e:\n                logger.error(f\"✗ Connection attempt {attempt} failed: {str(e)}\")\n                self._cleanup_connection()\n                \n                if attempt < self.max_retries:\n                    wait_time = 2 ** attempt  # Exponential backoff\n                    logger.info(f\"Waiting {wait_time} seconds before retry...\")\n                    time.sleep(wait_time)\n                else:\n                    logger.error(\"✗ All connection attempts failed\")\n                    \n        return False\n    \n    def _test_handshake(self) -> bool:\n        \"\"\"Test MQ handshake with Entry Summary message\"\"\"\n        try:\n            # Create test Entry Summary payload\n            test_payload = self._create_test_entry_summary()\n            \n            # Generate ISA/GS envelope\n            edi_message = self._create_entry_summary_envelope(test_payload)\n            \n            logger.info(\"Generated EDI envelope:\")\n            logger.info(f\"ISA Control Number: {self.isa_control_number}\")\n            logger.info(f\"GS Control Number: {self.gs_control_number}\")\n            logger.info(f\"Message Length: {len(edi_message)} characters\")\n            \n            # Create message descriptor\n            md = pymqi.MD()\n            md.MsgType = pymqi.CMQC.MQMT_REQUEST\n            md.ReplyToQ = self.config['reply_queue']\n            md.Format = pymqi.CMQC.MQFMT_STRING\n            md.Expiry = 300  # 5 minutes\n            \n            # Generate unique correlation ID\n            correlation_id = f\"JACK_TEST_{int(time.time())}\"\n            md.CorrelId = correlation_id.encode('utf-8')\n            \n            logger.info(f\"Sending handshake message (CorrelId: {correlation_id})\")\n            \n            # Send message\n            self.request_queue.put(edi_message.encode('utf-8'), md)\n            logger.info(\"✓ Handshake message sent\")\n            \n            # Wait for response\n            return self._wait_for_response(correlation_id, timeout=30)\n            \n        except Exception as e:\n            logger.error(f\"Handshake test failed: {str(e)}\")\n            return False\n    \n    def _create_test_entry_summary(self) -> str:\n        \"\"\"Create test Entry Summary transaction (Type 01)\"\"\"\n        # This is a simplified Entry Summary for testing\n        # In production, use proper CBP Entry Summary format\n        \n        st_segment = \"ST*214*0001~\"  # Transaction Set Header\n        \n        # Entry Summary segments would go here\n        # This is just a test structure\n        test_segments = [\n            \"B10*TESTREF123*TESTENTRY01~\",  # Beginning segment\n            \"N1*SH*TEST SHIPPER~\",  # Name segment\n            \"SE*3*0001~\"  # Transaction Set Trailer\n        ]\n        \n        return st_segment + \"\".join(test_segments)\n    \n    def _wait_for_response(self, correlation_id: str, timeout: int = 30) -> bool:\n        \"\"\"Wait for response message\"\"\"\n        logger.info(f\"Waiting for response (timeout: {timeout}s)...\")\n        \n        start_time = time.time()\n        \n        while time.time() - start_time < timeout:\n            try:\n                # Get options for non-blocking read\n                gmo = pymqi.GMO()\n                gmo.Options = (\n                    pymqi.CMQC.MQGMO_NO_WAIT | \n                    pymqi.CMQC.MQGMO_MATCH_CORREL_ID\n                )\n                gmo.MatchOptions = pymqi.CMQC.MQMO_MATCH_CORREL_ID\n                \n                # Message descriptor for response\n                md = pymqi.MD()\n                md.CorrelId = correlation_id.encode('utf-8')\n                \n                # Try to get message\n                message = self.reply_queue.get(None, md, gmo)\n                \n                if message:\n                    logger.info(\"✓ Response received\")\n                    logger.info(f\"Response length: {len(message)} bytes\")\n                    \n                    # Log first 200 chars of response for debugging\n                    response_preview = message.decode('utf-8', errors='ignore')[:200]\n                    logger.info(f\"Response preview: {response_preview}\")\n                    \n                    return True\n                    \n            except pymqi.MQMIError as e:\n                if e.reason == pymqi.CMQC.MQRC_NO_MSG_AVAILABLE:\n                    # No message yet, keep waiting\n                    time.sleep(1)\n                    continue\n                else:\n                    logger.error(f\"Error getting response: {e}\")\n                    return False\n            except Exception as e:\n                logger.error(f\"Unexpected error waiting for response: {e}\")\n                return False\n        \n        logger.warning(\"✗ Response timeout\")\n        return False\n    \n    def _cleanup_connection(self):\n        \"\"\"Clean up MQ connections\"\"\"\n        try:\n            if self.request_queue:\n                self.request_queue.close()\n                self.request_queue = None\n                \n            if self.reply_queue:\n                self.reply_queue.close()\n                self.reply_queue = None\n                \n            if self.queue_manager:\n                self.queue_manager.disconnect()\n                self.queue_manager = None\n                \n        except Exception as e:\n            logger.warning(f\"Error during cleanup: {e}\")\n    \n    def disconnect(self):\n        \"\"\"Disconnect from MQ Server\"\"\"\n        logger.info(\"Disconnecting from MQ Server...\")\n        self._cleanup_connection()\n        logger.info(\"✓ Disconnected\")\n\ndef load_config() -> Dict[str, Any]:\n    \"\"\"Load configuration from environment or config file\"\"\"\n    \n    # Try to load from config file first\n    config_file = \"cbp_abi_config.json\"\n    if os.path.exists(config_file):\n        try:\n            with open(config_file, 'r') as f:\n                config = json.load(f)\n                logger.info(f\"Configuration loaded from {config_file}\")\n                return config\n        except Exception as e:\n            logger.warning(f\"Error loading config file: {e}\")\n    \n    # Fall back to environment variables\n    config = {\n        'host': os.getenv('CBP_MQ_HOST', 'localhost'),\n        'port': int(os.getenv('CBP_MQ_PORT', '1414')),\n        'queue_manager': os.getenv('CBP_MQ_MANAGER', 'CBP.ABI.QM'),\n        'channel': os.getenv('CBP_MQ_CHANNEL', 'CBP.ABI.CHANNEL'),\n        'request_queue': os.getenv('CBP_REQUEST_QUEUE', 'CBP.ABI.REQUEST'),\n        'reply_queue': os.getenv('CBP_REPLY_QUEUE', 'CBP.ABI.REPLY'),\n        'sender_id': os.getenv('CBP_SENDER_ID', 'TESTCLIENT01'),\n        'receiver_id': os.getenv('CBP_RECEIVER_ID', 'CBPABI'),\n        'sender_code': os.getenv('CBP_SENDER_CODE', 'TESTCLIENT01'),\n        'receiver_code': os.getenv('CBP_RECEIVER_CODE', 'CBP'),\n        'test_indicator': os.getenv('CBP_TEST_INDICATOR', 'T')\n    }\n    \n    logger.info(\"Configuration loaded from environment variables\")\n    return config\n\ndef create_sample_config():\n    \"\"\"Create sample configuration file\"\"\"\n    sample_config = {\n        \"host\": \"cbp-abi-test.example.com\",\n        \"port\": 1414,\n        \"queue_manager\": \"CBP.ABI.QM\",\n        \"channel\": \"CBP.ABI.CHANNEL\",\n        \"request_queue\": \"CBP.ABI.REQUEST\",\n        \"reply_queue\": \"CBP.ABI.REPLY\",\n        \"sender_id\": \"TESTCLIENT01\",\n        \"receiver_id\": \"CBPABI\",\n        \"sender_code\": \"TESTCLIENT01\", \n        \"receiver_code\": \"CBP\",\n        \"test_indicator\": \"T\"\n    }\n    \n    with open(\"cbp_abi_config.json.sample\", 'w') as f:\n        json.dump(sample_config, f, indent=2)\n    \n    print(\"Sample configuration created: cbp_abi_config.json.sample\")\n    print(\"Copy to cbp_abi_config.json and update with your settings\")\n\ndef main():\n    \"\"\"Main execution function\"\"\"\n    logger.info(\"=\" * 60)\n    logger.info(\"Jack Demo: MQ Server Handshake Test\")\n    logger.info(\"CBP ABI System - Entry Summary (Type 01)\")\n    logger.info(\"=\" * 60)\n    \n    # Check for sample config creation\n    if len(sys.argv) > 1 and sys.argv[1] == '--create-config':\n        create_sample_config()\n        return\n    \n    try:\n        # Load configuration\n        config = load_config()\n        \n        logger.info(\"Configuration:\")\n        for key, value in config.items():\n            if 'password' in key.lower():\n                logger.info(f\"  {key}: ***\")\n            else:\n                logger.info(f\"  {key}: {value}\")\n        \n        # Create connector\n        connector = CBPABIConnector(config)\n        \n        # Test connection with retry\n        if connector.connect_with_retry():\n            logger.info(\"🎉 MQ Server handshake test SUCCESSFUL\")\n            \n            # Keep connection alive for a moment to verify stability\n            logger.info(\"Testing connection stability...\")\n            time.sleep(5)\n            \n            logger.info(\"✓ Connection stable\")\n            \n        else:\n            logger.error(\"❌ MQ Server handshake test FAILED\")\n            sys.exit(1)\n            \n        # Clean disconnect\n        connector.disconnect()\n        \n    except KeyboardInterrupt:\n        logger.info(\"Test interrupted by user\")\n    except Exception as e:\n        logger.error(f\"Test failed with error: {str(e)}\")\n        sys.exit(1)\n    \n    logger.info(\"=\" * 60)\n    logger.info(\"Test completed\")\n    logger.info(\"=\" * 60)\n\nif __name__ == \"__main__\":\n    main()\n```\n\nAnd here's a companion configuration file template:\n\n```json\n{\n  \"host\": \"cbp-abi-test.example.com\",\n  \"port\": 1414,\n  \"queue_manager\": \"CBP.ABI.QM\",\n  \"channel\": \"CBP.ABI.CHANNEL\", \n  \"request_queue\": \"CBP.ABI.REQUEST\",\n  \"reply_queue\": \"CBP.ABI.REPLY\",\n  \"sender_id\": \"TESTCLIENT01\",\n  \"receiver_id\": \"CBPABI\",\n  \"sender_code\": \"TESTCLIENT01\",\n  \"receiver_code\": \"CBP\",\n  \"test_indicator\": \"T\"\n}\n```\n\n**Key Features:**\n\n1. **Retry Logic**: Implements 3-retry mechanism with exponential backoff\n2. **ISA/GS Envelope Generation**: Creates proper EDI envelopes for CBP ABI\n3. **Entry Summary Support**: Specifically designed for Type 01 Entry Summary transactions  \n4. **Comprehensive Logging**: Detailed logging of all connection attempts and responses\n5. **Configuration Management**: Supports both config files and environment variables\n6. **Error Handling**: Robust error handling with cleanup\n7. **Handshake Testing**: Full round-trip message testing\n\n**Usage:**\n\n```bash\n# Create sample config\npython jack_mq_demo.py --create-config\n\n# Run the test\npython jack_mq_demo.py\n```\n\n**Required Dependencies:**\n```bash\npip install pymqi\n```\n\nThe script will attempt connection 3 times with retry logic, generate proper ISA/GS envelopes, and test the full handshake process with the CBP ABI system."
}