@asyncapi/generator 3.0.1 → 3.1.1

package/CHANGELOG.md

@@ -1,5 +1,77 @@
 # @asyncapi/generator
 
+## 3.1.1
+
+### Patch Changes
+
+- 2bfad27: Fix generator handling of template parameters with `false` default values, ensuring defaults are correctly injected and conditional generation works as expected.
+
+## 3.1.0
+
+### Minor Changes
+
+- 11a1b8d: - **Updated Component**: `OnMessage` (Python) - Added discriminator-based routing logic that automatically dispatches messages to operation-specific handlers before falling back to generic handlers
+
+  - **New Helpers**:
+    - `getMessageDiscriminatorData` - Extracts discriminator key and value from individual messages
+    - `getMessageDiscriminatorsFromOperations` - Collects all discriminator metadata from receive operations
+  - Enhanced Python webSocket client generation with **automatic operation-based message routing**:
+
+  ## How python routing works
+
+  - Generated WebSocket clients now automatically route incoming messages to operation-specific handlers based on message discriminators. Users can register handlers for specific message types without manually parsing or filtering messages.
+  - When a message arrives, the client checks it against registered discriminators (e.g., `type: "hello"`, `type: "events_api"`)
+  - If a match is found, the message is routed to the specific operation handler (e.g., `onHelloMessage`, `onEvent`)
+  - If no match is found, the message falls back to generic message handlers
+  - This enables clean separation of message handling logic based on message types
+
+  > `discriminator` is a `string` field that you can add to any AsyncAPI Schema. This also means that it is limited to AsyncAPI Schema only, and it won't work with other schema formats, like for example, Avro.
+
+  The implementation automatically derives discriminator information from your AsyncAPI document:
+
+  - Discriminator `key` is extracted from the `discriminator` field in your AsyncAPI spec
+  - Discriminator `value` is extracted from the `const` property defined in message schemas
+
+  Example AsyncAPI Schema with `discriminator` and `const`:
+
+  ```yaml
+  schemas:
+    hello:
+      type: object
+      discriminator: type # you specify name of property
+      properties:
+        type:
+          type: string
+          const: hello # you specify the value of the discriminator property that is used for routing
+          description: A hello string confirming WebSocket connection
+  ```
+
+  ## Fallback
+
+  When defaults aren't available in the AsyncAPI document, users must provide **both** `discriminator_key` and `discriminator_value` when registering handlers. Providing only one parameter is not supported - you must provide either both or neither.
+
+  > **Why this limitation exists**: When a receive operation has multiple messages sharing the same discriminator key (e.g., all use `"type"` field), we need the specific value (e.g., `"hello"`, `"disconnect"`) to distinguish between them. Without both pieces of information, the routing becomes ambiguous.
+
+  Example:
+
+  ```python
+  # Default case - discriminator info auto-derived from AsyncAPI doc
+  client.register_on_hello_message_handler(my_handler)
+
+  # Custom case - must provide both key AND value
+  client.register_on_hello_message_handler(
+      my_handler,
+      discriminator_key="message_type",
+      discriminator_value="custom_hello"
+  )
+  ```
+
+### Patch Changes
+
+- Updated dependencies [11a1b8d]
+  - @asyncapi/generator-components@0.5.0
+  - @asyncapi/generator-helpers@1.1.0
+
 ## 3.0.1
 
 ### Patch Changes

package/docs/configuration-file.md

@@ -159,7 +159,7 @@ The `generator` property from `package.json` file must contain a JSON object and
 | `conditionalFiles[filePath].validation` | Object | The `validation` is a JSON Schema Draft 07 object. This JSON Schema definition will be applied to the JSON value resulting from the `subject` query. If validation doesn't have errors, the condition is met, and therefore the given file will be rendered. Otherwise, the file is ignored. Check [JSON Schema Validation](https://json-schema.org/draft-07/json-schema-validation.html#rfc.section.6) for a list of all possible validation keywords. **Note: It is deprecated and will be removed with future releases. Use `conditionalGeneration` instead.** |
 | `conditionalGeneration` | Object[String, `{ subject?: String, parameter?: String, validation: Object }`] | An object containing all the file paths or directory names that should be conditionally rendered. Each key represents a file path or directory name and each value must be an object with the keys `subject`, `parameter` and `validation`. You can use either subject or parameter according to the use case. The path should be relative to the `template` directory inside the template. **Note: conditionalGeneration and conditionalFile are mutually exclusive, which means both cannot be configured at the same time in the template**. |
 | `conditionalGeneration[filePath/directoryName].subject` | String | The `subject` is a [JMESPath](http://jmespath.org/) query to grab the value you want to apply the condition to. It queries an object with the whole AsyncAPI document and, when specified, the given server. The object looks like this: `{ asyncapi: { ... }, server: { ... } }`. If the template supports the `server` parameter, you can access server details like, for example, protocol this way: `server.protocol`. During validation with `conditionalGeneration`, only the server that the template user selected is available in the specification file. For more information about the `server` parameter [read about special parameters](#special-parameters). |
-| `conditionalGeneration[filePath/directoryName].parameter` | String | The `parameter` is the name of a custom template parameter passed through `templateParams` that controls whether a specific file or folder should be included in the generated output. You must define a `validation` rule using a JSON Schema fragment to apply the condition. For example, if you define `"parameter": "includeDocs"` with `"validation": { "const": true }`, the corresponding folder (e.g., `docs/`) will only be generated when the user passes `{ includeDocs: true }`. If `includeDocs` is `false`, it will be skipped. |
+| `conditionalGeneration[filePath/directoryName].parameter` | String | The `parameter` is the name of a custom template parameter passed through `templateParams` that controls whether a specific file or folder should be included in the generated output. You must define a `validation` rule using a JSON Schema fragment to apply the condition. For example, if you define `"parameter": "includeDocs"` with `"validation": { "const": true }`, the corresponding folder (e.g., `docs/`) will only be generated when the user passes `{ includeDocs: true }`. If `includeDocs` is `false`, it will be skipped. When you rely on a parameter here, make sure its default value declared in the `parameters` section uses the same data type your validation expects; mismatched types lead to failing conditions even when the defaults should allow generation. |
 | `conditionalGeneration[filePath/directoryName].validation` | Object (JSON Schema fragment) | The validation defines the condition under which the file or directory will be generated. It must be a valid JSON Schema fragment that validates the value of the parameter. For example, if you want to include a folder only when includeDocs is true, use "validation": `{ "const": true }`. You can also use more complex validation logic, like "enum": ["yes", "true"] or "type": "string" with a "pattern" constraint. If the parameter fails validation, the file or folder will not be included in the generated output. This allows for powerful and flexible control over template generation based on user input. |
 | `nonRenderableFiles` | [String] | A list of file paths or [globs](https://en.wikipedia.org/wiki/Glob_(programming)) that must be copied "as-is" to the target directory, i.e., without performing any rendering process. This is useful when you want to copy binary files. |
 | `generator` | [String] | A string representing the generator version-range the template is compatible with. This value must follow the [semver](https://nodejs.dev/learn/semantic-versioning-using-npm) syntax. E.g., `>=1.0.0`, `>=1.0.0 <=2.0.0`, `~1.0.0`, `^1.0.0`, `1.0.0`, etc. [Read more about semver](https://docs.npmjs.com/about-semantic-versioning). |

package/docs/template.md

@@ -29,7 +29,7 @@ Baked-in templates benefit from:
 
 In contrast, **standalone templates** (described below) are maintained as independent Node.js packages, may live in separate repositories, and can be managed or installed separately from the core generator.
 
-Learn more from document [Baked-in Templates](#baked-in-templates).
+Learn more from document [Baked-in Templates](baked-in-templates).
 
 ### Standalone templates
 

package/lib/templates/bakedInTemplates/core-template-client-kafka-java-quarkus/package.json

@@ -18,7 +18,7 @@
     "lint:fix": "eslint --fix --max-warnings 0 --config ../../../../../../.eslintrc --ignore-path ../../../../../../.eslintignore ."
   },
   "dependencies": {
-    "@asyncapi/generator-helpers": "1.0.1",
+    "@asyncapi/generator-helpers": "1.1.0",
     "@asyncapi/generator-react-sdk": "^1.1.2"
   },
   "devDependencies": {

package/lib/templates/bakedInTemplates/core-template-client-websocket-dart/package.json

@@ -12,8 +12,8 @@
   "license": "Apache-2.0",
   "dependencies": {
     "@asyncapi/generator-react-sdk": "*",
-    "@asyncapi/generator-helpers": "1.0.1",
-    "@asyncapi/generator-components": "0.4.1"
+    "@asyncapi/generator-helpers": "1.1.0",
+    "@asyncapi/generator-components": "0.5.0"
   },
   "devDependencies": {
     "@asyncapi/parser": "^3.4.0",

package/lib/templates/bakedInTemplates/core-template-client-websocket-java-quarkus/package.json

@@ -18,8 +18,8 @@
     "lint:fix": "eslint --fix --max-warnings 0 --config ../../../../../../.eslintrc --ignore-path ../../../../../../.eslintignore ."
   },
   "dependencies": {
-    "@asyncapi/generator-helpers": "1.0.1",
-    "@asyncapi/generator-components": "0.4.1",
+    "@asyncapi/generator-helpers": "1.1.0",
+    "@asyncapi/generator-components": "0.5.0",
     "@asyncapi/generator-react-sdk": "^1.1.2"
   },
   "devDependencies": {

package/lib/templates/bakedInTemplates/core-template-client-websocket-javascript/package.json

@@ -11,9 +11,9 @@
   "author": "Lukasz Gornicki <lpgornicki@gmail.com>",
   "license": "Apache-2.0",
   "dependencies": {
-    "@asyncapi/generator-helpers": "1.0.1",
+    "@asyncapi/generator-helpers": "1.1.0",
     "@asyncapi/generator-react-sdk": "*",
-    "@asyncapi/generator-components": "0.4.1",
+    "@asyncapi/generator-components": "0.5.0",
     "@asyncapi/keeper": "0.5.0"
   },
   "devDependencies": {

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/README.md

@@ -66,3 +66,69 @@ You can use our AsyncAPI's credentials to access different set of events produce
     {"type":"hello","num_connections":1,"debug_info":{"host":"applink-3","build_number":118,"approximate_connection_time":18060},"connection_info":{"app_id":"A08NKKBFGBD"}}
     ```
     If you did not receive it, you probably connect with wrong credentials. Remember that generated `ticket` can be used only once to establish a websocket connection.
+
+## Client for Slack with Auto-Routing
+
+To run the Slack Client example with auto-routing, first follow the "Client for Slack" setup above (generate credentials and install dependencies), then:
+- Use `example-slack-with-routing.py` instead of `example-slack.py`.
+- This example demonstrates auto-routing of messages to registered handlers for different event types (hello, event, disconnect, and unrecognized messages). The client automatically dispatches incoming messages based on their type without manual parsing.
+
+1. Start the example that uses generated client with auto-routing. Examine events, and modify example as you want:
+    
+    Linux/MacOs
+    ```bash
+    TICKET=6b150bb1-82b4-457f-a09d-6ff0af1fd2d1 APP_ID=00dfdcccb53a2645dd3f1773fcb10fa7b0a598cf333a990a9db12375ef1865dd python example-slack-with-routing.py
+    ```
+    Windows
+    ```powershell
+    $env:TICKET="dcaa9dc7-b728-40dd-ac40-16dd5f2f8710"; $env:APP_ID="00dfdcccb53a2645dd3f1773fcb10fa7b0a598cf333a990a9db12375ef1865dd"; python example-slack-with-routing.py
+    ```
+
+1. By default, `discriminator_key` is derived from the discriminator field, and `discriminator_value` from the corresponding `const` in the AsyncAPI document. For non-default cases, users must provide both `discriminator_key` and `discriminator_value` explicitly in register_handlers. Partial inputs are not supported to avoid ambiguity in message routing.
+
+### How Routing Works
+
+- Generated WebSocket clients now automatically route incoming messages to operation-specific handlers based on message discriminators. Users can register handlers for specific message types without manually parsing or filtering messages.
+- When a message arrives, the client checks it against registered discriminators (e.g., `type: "hello"`, `type: "events_api"`)
+- If a match is found, the message is routed to the specific operation handler (e.g., `onHelloMessage`, `onEvent`)
+- If no match is found, the message falls back to generic message handlers
+- This enables clean separation of message handling logic based on message types
+
+> `discriminator` is a `string` field that you can add to any AsyncAPI Schema. This also means that it is limited to AsyncAPI Schema only, and it won't work with other schema formats, like for example, Avro. 
+
+The implementation automatically derives discriminator information from your AsyncAPI document:
+- Discriminator `key` is extracted from the `discriminator` field in your AsyncAPI spec
+- Discriminator `value` is extracted from the `const` property defined in message schemas
+
+Example AsyncAPI Schema with `discriminator` and `const`:
+```yaml
+schemas:
+    hello:
+      type: object
+      discriminator: type # you specify name of property
+      properties:
+        type:
+          type: string
+          const: hello # you specify the value of the discriminator property that is used for routing
+          description: A hello string confirming WebSocket connection
+```
+
+### Fallback
+
+When defaults aren't available in the AsyncAPI document, users must provide **both** `discriminator_key` and `discriminator_value` when registering handlers. Providing only one parameter is not supported - you must provide either both or neither.
+
+> **Why this limitation exists**: When a receive operation has multiple messages sharing the same discriminator key (e.g., all use `"type"` field), we need the specific value (e.g., `"hello"`, `"disconnect"`) to distinguish between them. Without both pieces of information, the routing becomes ambiguous.
+
+Example:
+
+```python
+# Default case - discriminator info auto-derived from AsyncAPI doc
+client.register_on_hello_message_handler(my_handler)
+
+# Custom case - must provide both key AND value
+client.register_on_hello_message_handler(
+    my_handler,
+    discriminator_key="message_type",
+    discriminator_value="custom_hello"
+)
+```

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/ClientClass.js

@@ -10,9 +10,11 @@ var Constructor = require('./Constructor.js');
 var generatorComponents = require('@asyncapi/generator-components');
 var RegisterOutgoingProcessor = require('./RegisterOutgoingProcessor.js');
 var HandleError = require('./HandleError.js');
+var RegisterReceiveOperations = require('./RegisterReceiveOperations.js');
 var jsxRuntime = require('/home/runner/work/generator/generator/node_modules/react/cjs/react-jsx-runtime.production.min.js');
 require('./QueryParamsArgumentsDocs.js');
 require('./InitSignature.js');
+require('./ReceiveOperationsDiscriminators.js');
 
 function ClientClass({
   asyncapi,
@@ -25,13 +27,15 @@ function ClientClass({
   const serverUrl = generatorHelpers.getServerUrl(server);
   const operations = asyncapi.operations();
   const sendOperations = operations.filterBySend();
+  const receiveOperations = operations.filterByReceive();
   return /*#__PURE__*/jsxRuntime.jsxs(generatorReactSdk.Text, {
     children: [/*#__PURE__*/jsxRuntime.jsx(generatorReactSdk.Text, {
       newLines: 2,
       children: `class ${clientName}:`
     }), /*#__PURE__*/jsxRuntime.jsx(Constructor.Constructor, {
       serverUrl: serverUrl,
-      query: queryParams
+      query: queryParams,
+      receiveOperations: receiveOperations
     }), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.Connect, {
       language: "python",
       title: title
@@ -50,7 +54,9 @@ function ClientClass({
       methodName: "handle_message",
       methodParams: ['self', 'message'],
       preExecutionCode: "\"\"\"Pass the incoming message to all registered message handlers. \"\"\""
-    }), /*#__PURE__*/jsxRuntime.jsx(HandleError.HandleError, {}), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.SendOperations, {
+    }), /*#__PURE__*/jsxRuntime.jsx(HandleError.HandleError, {}), /*#__PURE__*/jsxRuntime.jsx(RegisterReceiveOperations.RegisterReceiveOperations, {
+      receiveOperations: receiveOperations
+    }), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.SendOperations, {
       language: "python",
       sendOperations: sendOperations,
       clientName: clientName

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/ClientClass.js.map

@@ -1 +1 @@
-{"version":3,"file":"ClientClass.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/ClientClass.js"],"sourcesContent":["import { Text } from '@asyncapi/generator-react-sdk';\nimport { getClientName, getServerUrl, getServer, getQueryParams, getTitle } from '@asyncapi/generator-helpers';\nimport { Send } from './Send';\nimport { Constructor } from './Constructor';\nimport { CloseConnection, RegisterMessageHandler, RegisterErrorHandler, SendOperations, Connect, HandleMessage } from '@asyncapi/generator-components';\nimport { RegisterOutgoingProcessor } from './RegisterOutgoingProcessor';\nimport { HandleError } from './HandleError';\n\nexport function ClientClass({ asyncapi, params }) {\n  const server = getServer(asyncapi.servers(), params.server);\n  const title = getTitle(asyncapi);\n  const queryParams = getQueryParams(asyncapi.channels());\n  const clientName = getClientName(asyncapi, params.appendClientSuffix, params.customClientName);\n  const serverUrl = getServerUrl(server);\n  const operations = asyncapi.operations();\n  const sendOperations = operations.filterBySend();\n  return (\n    <Text>\n      <Text newLines={2}>\n        {`class ${clientName}:`}\n      </Text>\n      <Constructor serverUrl={serverUrl} query={queryParams} />\n      <Connect language=\"python\" title={title} />\n      <RegisterMessageHandler\n        language=\"python\"\n        methodName='register_message_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process incoming messages.\"\"\"'\n      />\n      <RegisterErrorHandler\n        language=\"python\"\n        methodName='register_error_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process errors.\"\"\"'\n      />\n      <RegisterOutgoingProcessor />\n      <HandleMessage\n        language=\"python\"\n        methodName='handle_message'\n        methodParams={['self', 'message']}\n        preExecutionCode='\"\"\"Pass the incoming message to all registered message handlers. \"\"\"'\n      />\n      <HandleError />\n      <SendOperations \n        language=\"python\"\n        sendOperations={sendOperations} \n        clientName={clientName} \n      />\n      <Send sendOperations={sendOperations} />\n      <CloseConnection \n        language=\"python\" \n        methodParams={['self']}\n        preExecutionCode='\"\"\"Cleanly close the WebSocket connection.\"\"\"'\n      />\n    </Text>\n  );\n}\n"],"names":["ClientClass","asyncapi","params","server","getServer","servers","title","getTitle","queryParams","getQueryParams","channels","clientName","getClientName","appendClientSuffix","customClientName","serverUrl","getServerUrl","operations","sendOperations","filterBySend","_jsxs","Text","children","_jsx","newLines","Constructor","query","Connect","language","RegisterMessageHandler","methodName","methodParams","preExecutionCode","RegisterErrorHandler","RegisterOutgoingProcessor","HandleMessage","HandleError","SendOperations","Send","CloseConnection"],"mappings":";;;;;;;;;;;;;;;;AAQO,SAASA,WAAWA,CAAC;EAAEC,QAAQ;AAAEC,EAAAA,MAAAA;AAAO,CAAC,EAAE;AAChD,EAAA,MAAMC,MAAM,GAAGC,0BAAS,CAACH,QAAQ,CAACI,OAAO,EAAE,EAAEH,MAAM,CAACC,MAAM,CAAC,CAAA;AAC3D,EAAA,MAAMG,KAAK,GAAGC,yBAAQ,CAACN,QAAQ,CAAC,CAAA;EAChC,MAAMO,WAAW,GAAGC,+BAAc,CAACR,QAAQ,CAACS,QAAQ,EAAE,CAAC,CAAA;AACvD,EAAA,MAAMC,UAAU,GAAGC,8BAAa,CAACX,QAAQ,EAAEC,MAAM,CAACW,kBAAkB,EAAEX,MAAM,CAACY,gBAAgB,CAAC,CAAA;AAC9F,EAAA,MAAMC,SAAS,GAAGC,6BAAY,CAACb,MAAM,CAAC,CAAA;AACtC,EAAA,MAAMc,UAAU,GAAGhB,QAAQ,CAACgB,UAAU,EAAE,CAAA;AACxC,EAAA,MAAMC,cAAc,GAAGD,UAAU,CAACE,YAAY,EAAE,CAAA;EAChD,oBACEC,eAAA,CAACC,sBAAI,EAAA;IAAAC,QAAA,EAAA,cACHC,cAAA,CAACF,sBAAI,EAAA;AAACG,MAAAA,QAAQ,EAAE,CAAE;MAAAF,QAAA,EACf,SAASX,UAAU,CAAA,CAAA,CAAA;AAAG,KACnB,CAAC,eACPY,cAAA,CAACE,uBAAW,EAAA;AAACV,MAAAA,SAAS,EAAEA,SAAU;AAACW,MAAAA,KAAK,EAAElB,WAAAA;AAAY,KAAE,CAAC,eACzDe,cAAA,CAACI,2BAAO,EAAA;AAACC,MAAAA,QAAQ,EAAC,QAAQ;AAACtB,MAAAA,KAAK,EAAEA,KAAAA;AAAM,KAAE,CAAC,eAC3CiB,cAAA,CAACM,0CAAsB,EAAA;AACrBD,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,UAAU,EAAC,0BAA0B;AACrCC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,+DAAA;AAAyD,KAC3E,CAAC,eACFT,cAAA,CAACU,wCAAoB,EAAA;AACnBL,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,UAAU,EAAC,wBAAwB;AACnCC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,oDAAA;KAClB,CAAC,eACFT,cAAA,CAACW,mDAAyB,IAAE,CAAC,eAC7BX,cAAA,CAACY,iCAAa,EAAA;AACZP,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,UAAU,EAAC,gBAAgB;AAC3BC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,4EAAA;KAClB,CAAC,eACFT,cAAA,CAACa,uBAAW,IAAE,CAAC,eACfb,cAAA,CAACc,kCAAc,EAAA;AACbT,MAAAA,QAAQ,EAAC,QAAQ;AACjBV,MAAAA,cAAc,EAAEA,cAAe;AAC/BP,MAAAA,UAAU,EAAEA,UAAAA;AAAW,KACxB,CAAC,eACFY,cAAA,CAACe,SAAI,EAAA;AAACpB,MAAAA,cAAc,EAAEA,cAAAA;AAAe,KAAE,CAAC,eACxCK,cAAA,CAACgB,mCAAe,EAAA;AACdX,MAAAA,QAAQ,EAAC,QAAQ;MACjBG,YAAY,EAAE,CAAC,MAAM,CAAE;AACvBC,MAAAA,gBAAgB,EAAC,qDAAA;AAA+C,KACjE,CAAC,CAAA;AAAA,GACE,CAAC,CAAA;AAEX;;;;"}
+{"version":3,"file":"ClientClass.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/ClientClass.js"],"sourcesContent":["import { Text } from '@asyncapi/generator-react-sdk';\nimport { getClientName, getServerUrl, getServer, getQueryParams, getTitle } from '@asyncapi/generator-helpers';\nimport { Send } from './Send';\nimport { Constructor } from './Constructor';\nimport { CloseConnection, RegisterMessageHandler, RegisterErrorHandler, SendOperations, Connect, HandleMessage } from '@asyncapi/generator-components';\nimport { RegisterOutgoingProcessor } from './RegisterOutgoingProcessor';\nimport { HandleError } from './HandleError';\nimport { RegisterReceiveOperations } from './RegisterReceiveOperations';\n\nexport function ClientClass({ asyncapi, params }) {\n  const server = getServer(asyncapi.servers(), params.server);\n  const title = getTitle(asyncapi);\n  const queryParams = getQueryParams(asyncapi.channels());\n  const clientName = getClientName(asyncapi, params.appendClientSuffix, params.customClientName);\n  const serverUrl = getServerUrl(server);\n  const operations = asyncapi.operations();\n  const sendOperations = operations.filterBySend();\n  const receiveOperations = operations.filterByReceive();\n\n  return (\n    <Text>\n      <Text newLines={2}>\n        {`class ${clientName}:`}\n      </Text>\n      <Constructor serverUrl={serverUrl} query={queryParams} receiveOperations={receiveOperations} />\n      <Connect language=\"python\" title={title} />\n      <RegisterMessageHandler\n        language=\"python\"\n        methodName='register_message_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process incoming messages.\"\"\"'\n      />\n      <RegisterErrorHandler\n        language=\"python\"\n        methodName='register_error_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process errors.\"\"\"'\n      />\n      <RegisterOutgoingProcessor />\n      <HandleMessage\n        language=\"python\"\n        methodName='handle_message'\n        methodParams={['self', 'message']}\n        preExecutionCode='\"\"\"Pass the incoming message to all registered message handlers. \"\"\"'\n      />\n      <HandleError />\n      <RegisterReceiveOperations receiveOperations={receiveOperations} />\n      <SendOperations \n        language=\"python\"\n        sendOperations={sendOperations} \n        clientName={clientName} \n      />\n      <Send sendOperations={sendOperations} />\n      <CloseConnection \n        language=\"python\" \n        methodParams={['self']}\n        preExecutionCode='\"\"\"Cleanly close the WebSocket connection.\"\"\"'\n      />\n    </Text>\n  );\n}\n"],"names":["ClientClass","asyncapi","params","server","getServer","servers","title","getTitle","queryParams","getQueryParams","channels","clientName","getClientName","appendClientSuffix","customClientName","serverUrl","getServerUrl","operations","sendOperations","filterBySend","receiveOperations","filterByReceive","_jsxs","Text","children","_jsx","newLines","Constructor","query","Connect","language","RegisterMessageHandler","methodName","methodParams","preExecutionCode","RegisterErrorHandler","RegisterOutgoingProcessor","HandleMessage","HandleError","RegisterReceiveOperations","SendOperations","Send","CloseConnection"],"mappings":";;;;;;;;;;;;;;;;;;AASO,SAASA,WAAWA,CAAC;EAAEC,QAAQ;AAAEC,EAAAA,MAAAA;AAAO,CAAC,EAAE;AAChD,EAAA,MAAMC,MAAM,GAAGC,0BAAS,CAACH,QAAQ,CAACI,OAAO,EAAE,EAAEH,MAAM,CAACC,MAAM,CAAC,CAAA;AAC3D,EAAA,MAAMG,KAAK,GAAGC,yBAAQ,CAACN,QAAQ,CAAC,CAAA;EAChC,MAAMO,WAAW,GAAGC,+BAAc,CAACR,QAAQ,CAACS,QAAQ,EAAE,CAAC,CAAA;AACvD,EAAA,MAAMC,UAAU,GAAGC,8BAAa,CAACX,QAAQ,EAAEC,MAAM,CAACW,kBAAkB,EAAEX,MAAM,CAACY,gBAAgB,CAAC,CAAA;AAC9F,EAAA,MAAMC,SAAS,GAAGC,6BAAY,CAACb,MAAM,CAAC,CAAA;AACtC,EAAA,MAAMc,UAAU,GAAGhB,QAAQ,CAACgB,UAAU,EAAE,CAAA;AACxC,EAAA,MAAMC,cAAc,GAAGD,UAAU,CAACE,YAAY,EAAE,CAAA;AAChD,EAAA,MAAMC,iBAAiB,GAAGH,UAAU,CAACI,eAAe,EAAE,CAAA;EAEtD,oBACEC,eAAA,CAACC,sBAAI,EAAA;IAAAC,QAAA,EAAA,cACHC,cAAA,CAACF,sBAAI,EAAA;AAACG,MAAAA,QAAQ,EAAE,CAAE;MAAAF,QAAA,EACf,SAASb,UAAU,CAAA,CAAA,CAAA;AAAG,KACnB,CAAC,eACPc,cAAA,CAACE,uBAAW,EAAA;AAACZ,MAAAA,SAAS,EAAEA,SAAU;AAACa,MAAAA,KAAK,EAAEpB,WAAY;AAACY,MAAAA,iBAAiB,EAAEA,iBAAAA;AAAkB,KAAE,CAAC,eAC/FK,cAAA,CAACI,2BAAO,EAAA;AAACC,MAAAA,QAAQ,EAAC,QAAQ;AAACxB,MAAAA,KAAK,EAAEA,KAAAA;AAAM,KAAE,CAAC,eAC3CmB,cAAA,CAACM,0CAAsB,EAAA;AACrBD,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,UAAU,EAAC,0BAA0B;AACrCC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,+DAAA;AAAyD,KAC3E,CAAC,eACFT,cAAA,CAACU,wCAAoB,EAAA;AACnBL,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,UAAU,EAAC,wBAAwB;AACnCC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,oDAAA;KAClB,CAAC,eACFT,cAAA,CAACW,mDAAyB,IAAE,CAAC,eAC7BX,cAAA,CAACY,iCAAa,EAAA;AACZP,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,UAAU,EAAC,gBAAgB;AAC3BC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,4EAAA;KAClB,CAAC,eACFT,cAAA,CAACa,uBAAW,IAAE,CAAC,eACfb,cAAA,CAACc,mDAAyB,EAAA;AAACnB,MAAAA,iBAAiB,EAAEA,iBAAAA;AAAkB,KAAE,CAAC,eACnEK,cAAA,CAACe,kCAAc,EAAA;AACbV,MAAAA,QAAQ,EAAC,QAAQ;AACjBZ,MAAAA,cAAc,EAAEA,cAAe;AAC/BP,MAAAA,UAAU,EAAEA,UAAAA;AAAW,KACxB,CAAC,eACFc,cAAA,CAACgB,SAAI,EAAA;AAACvB,MAAAA,cAAc,EAAEA,cAAAA;AAAe,KAAE,CAAC,eACxCO,cAAA,CAACiB,mCAAe,EAAA;AACdZ,MAAAA,QAAQ,EAAC,QAAQ;MACjBG,YAAY,EAAE,CAAC,MAAM,CAAE;AACvBC,MAAAA,gBAAgB,EAAC,qDAAA;AAA+C,KACjE,CAAC,CAAA;AAAA,GACE,CAAC,CAAA;AAEX;;;;"}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/Constructor.js

@@ -7,11 +7,14 @@ var generatorReactSdk = require('@asyncapi/generator-react-sdk');
 var generatorComponents = require('@asyncapi/generator-components');
 var QueryParamsArgumentsDocs = require('./QueryParamsArgumentsDocs.js');
 var InitSignature = require('./InitSignature.js');
+var ReceiveOperationsDiscriminators = require('./ReceiveOperationsDiscriminators.js');
 var jsxRuntime = require('/home/runner/work/generator/generator/node_modules/react/cjs/react-jsx-runtime.production.min.js');
+require('@asyncapi/generator-helpers');
 
 function Constructor({
   serverUrl,
-  query
+  query,
+  receiveOperations
 }) {
   const queryParamsArray = query && Array.from(query.entries());
   return /*#__PURE__*/jsxRuntime.jsxs(jsxRuntime.Fragment, {
@@ -38,8 +41,9 @@ function Constructor({
       self.error_handlers = []        # Callables for errors
       self.outgoing_processors = []   # Callables to process outgoing messages
       self._stop_event = threading.Event()
-      
       ${query ? 'params = {}' : ''}`
+    }), /*#__PURE__*/jsxRuntime.jsx(ReceiveOperationsDiscriminators.ReceiveOperationsDiscriminators, {
+      receiveOperations: receiveOperations
     }), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.QueryParamsVariables, {
       language: "python",
       queryParams: queryParamsArray

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/Constructor.js.map

@@ -1 +1 @@
-{"version":3,"file":"Constructor.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/Constructor.js"],"sourcesContent":["import { Text } from '@asyncapi/generator-react-sdk';\nimport { QueryParamsVariables } from '@asyncapi/generator-components';\nimport { QueryParamsArgumentsDocs } from './QueryParamsArgumentsDocs';\nimport { InitSignature } from './InitSignature';\n\nexport function Constructor({ serverUrl, query}) {\n  const queryParamsArray = query && Array.from(query.entries());\n  \n  return (\n    <>\n      <InitSignature queryParams={queryParamsArray} serverUrl={serverUrl} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      Constructor to initialize the WebSocket client.\n\n      Args:\n          url (str, optional): The WebSocket server URL. Use it if the server URL is \n          different from the default one taken from the AsyncAPI document.`}\n      </Text>\n      <QueryParamsArgumentsDocs queryParams={queryParamsArray} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      self.ws_app = None  # Instance of WebSocketApp\n      self.message_handlers = []      # Callables for incoming messages\n      self.error_handlers = []        # Callables for errors\n      self.outgoing_processors = []   # Callables to process outgoing messages\n      self._stop_event = threading.Event()\n      \n      ${ query ? 'params = {}' : ''}`\n        }\n      </Text>\n      <QueryParamsVariables\n        language=\"python\"\n        queryParams={queryParamsArray} \n      />\n      <Text newLines={2}>\n        {`\n        ${query ? 'qs = urlencode(params) if params else \"\"' : ''}\n        ${query ? 'self.url = f\"{url}{f\\'?{qs}\\' if qs else \\'\\'}\"' : 'self.url = url'}`}\n      </Text>\n    </>\n  );\n}"],"names":["Constructor","serverUrl","query","queryParamsArray","Array","from","entries","_jsxs","_Fragment","children","_jsx","InitSignature","queryParams","Text","indent","QueryParamsArgumentsDocs","QueryParamsVariables","language","newLines"],"mappings":";;;;;;;;;;;AAKO,SAASA,WAAWA,CAAC;EAAEC,SAAS;AAAEC,EAAAA,KAAAA;AAAK,CAAC,EAAE;AAC/C,EAAA,MAAMC,gBAAgB,GAAGD,KAAK,IAAIE,KAAK,CAACC,IAAI,CAACH,KAAK,CAACI,OAAO,EAAE,CAAC,CAAA;EAE7D,oBACEC,eAAA,CAAAC,mBAAA,EAAA;IAAAC,QAAA,EAAA,cACEC,cAAA,CAACC,2BAAa,EAAA;AAACC,MAAAA,WAAW,EAAET,gBAAiB;AAACF,MAAAA,SAAS,EAAEA,SAAAA;AAAU,KAAE,CAAC,eACtES,cAAA,CAACG,sBAAI,EAAA;AAACC,MAAAA,MAAM,EAAE,CAAE;AAAAL,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA,0EAAA,CAAA;AAA2E,KAC/D,CAAC,eACPC,cAAA,CAACK,iDAAwB,EAAA;AAACH,MAAAA,WAAW,EAAET,gBAAAA;AAAiB,KAAE,CAAC,eAC3DO,cAAA,CAACG,sBAAI,EAAA;AAACC,MAAAA,MAAM,EAAE,CAAE;AAAAL,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA;AACA;AACA,MAAA,EAASP,KAAK,GAAG,aAAa,GAAG,EAAE,CAAA,CAAA;AAAE,KAEzB,CAAC,eACPQ,cAAA,CAACM,wCAAoB,EAAA;AACnBC,MAAAA,QAAQ,EAAC,QAAQ;AACjBL,MAAAA,WAAW,EAAET,gBAAAA;AAAiB,KAC/B,CAAC,eACFO,cAAA,CAACG,sBAAI,EAAA;AAACK,MAAAA,QAAQ,EAAE,CAAE;AAAAT,MAAAA,QAAA,EACf,CAAA;AACT,QAAA,EAAUP,KAAK,GAAG,0CAA0C,GAAG,EAAE,CAAA;AACjE,QAAA,EAAUA,KAAK,GAAG,iDAAiD,GAAG,gBAAgB,CAAA,CAAA;AAAE,KAC5E,CAAC,CAAA;AAAA,GACP,CAAC,CAAA;AAEP;;;;"}
+{"version":3,"file":"Constructor.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/Constructor.js"],"sourcesContent":["import { Text } from '@asyncapi/generator-react-sdk';\nimport { QueryParamsVariables } from '@asyncapi/generator-components';\nimport { QueryParamsArgumentsDocs } from './QueryParamsArgumentsDocs';\nimport { InitSignature } from './InitSignature';\nimport { ReceiveOperationsDiscriminators } from './ReceiveOperationsDiscriminators';\n\nexport function Constructor({ serverUrl, query, receiveOperations }) {\n  const queryParamsArray = query && Array.from(query.entries());\n\n  return (\n    <>\n      <InitSignature queryParams={queryParamsArray} serverUrl={serverUrl} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      Constructor to initialize the WebSocket client.\n\n      Args:\n          url (str, optional): The WebSocket server URL. Use it if the server URL is \n          different from the default one taken from the AsyncAPI document.`}\n      </Text>\n      <QueryParamsArgumentsDocs queryParams={queryParamsArray} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      self.ws_app = None  # Instance of WebSocketApp\n      self.message_handlers = []      # Callables for incoming messages\n      self.error_handlers = []        # Callables for errors\n      self.outgoing_processors = []   # Callables to process outgoing messages\n      self._stop_event = threading.Event()\n      ${ query ? 'params = {}' : ''}`\n        }\n      </Text>\n      <ReceiveOperationsDiscriminators receiveOperations={receiveOperations} />\n      <QueryParamsVariables\n        language=\"python\"\n        queryParams={queryParamsArray} \n      />\n      <Text newLines={2}>\n        {`\n        ${query ? 'qs = urlencode(params) if params else \"\"' : ''}\n        ${query ? 'self.url = f\"{url}{f\\'?{qs}\\' if qs else \\'\\'}\"' : 'self.url = url'}`}\n      </Text>\n    </>\n  );\n}"],"names":["Constructor","serverUrl","query","receiveOperations","queryParamsArray","Array","from","entries","_jsxs","_Fragment","children","_jsx","InitSignature","queryParams","Text","indent","QueryParamsArgumentsDocs","ReceiveOperationsDiscriminators","QueryParamsVariables","language","newLines"],"mappings":";;;;;;;;;;;;;AAMO,SAASA,WAAWA,CAAC;EAAEC,SAAS;EAAEC,KAAK;AAAEC,EAAAA,iBAAAA;AAAkB,CAAC,EAAE;AACnE,EAAA,MAAMC,gBAAgB,GAAGF,KAAK,IAAIG,KAAK,CAACC,IAAI,CAACJ,KAAK,CAACK,OAAO,EAAE,CAAC,CAAA;EAE7D,oBACEC,eAAA,CAAAC,mBAAA,EAAA;IAAAC,QAAA,EAAA,cACEC,cAAA,CAACC,2BAAa,EAAA;AAACC,MAAAA,WAAW,EAAET,gBAAiB;AAACH,MAAAA,SAAS,EAAEA,SAAAA;AAAU,KAAE,CAAC,eACtEU,cAAA,CAACG,sBAAI,EAAA;AAACC,MAAAA,MAAM,EAAE,CAAE;AAAAL,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA,0EAAA,CAAA;AAA2E,KAC/D,CAAC,eACPC,cAAA,CAACK,iDAAwB,EAAA;AAACH,MAAAA,WAAW,EAAET,gBAAAA;AAAiB,KAAE,CAAC,eAC3DO,cAAA,CAACG,sBAAI,EAAA;AAACC,MAAAA,MAAM,EAAE,CAAE;AAAAL,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA;AACA,MAAA,EAASR,KAAK,GAAG,aAAa,GAAG,EAAE,CAAA,CAAA;AAAE,KAEzB,CAAC,eACPS,cAAA,CAACM,+DAA+B,EAAA;AAACd,MAAAA,iBAAiB,EAAEA,iBAAAA;AAAkB,KAAE,CAAC,eACzEQ,cAAA,CAACO,wCAAoB,EAAA;AACnBC,MAAAA,QAAQ,EAAC,QAAQ;AACjBN,MAAAA,WAAW,EAAET,gBAAAA;AAAiB,KAC/B,CAAC,eACFO,cAAA,CAACG,sBAAI,EAAA;AAACM,MAAAA,QAAQ,EAAE,CAAE;AAAAV,MAAAA,QAAA,EACf,CAAA;AACT,QAAA,EAAUR,KAAK,GAAG,0CAA0C,GAAG,EAAE,CAAA;AACjE,QAAA,EAAUA,KAAK,GAAG,iDAAiD,GAAG,gBAAgB,CAAA,CAAA;AAAE,KAC5E,CAAC,CAAA;AAAA,GACP,CAAC,CAAA;AAEP;;;;"}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/ReceiveOperationsDiscriminators.js

@@ -0,0 +1,29 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+require('source-map-support/register');
+var generatorReactSdk = require('@asyncapi/generator-react-sdk');
+var generatorHelpers = require('@asyncapi/generator-helpers');
+var jsxRuntime = require('/home/runner/work/generator/generator/node_modules/react/cjs/react-jsx-runtime.production.min.js');
+
+function ReceiveOperationsDiscriminators({
+  receiveOperations
+}) {
+  const hasOperations = Array.isArray(receiveOperations) && receiveOperations.length > 0;
+  if (!hasOperations) {
+    return null;
+  }
+  const operationDiscriminators = generatorHelpers.getMessageDiscriminatorsFromOperations(receiveOperations);
+  const serializedDiscriminators = JSON.stringify(operationDiscriminators);
+  return /*#__PURE__*/jsxRuntime.jsx(generatorReactSdk.Text, {
+    indent: 2,
+    newLines: 2,
+    children: `
+      self.receive_operation_handlers = {}
+      self.receive_operation_discriminators = ${serializedDiscriminators}`
+  });
+}
+
+exports.ReceiveOperationsDiscriminators = ReceiveOperationsDiscriminators;
+//# sourceMappingURL=ReceiveOperationsDiscriminators.js.map

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/ReceiveOperationsDiscriminators.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ReceiveOperationsDiscriminators.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/ReceiveOperationsDiscriminators.js"],"sourcesContent":["import { Text } from '@asyncapi/generator-react-sdk';\nimport { getMessageDiscriminatorsFromOperations } from '@asyncapi/generator-helpers';\n\n/**\n * Component that generates initialization code for receive operation discriminators.\n * \n * @param {Object} props - Component properties\n * @param {Array<Object>} [props.receiveOperations] - Receive operations from AsyncAPI document\n * @returns {React.Element|null} Rendered initialization code or null.\n */\nexport function ReceiveOperationsDiscriminators({ receiveOperations }) {\n  const hasOperations = Array.isArray(receiveOperations) && receiveOperations.length > 0;\n  if (!hasOperations) {\n    return null;\n  }\n\n  const operationDiscriminators = getMessageDiscriminatorsFromOperations(receiveOperations);\n  const serializedDiscriminators = JSON.stringify(operationDiscriminators);\n\n  return (\n    <Text indent={2} newLines={2}>\n      {`\n      self.receive_operation_handlers = {}\n      self.receive_operation_discriminators = ${serializedDiscriminators}`\n      }\n    </Text>\n  );\n}\n"],"names":["ReceiveOperationsDiscriminators","receiveOperations","hasOperations","Array","isArray","length","operationDiscriminators","getMessageDiscriminatorsFromOperations","serializedDiscriminators","JSON","stringify","_jsx","Text","indent","newLines","children"],"mappings":";;;;;;;;;AAUO,SAASA,+BAA+BA,CAAC;AAAEC,EAAAA,iBAAAA;AAAkB,CAAC,EAAE;AACrE,EAAA,MAAMC,aAAa,GAAGC,KAAK,CAACC,OAAO,CAACH,iBAAiB,CAAC,IAAIA,iBAAiB,CAACI,MAAM,GAAG,CAAC,CAAA;EACtF,IAAI,CAACH,aAAa,EAAE;AAClB,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;AAEA,EAAA,MAAMI,uBAAuB,GAAGC,uDAAsC,CAACN,iBAAiB,CAAC,CAAA;AACzF,EAAA,MAAMO,wBAAwB,GAAGC,IAAI,CAACC,SAAS,CAACJ,uBAAuB,CAAC,CAAA;EAExE,oBACEK,cAAA,CAACC,sBAAI,EAAA;AAACC,IAAAA,MAAM,EAAE,CAAE;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAC1B,CAAA;AACP;AACA,8CAAA,EAAgDP,wBAAwB,CAAA,CAAA;AAAE,GAEhE,CAAC,CAAA;AAEX;;;;"}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/RegisterReceiveOperations.js

@@ -0,0 +1,66 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+require('source-map-support/register');
+var generatorReactSdk = require('@asyncapi/generator-react-sdk');
+var generatorHelpers = require('@asyncapi/generator-helpers');
+var jsxRuntime = require('/home/runner/work/generator/generator/node_modules/react/cjs/react-jsx-runtime.production.min.js');
+
+function RegisterReceiveOperations({
+  receiveOperations
+}) {
+  if (!receiveOperations || receiveOperations.length === 0) {
+    return null;
+  }
+  return receiveOperations.map(operation => {
+    const operationId = operation.id();
+    const methodName = `register_${generatorHelpers.toSnakeCase(operationId)}_handler`;
+    return /*#__PURE__*/jsxRuntime.jsx(generatorReactSdk.Text, {
+      indent: 2,
+      newLines: 2,
+      children: `def ${methodName}(self, handler, discriminator_key=None, discriminator_value=None):
+    """
+    Register a handler for ${operationId} operation.
+
+    Args:
+        handler (callable): Handler function that receives raw_message as argument
+        discriminator_key (str): Message field to use for routing (e.g., 'type', 'event_type')
+        discriminator_value (str): Expected value for routing (e.g., 'hello', 'user_joined')
+    """
+    if not callable(handler):
+        print("Handler must be callable")
+        return
+
+    # Validate that either both discriminator_key and discriminator_value are provided or neither
+    if (discriminator_key is not None and discriminator_value is None) or (discriminator_key is None and discriminator_value is not None):
+        print("Error: Both discriminator_key and discriminator_value must be provided together")
+        return
+
+    # Register handler
+    self.receive_operation_handlers["${operationId}"] = handler
+
+    # Add discriminator entry to the list if both key and value are provided
+    if discriminator_key is not None and discriminator_value is not None:
+        discriminator_entry = {
+            "key": discriminator_key,
+            "value": discriminator_value,
+            "operation_id": "${operationId}"
+        }
+        
+        # Check if this discriminator already exists
+        exists = any(
+            d.get("key") == discriminator_key and 
+            d.get("value") == discriminator_value and 
+            d.get("operation_id") == "${operationId}" 
+            for d in self.receive_operation_discriminators
+        )
+
+        if not exists:
+          self.receive_operation_discriminators.append(discriminator_entry)`
+    }, operationId);
+  });
+}
+
+exports.RegisterReceiveOperations = RegisterReceiveOperations;
+//# sourceMappingURL=RegisterReceiveOperations.js.map

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/components/RegisterReceiveOperations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RegisterReceiveOperations.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/RegisterReceiveOperations.js"],"sourcesContent":["import { Text } from '@asyncapi/generator-react-sdk';\nimport { toSnakeCase } from '@asyncapi/generator-helpers';\n\n/**\n * Component for rendering receive operation handler registration methods\n * Generates one specific method per operation for better DX\n * \n* @param {Object} props - Component props\n* @param {Array} props.receiveOperations - Array of receive operations from AsyncAPI document\n* @returns {Array<React.Element>|null} Array of Text components containing Python handler registration methods, or null if no operations\n */\nexport function RegisterReceiveOperations({ receiveOperations }) {\n  if (!receiveOperations || receiveOperations.length === 0) {\n    return null;\n  }\n\n  return receiveOperations.map((operation) => {\n    const operationId = operation.id();\n    const methodName = `register_${toSnakeCase(operationId)}_handler`;\n\n    return (\n      <Text indent={2} newLines={2} key={operationId}>\n        {`def ${methodName}(self, handler, discriminator_key=None, discriminator_value=None):\n    \"\"\"\n    Register a handler for ${operationId} operation.\n\n    Args:\n        handler (callable): Handler function that receives raw_message as argument\n        discriminator_key (str): Message field to use for routing (e.g., 'type', 'event_type')\n        discriminator_value (str): Expected value for routing (e.g., 'hello', 'user_joined')\n    \"\"\"\n    if not callable(handler):\n        print(\"Handler must be callable\")\n        return\n\n    # Validate that either both discriminator_key and discriminator_value are provided or neither\n    if (discriminator_key is not None and discriminator_value is None) or (discriminator_key is None and discriminator_value is not None):\n        print(\"Error: Both discriminator_key and discriminator_value must be provided together\")\n        return\n\n    # Register handler\n    self.receive_operation_handlers[\"${operationId}\"] = handler\n\n    # Add discriminator entry to the list if both key and value are provided\n    if discriminator_key is not None and discriminator_value is not None:\n        discriminator_entry = {\n            \"key\": discriminator_key,\n            \"value\": discriminator_value,\n            \"operation_id\": \"${operationId}\"\n        }\n        \n        # Check if this discriminator already exists\n        exists = any(\n            d.get(\"key\") == discriminator_key and \n            d.get(\"value\") == discriminator_value and \n            d.get(\"operation_id\") == \"${operationId}\" \n            for d in self.receive_operation_discriminators\n        )\n\n        if not exists:\n          self.receive_operation_discriminators.append(discriminator_entry)`}\n      </Text>\n    );\n  });\n}"],"names":["RegisterReceiveOperations","receiveOperations","length","map","operation","operationId","id","methodName","toSnakeCase","_jsx","Text","indent","newLines","children"],"mappings":";;;;;;;;;AAWO,SAASA,yBAAyBA,CAAC;AAAEC,EAAAA,iBAAAA;AAAkB,CAAC,EAAE;EAC/D,IAAI,CAACA,iBAAiB,IAAIA,iBAAiB,CAACC,MAAM,KAAK,CAAC,EAAE;AACxD,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;AAEA,EAAA,OAAOD,iBAAiB,CAACE,GAAG,CAAEC,SAAS,IAAK;AAC1C,IAAA,MAAMC,WAAW,GAAGD,SAAS,CAACE,EAAE,EAAE,CAAA;AAClC,IAAA,MAAMC,UAAU,GAAG,CAAA,SAAA,EAAYC,4BAAW,CAACH,WAAW,CAAC,CAAU,QAAA,CAAA,CAAA;IAEjE,oBACEI,cAAA,CAACC,sBAAI,EAAA;AAACC,MAAAA,MAAM,EAAE,CAAE;AAACC,MAAAA,QAAQ,EAAE,CAAE;MAAAC,QAAA,EAC1B,OAAON,UAAU,CAAA;AAC1B;AACA,2BAAA,EAA6BF,WAAW,CAAA;AACxC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,qCAAA,EAAuCA,WAAW,CAAA;AAClD;AACA;AACA;AACA;AACA;AACA;AACA,6BAAA,EAA+BA,WAAW,CAAA;AAC1C;AACA;AACA;AACA;AACA;AACA;AACA,sCAAA,EAAwCA,WAAW,CAAA;AACnD;AACA;AACA;AACA;AACA,2EAAA,CAAA;AAA4E,KAAA,EAvCnCA,WAwC7B,CAAC,CAAA;AAEX,GAAC,CAAC,CAAA;AACJ;;;;"}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/template/client.py.js

@@ -92,9 +92,28 @@ function InitSignature({
   });
 }
 
+function ReceiveOperationsDiscriminators({
+  receiveOperations
+}) {
+  const hasOperations = Array.isArray(receiveOperations) && receiveOperations.length > 0;
+  if (!hasOperations) {
+    return null;
+  }
+  const operationDiscriminators = generatorHelpers.getMessageDiscriminatorsFromOperations(receiveOperations);
+  const serializedDiscriminators = JSON.stringify(operationDiscriminators);
+  return /*#__PURE__*/jsxRuntime.jsx(generatorReactSdk.Text, {
+    indent: 2,
+    newLines: 2,
+    children: `
+      self.receive_operation_handlers = {}
+      self.receive_operation_discriminators = ${serializedDiscriminators}`
+  });
+}
+
 function Constructor({
   serverUrl,
-  query
+  query,
+  receiveOperations
 }) {
   const queryParamsArray = query && Array.from(query.entries());
   return /*#__PURE__*/jsxRuntime.jsxs(jsxRuntime.Fragment, {
@@ -121,8 +140,9 @@ function Constructor({
       self.error_handlers = []        # Callables for errors
       self.outgoing_processors = []   # Callables to process outgoing messages
       self._stop_event = threading.Event()
-      
       ${query ? 'params = {}' : ''}`
+    }), /*#__PURE__*/jsxRuntime.jsx(ReceiveOperationsDiscriminators, {
+      receiveOperations: receiveOperations
     }), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.QueryParamsVariables, {
       language: "python",
       queryParams: queryParamsArray
@@ -166,6 +186,61 @@ function HandleError() {
   });
 }
 
+function RegisterReceiveOperations({
+  receiveOperations
+}) {
+  if (!receiveOperations || receiveOperations.length === 0) {
+    return null;
+  }
+  return receiveOperations.map(operation => {
+    const operationId = operation.id();
+    const methodName = `register_${generatorHelpers.toSnakeCase(operationId)}_handler`;
+    return /*#__PURE__*/jsxRuntime.jsx(generatorReactSdk.Text, {
+      indent: 2,
+      newLines: 2,
+      children: `def ${methodName}(self, handler, discriminator_key=None, discriminator_value=None):
+    """
+    Register a handler for ${operationId} operation.
+
+    Args:
+        handler (callable): Handler function that receives raw_message as argument
+        discriminator_key (str): Message field to use for routing (e.g., 'type', 'event_type')
+        discriminator_value (str): Expected value for routing (e.g., 'hello', 'user_joined')
+    """
+    if not callable(handler):
+        print("Handler must be callable")
+        return
+
+    # Validate that either both discriminator_key and discriminator_value are provided or neither
+    if (discriminator_key is not None and discriminator_value is None) or (discriminator_key is None and discriminator_value is not None):
+        print("Error: Both discriminator_key and discriminator_value must be provided together")
+        return
+
+    # Register handler
+    self.receive_operation_handlers["${operationId}"] = handler
+
+    # Add discriminator entry to the list if both key and value are provided
+    if discriminator_key is not None and discriminator_value is not None:
+        discriminator_entry = {
+            "key": discriminator_key,
+            "value": discriminator_value,
+            "operation_id": "${operationId}"
+        }
+        
+        # Check if this discriminator already exists
+        exists = any(
+            d.get("key") == discriminator_key and 
+            d.get("value") == discriminator_value and 
+            d.get("operation_id") == "${operationId}" 
+            for d in self.receive_operation_discriminators
+        )
+
+        if not exists:
+          self.receive_operation_discriminators.append(discriminator_entry)`
+    }, operationId);
+  });
+}
+
 function ClientClass({
   asyncapi,
   params
@@ -177,13 +252,15 @@ function ClientClass({
   const serverUrl = generatorHelpers.getServerUrl(server);
   const operations = asyncapi.operations();
   const sendOperations = operations.filterBySend();
+  const receiveOperations = operations.filterByReceive();
   return /*#__PURE__*/jsxRuntime.jsxs(generatorReactSdk.Text, {
     children: [/*#__PURE__*/jsxRuntime.jsx(generatorReactSdk.Text, {
       newLines: 2,
       children: `class ${clientName}:`
     }), /*#__PURE__*/jsxRuntime.jsx(Constructor, {
       serverUrl: serverUrl,
-      query: queryParams
+      query: queryParams,
+      receiveOperations: receiveOperations
     }), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.Connect, {
       language: "python",
       title: title
@@ -202,7 +279,9 @@ function ClientClass({
       methodName: "handle_message",
       methodParams: ['self', 'message'],
       preExecutionCode: "\"\"\"Pass the incoming message to all registered message handlers. \"\"\""
-    }), /*#__PURE__*/jsxRuntime.jsx(HandleError, {}), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.SendOperations, {
+    }), /*#__PURE__*/jsxRuntime.jsx(HandleError, {}), /*#__PURE__*/jsxRuntime.jsx(RegisterReceiveOperations, {
+      receiveOperations: receiveOperations
+    }), /*#__PURE__*/jsxRuntime.jsx(generatorComponents.SendOperations, {
       language: "python",
       sendOperations: sendOperations,
       clientName: clientName

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/__transpiled/template/client.py.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.py.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/Requires.js","../../../../../../../../packages/templates/clients/websocket/python/components/Send.js","../../../../../../../../packages/templates/clients/websocket/python/components/QueryParamsArgumentsDocs.js","../../../../../../../../packages/templates/clients/websocket/python/components/InitSignature.js","../../../../../../../../packages/templates/clients/websocket/python/components/Constructor.js","../../../../../../../../packages/templates/clients/websocket/python/components/RegisterOutgoingProcessor.js","../../../../../../../../packages/templates/clients/websocket/python/components/HandleError.js","../../../../../../../../packages/templates/clients/websocket/python/components/ClientClass.js","../../../../../../../../packages/templates/clients/websocket/python/template/client.py.js"],"sourcesContent":["import { DependencyProvider } from '@asyncapi/generator-components';\n\nexport function Requires({ query }) {\n  const additionalDependencies = [];\n  if (query) {\n    additionalDependencies.push('import os');\n    additionalDependencies.push('from urllib.parse import urlencode');\n  }\n  return (\n    <DependencyProvider\n      language=\"python\"\n      additionalDependencies={additionalDependencies}\n    />\n  );\n}","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function Send({ sendOperations }) {\n  if (!sendOperations || sendOperations.length === 0) {\n    return null;\n  }\n\n  return (\n    <Text newLines={2} indent={2}>\n      {\n        `\n@staticmethod\nasync def _send(message, socket):\n    \"\"\"\n    Internal helper to handle the actual sending logic.\n\n    Args:\n        message (dict or str): The message to send.\n        socket (websockets.WebSocketCommonProtocol): The WebSocket to send through.\n\n    Notes:\n        If message is a dictionary, it will be automatically converted to JSON.\n    \"\"\"\n    try:\n        if isinstance(message, dict):\n            message = json.dumps(message)\n        await socket.send(message)\n    except Exception as e:\n        print(\"Error sending:\", e)`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function QueryParamsArgumentsDocs({ queryParams }) {\n  if (!queryParams) {\n    return null;\n  }\n\n  return queryParams.map((param) => {\n    const paramName = param[0];\n    const firstLine = `${paramName} (str, optional):`;\n    const secondLine = `If provided (or if ${paramName.toUpperCase()} environment variable is set), added as ?${paramName}=… to URL`;\n    return (\n      <Text indent={1}>\n        {firstLine}\n        {secondLine}\n      </Text>\n    );\n  });\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function InitSignature({ queryParams, serverUrl }) {\n  if (!queryParams) {\n    return (\n      <Text indent={2} newLines={2}>\n        {`def __init__(self, url: str = \"${serverUrl}\"):`}\n      </Text>\n    );\n  }\n  \n  const queryParamsArguments = queryParams?.map((param) => {\n    const paramName = param[0];\n    const paramDefaultValue = param[1];\n    const defaultValue = paramDefaultValue ? `\"${paramDefaultValue}\"` : 'None';\n    return `${paramName}: str = ${defaultValue}`;\n  }).join(', ');\n  \n  return (\n    <Text indent={2} newLines={2}>\n      {`def __init__(self, url: str = \"${serverUrl}\", ${queryParamsArguments}):`}\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\nimport { QueryParamsVariables } from '@asyncapi/generator-components';\nimport { QueryParamsArgumentsDocs } from './QueryParamsArgumentsDocs';\nimport { InitSignature } from './InitSignature';\n\nexport function Constructor({ serverUrl, query}) {\n  const queryParamsArray = query && Array.from(query.entries());\n  \n  return (\n    <>\n      <InitSignature queryParams={queryParamsArray} serverUrl={serverUrl} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      Constructor to initialize the WebSocket client.\n\n      Args:\n          url (str, optional): The WebSocket server URL. Use it if the server URL is \n          different from the default one taken from the AsyncAPI document.`}\n      </Text>\n      <QueryParamsArgumentsDocs queryParams={queryParamsArray} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      self.ws_app = None  # Instance of WebSocketApp\n      self.message_handlers = []      # Callables for incoming messages\n      self.error_handlers = []        # Callables for errors\n      self.outgoing_processors = []   # Callables to process outgoing messages\n      self._stop_event = threading.Event()\n      \n      ${ query ? 'params = {}' : ''}`\n        }\n      </Text>\n      <QueryParamsVariables\n        language=\"python\"\n        queryParams={queryParamsArray} \n      />\n      <Text newLines={2}>\n        {`\n        ${query ? 'qs = urlencode(params) if params else \"\"' : ''}\n        ${query ? 'self.url = f\"{url}{f\\'?{qs}\\' if qs else \\'\\'}\"' : 'self.url = url'}`}\n      </Text>\n    </>\n  );\n}","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function RegisterOutgoingProcessor() {\n  return (\n    <Text newLines={2} indent={2}>\n      {\n        `def register_outgoing_processor(self, processor):\n    \"\"\"\n    Register a callable that processes outgoing messages automatically.\n    These processors run in sequence before each message is sent.\n    \"\"\"\n    if callable(processor):\n        self.outgoing_processors.append(processor)\n    else:\n        print(\"Outgoing processor must be callable\")`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function HandleError() {\n  return (\n    <Text newLines={2} indent={2}>\n      {\n        `def handle_error(self, error):\n    \"\"\"Pass the error to all registered error handlers. Generic log message is printed if no handlers are registered.\"\"\"\n    if len(self.error_handlers) == 0:\n      print(\"\\\\033[91mError occurred:\\\\033[0m\", error)\n    else:\n      # Call custom error handlers\n      for handler in self.error_handlers:\n        handler(error)`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\nimport { getClientName, getServerUrl, getServer, getQueryParams, getTitle } from '@asyncapi/generator-helpers';\nimport { Send } from './Send';\nimport { Constructor } from './Constructor';\nimport { CloseConnection, RegisterMessageHandler, RegisterErrorHandler, SendOperations, Connect, HandleMessage } from '@asyncapi/generator-components';\nimport { RegisterOutgoingProcessor } from './RegisterOutgoingProcessor';\nimport { HandleError } from './HandleError';\n\nexport function ClientClass({ asyncapi, params }) {\n  const server = getServer(asyncapi.servers(), params.server);\n  const title = getTitle(asyncapi);\n  const queryParams = getQueryParams(asyncapi.channels());\n  const clientName = getClientName(asyncapi, params.appendClientSuffix, params.customClientName);\n  const serverUrl = getServerUrl(server);\n  const operations = asyncapi.operations();\n  const sendOperations = operations.filterBySend();\n  return (\n    <Text>\n      <Text newLines={2}>\n        {`class ${clientName}:`}\n      </Text>\n      <Constructor serverUrl={serverUrl} query={queryParams} />\n      <Connect language=\"python\" title={title} />\n      <RegisterMessageHandler\n        language=\"python\"\n        methodName='register_message_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process incoming messages.\"\"\"'\n      />\n      <RegisterErrorHandler\n        language=\"python\"\n        methodName='register_error_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process errors.\"\"\"'\n      />\n      <RegisterOutgoingProcessor />\n      <HandleMessage\n        language=\"python\"\n        methodName='handle_message'\n        methodParams={['self', 'message']}\n        preExecutionCode='\"\"\"Pass the incoming message to all registered message handlers. \"\"\"'\n      />\n      <HandleError />\n      <SendOperations \n        language=\"python\"\n        sendOperations={sendOperations} \n        clientName={clientName} \n      />\n      <Send sendOperations={sendOperations} />\n      <CloseConnection \n        language=\"python\" \n        methodParams={['self']}\n        preExecutionCode='\"\"\"Cleanly close the WebSocket connection.\"\"\"'\n      />\n    </Text>\n  );\n}\n","import { File } from '@asyncapi/generator-react-sdk';\nimport { getServer, getQueryParams, getInfo } from '@asyncapi/generator-helpers';\nimport { FileHeaderInfo } from '@asyncapi/generator-components';\nimport { Requires } from '../components/Requires';\nimport { ClientClass } from '../components/ClientClass';\n\nexport default function ({ asyncapi, params }) {\n  const server = getServer(asyncapi.servers(), params.server);\n  const info = getInfo(asyncapi);\n  const queryParams = getQueryParams(asyncapi.channels());\n  return (\n    // The clientFileName default values can be found and modified under the .ageneratorrc\n    <File name={params.clientFileName}>\n      <FileHeaderInfo\n        info={info}\n        server={server}\n        language=\"python\"\n      />\n      <Requires query={queryParams} />\n      <ClientClass asyncapi={asyncapi} params={params} />\n    </File>\n  );\n}\n"],"names":["Requires","query","additionalDependencies","push","_jsx","DependencyProvider","language","Send","sendOperations","length","Text","newLines","indent","children","QueryParamsArgumentsDocs","queryParams","map","param","paramName","firstLine","secondLine","toUpperCase","_jsxs","InitSignature","serverUrl","queryParamsArguments","paramDefaultValue","defaultValue","join","Constructor","queryParamsArray","Array","from","entries","_Fragment","QueryParamsVariables","RegisterOutgoingProcessor","HandleError","ClientClass","asyncapi","params","server","getServer","servers","title","getTitle","getQueryParams","channels","clientName","getClientName","appendClientSuffix","customClientName","getServerUrl","operations","filterBySend","Connect","RegisterMessageHandler","methodName","methodParams","preExecutionCode","RegisterErrorHandler","HandleMessage","SendOperations","CloseConnection","info","getInfo","File","name","clientFileName","FileHeaderInfo"],"mappings":";;;;;;;;AAEO,SAASA,QAAQA,CAAC;AAAEC,EAAAA,KAAAA;AAAM,CAAC,EAAE;EAClC,MAAMC,sBAAsB,GAAG,EAAE,CAAA;AACjC,EAAA,IAAID,KAAK,EAAE;AACTC,IAAAA,sBAAsB,CAACC,IAAI,CAAC,WAAW,CAAC,CAAA;AACxCD,IAAAA,sBAAsB,CAACC,IAAI,CAAC,oCAAoC,CAAC,CAAA;AACnE,GAAA;EACA,oBACEC,cAAA,CAACC,sCAAkB,EAAA;AACjBC,IAAAA,QAAQ,EAAC,QAAQ;AACjBJ,IAAAA,sBAAsB,EAAEA,sBAAAA;AAAuB,GAChD,CAAC,CAAA;AAEN;;ACZO,SAASK,IAAIA,CAAC;AAAEC,EAAAA,cAAAA;AAAe,CAAC,EAAE;EACvC,IAAI,CAACA,cAAc,IAAIA,cAAc,CAACC,MAAM,KAAK,CAAC,EAAE;AAClD,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;EAEA,oBACEL,cAAA,CAACM,sBAAI,EAAA;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAACC,IAAAA,MAAM,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAEzB,CAAA;AACR;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,kCAAA,CAAA;AAAmC,GAEzB,CAAC,CAAA;AAEX;;AC9BO,SAASC,wBAAwBA,CAAC;AAAEC,EAAAA,WAAAA;AAAY,CAAC,EAAE;EACxD,IAAI,CAACA,WAAW,EAAE;AAChB,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;AAEA,EAAA,OAAOA,WAAW,CAACC,GAAG,CAAEC,KAAK,IAAK;AAChC,IAAA,MAAMC,SAAS,GAAGD,KAAK,CAAC,CAAC,CAAC,CAAA;AAC1B,IAAA,MAAME,SAAS,GAAG,CAAGD,EAAAA,SAAS,CAAmB,iBAAA,CAAA,CAAA;IACjD,MAAME,UAAU,GAAG,CAAA,mBAAA,EAAsBF,SAAS,CAACG,WAAW,EAAE,CAA4CH,yCAAAA,EAAAA,SAAS,CAAW,SAAA,CAAA,CAAA;IAChI,oBACEI,eAAA,CAACZ,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;MAAAC,QAAA,EAAA,CACbM,SAAS,EACTC,UAAU,CAAA;AAAA,KACP,CAAC,CAAA;AAEX,GAAC,CAAC,CAAA;AACJ;;AChBO,SAASG,aAAaA,CAAC;EAAER,WAAW;AAAES,EAAAA,SAAAA;AAAU,CAAC,EAAE;EACxD,IAAI,CAACT,WAAW,EAAE;IAChB,oBACEX,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAACD,MAAAA,QAAQ,EAAE,CAAE;MAAAE,QAAA,EAC1B,kCAAkCW,SAAS,CAAA,GAAA,CAAA;AAAK,KAC7C,CAAC,CAAA;AAEX,GAAA;EAEA,MAAMC,oBAAoB,GAAGV,WAAW,KAAXA,IAAAA,IAAAA,WAAW,KAAXA,KAAAA,CAAAA,GAAAA,KAAAA,CAAAA,GAAAA,WAAW,CAAEC,GAAG,CAAEC,KAAK,IAAK;AACvD,IAAA,MAAMC,SAAS,GAAGD,KAAK,CAAC,CAAC,CAAC,CAAA;AAC1B,IAAA,MAAMS,iBAAiB,GAAGT,KAAK,CAAC,CAAC,CAAC,CAAA;IAClC,MAAMU,YAAY,GAAGD,iBAAiB,GAAG,IAAIA,iBAAiB,CAAA,CAAA,CAAG,GAAG,MAAM,CAAA;AAC1E,IAAA,OAAO,CAAGR,EAAAA,SAAS,CAAWS,QAAAA,EAAAA,YAAY,CAAE,CAAA,CAAA;AAC9C,GAAC,CAAC,CAACC,IAAI,CAAC,IAAI,CAAC,CAAA;EAEb,oBACExB,cAAA,CAACM,sBAAI,EAAA;AAACE,IAAAA,MAAM,EAAE,CAAE;AAACD,IAAAA,QAAQ,EAAE,CAAE;AAAAE,IAAAA,QAAA,EAC1B,CAAA,+BAAA,EAAkCW,SAAS,CAAA,GAAA,EAAMC,oBAAoB,CAAA,EAAA,CAAA;AAAI,GACtE,CAAC,CAAA;AAEX;;AClBO,SAASI,WAAWA,CAAC;EAAEL,SAAS;AAAEvB,EAAAA,KAAAA;AAAK,CAAC,EAAE;AAC/C,EAAA,MAAM6B,gBAAgB,GAAG7B,KAAK,IAAI8B,KAAK,CAACC,IAAI,CAAC/B,KAAK,CAACgC,OAAO,EAAE,CAAC,CAAA;EAE7D,oBACEX,eAAA,CAAAY,mBAAA,EAAA;IAAArB,QAAA,EAAA,cACET,cAAA,CAACmB,aAAa,EAAA;AAACR,MAAAA,WAAW,EAAEe,gBAAiB;AAACN,MAAAA,SAAS,EAAEA,SAAAA;AAAU,KAAE,CAAC,eACtEpB,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAAAC,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA,0EAAA,CAAA;AAA2E,KAC/D,CAAC,eACPT,cAAA,CAACU,wBAAwB,EAAA;AAACC,MAAAA,WAAW,EAAEe,gBAAAA;AAAiB,KAAE,CAAC,eAC3D1B,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAAAC,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA;AACA;AACA,MAAA,EAASZ,KAAK,GAAG,aAAa,GAAG,EAAE,CAAA,CAAA;AAAE,KAEzB,CAAC,eACPG,cAAA,CAAC+B,wCAAoB,EAAA;AACnB7B,MAAAA,QAAQ,EAAC,QAAQ;AACjBS,MAAAA,WAAW,EAAEe,gBAAAA;AAAiB,KAC/B,CAAC,eACF1B,cAAA,CAACM,sBAAI,EAAA;AAACC,MAAAA,QAAQ,EAAE,CAAE;AAAAE,MAAAA,QAAA,EACf,CAAA;AACT,QAAA,EAAUZ,KAAK,GAAG,0CAA0C,GAAG,EAAE,CAAA;AACjE,QAAA,EAAUA,KAAK,GAAG,iDAAiD,GAAG,gBAAgB,CAAA,CAAA;AAAE,KAC5E,CAAC,CAAA;AAAA,GACP,CAAC,CAAA;AAEP;;AC1CO,SAASmC,yBAAyBA,GAAG;EAC1C,oBACEhC,cAAA,CAACM,sBAAI,EAAA;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAACC,IAAAA,MAAM,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAEzB,CAAA;AACR;AACA;AACA;AACA;AACA;AACA;AACA;AACA,oDAAA,CAAA;AAAqD,GAE3C,CAAC,CAAA;AAEX;;AChBO,SAASwB,WAAWA,GAAG;EAC5B,oBACEjC,cAAA,CAACM,sBAAI,EAAA;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAACC,IAAAA,MAAM,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAEzB,CAAA;AACR;AACA;AACA;AACA;AACA;AACA;AACA,sBAAA,CAAA;AAAuB,GAEb,CAAC,CAAA;AAEX;;ACTO,SAASyB,WAAWA,CAAC;EAAEC,QAAQ;AAAEC,EAAAA,MAAAA;AAAO,CAAC,EAAE;AAChD,EAAA,MAAMC,MAAM,GAAGC,0BAAS,CAACH,QAAQ,CAACI,OAAO,EAAE,EAAEH,MAAM,CAACC,MAAM,CAAC,CAAA;AAC3D,EAAA,MAAMG,KAAK,GAAGC,yBAAQ,CAACN,QAAQ,CAAC,CAAA;EAChC,MAAMxB,WAAW,GAAG+B,+BAAc,CAACP,QAAQ,CAACQ,QAAQ,EAAE,CAAC,CAAA;AACvD,EAAA,MAAMC,UAAU,GAAGC,8BAAa,CAACV,QAAQ,EAAEC,MAAM,CAACU,kBAAkB,EAAEV,MAAM,CAACW,gBAAgB,CAAC,CAAA;AAC9F,EAAA,MAAM3B,SAAS,GAAG4B,6BAAY,CAACX,MAAM,CAAC,CAAA;AACtC,EAAA,MAAMY,UAAU,GAAGd,QAAQ,CAACc,UAAU,EAAE,CAAA;AACxC,EAAA,MAAM7C,cAAc,GAAG6C,UAAU,CAACC,YAAY,EAAE,CAAA;EAChD,oBACEhC,eAAA,CAACZ,sBAAI,EAAA;IAAAG,QAAA,EAAA,cACHT,cAAA,CAACM,sBAAI,EAAA;AAACC,MAAAA,QAAQ,EAAE,CAAE;MAAAE,QAAA,EACf,SAASmC,UAAU,CAAA,CAAA,CAAA;AAAG,KACnB,CAAC,eACP5C,cAAA,CAACyB,WAAW,EAAA;AAACL,MAAAA,SAAS,EAAEA,SAAU;AAACvB,MAAAA,KAAK,EAAEc,WAAAA;AAAY,KAAE,CAAC,eACzDX,cAAA,CAACmD,2BAAO,EAAA;AAACjD,MAAAA,QAAQ,EAAC,QAAQ;AAACsC,MAAAA,KAAK,EAAEA,KAAAA;AAAM,KAAE,CAAC,eAC3CxC,cAAA,CAACoD,0CAAsB,EAAA;AACrBlD,MAAAA,QAAQ,EAAC,QAAQ;AACjBmD,MAAAA,UAAU,EAAC,0BAA0B;AACrCC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,+DAAA;AAAyD,KAC3E,CAAC,eACFvD,cAAA,CAACwD,wCAAoB,EAAA;AACnBtD,MAAAA,QAAQ,EAAC,QAAQ;AACjBmD,MAAAA,UAAU,EAAC,wBAAwB;AACnCC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,oDAAA;KAClB,CAAC,eACFvD,cAAA,CAACgC,yBAAyB,IAAE,CAAC,eAC7BhC,cAAA,CAACyD,iCAAa,EAAA;AACZvD,MAAAA,QAAQ,EAAC,QAAQ;AACjBmD,MAAAA,UAAU,EAAC,gBAAgB;AAC3BC,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,4EAAA;KAClB,CAAC,eACFvD,cAAA,CAACiC,WAAW,IAAE,CAAC,eACfjC,cAAA,CAAC0D,kCAAc,EAAA;AACbxD,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,cAAc,EAAEA,cAAe;AAC/BwC,MAAAA,UAAU,EAAEA,UAAAA;AAAW,KACxB,CAAC,eACF5C,cAAA,CAACG,IAAI,EAAA;AAACC,MAAAA,cAAc,EAAEA,cAAAA;AAAe,KAAE,CAAC,eACxCJ,cAAA,CAAC2D,mCAAe,EAAA;AACdzD,MAAAA,QAAQ,EAAC,QAAQ;MACjBoD,YAAY,EAAE,CAAC,MAAM,CAAE;AACvBC,MAAAA,gBAAgB,EAAC,qDAAA;AAA+C,KACjE,CAAC,CAAA;AAAA,GACE,CAAC,CAAA;AAEX;;AClDe,kBAAU,EAAA;EAAEpB,QAAQ;AAAEC,EAAAA,MAAAA;AAAO,CAAC,EAAE;AAC7C,EAAA,MAAMC,MAAM,GAAGC,0BAAS,CAACH,QAAQ,CAACI,OAAO,EAAE,EAAEH,MAAM,CAACC,MAAM,CAAC,CAAA;AAC3D,EAAA,MAAMuB,IAAI,GAAGC,wBAAO,CAAC1B,QAAQ,CAAC,CAAA;EAC9B,MAAMxB,WAAW,GAAG+B,+BAAc,CAACP,QAAQ,CAACQ,QAAQ,EAAE,CAAC,CAAA;AACvD,EAAA;AAAA;AACE;AACAzB,IAAAA,eAAA,CAAC4C,sBAAI,EAAA;MAACC,IAAI,EAAE3B,MAAM,CAAC4B,cAAe;MAAAvD,QAAA,EAAA,cAChCT,cAAA,CAACiE,kCAAc,EAAA;AACbL,QAAAA,IAAI,EAAEA,IAAK;AACXvB,QAAAA,MAAM,EAAEA,MAAO;AACfnC,QAAAA,QAAQ,EAAC,QAAA;AAAQ,OAClB,CAAC,eACFF,cAAA,CAACJ,QAAQ,EAAA;AAACC,QAAAA,KAAK,EAAEc,WAAAA;AAAY,OAAE,CAAC,eAChCX,cAAA,CAACkC,WAAW,EAAA;AAACC,QAAAA,QAAQ,EAAEA,QAAS;AAACC,QAAAA,MAAM,EAAEA,MAAAA;AAAO,OAAE,CAAC,CAAA;KAC/C,CAAA;AAAC,IAAA;AAEX;;;;"}
+{"version":3,"file":"client.py.js","sources":["../../../../../../../../packages/templates/clients/websocket/python/components/Requires.js","../../../../../../../../packages/templates/clients/websocket/python/components/Send.js","../../../../../../../../packages/templates/clients/websocket/python/components/QueryParamsArgumentsDocs.js","../../../../../../../../packages/templates/clients/websocket/python/components/InitSignature.js","../../../../../../../../packages/templates/clients/websocket/python/components/ReceiveOperationsDiscriminators.js","../../../../../../../../packages/templates/clients/websocket/python/components/Constructor.js","../../../../../../../../packages/templates/clients/websocket/python/components/RegisterOutgoingProcessor.js","../../../../../../../../packages/templates/clients/websocket/python/components/HandleError.js","../../../../../../../../packages/templates/clients/websocket/python/components/RegisterReceiveOperations.js","../../../../../../../../packages/templates/clients/websocket/python/components/ClientClass.js","../../../../../../../../packages/templates/clients/websocket/python/template/client.py.js"],"sourcesContent":["import { DependencyProvider } from '@asyncapi/generator-components';\n\nexport function Requires({ query }) {\n  const additionalDependencies = [];\n  if (query) {\n    additionalDependencies.push('import os');\n    additionalDependencies.push('from urllib.parse import urlencode');\n  }\n  return (\n    <DependencyProvider\n      language=\"python\"\n      additionalDependencies={additionalDependencies}\n    />\n  );\n}","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function Send({ sendOperations }) {\n  if (!sendOperations || sendOperations.length === 0) {\n    return null;\n  }\n\n  return (\n    <Text newLines={2} indent={2}>\n      {\n        `\n@staticmethod\nasync def _send(message, socket):\n    \"\"\"\n    Internal helper to handle the actual sending logic.\n\n    Args:\n        message (dict or str): The message to send.\n        socket (websockets.WebSocketCommonProtocol): The WebSocket to send through.\n\n    Notes:\n        If message is a dictionary, it will be automatically converted to JSON.\n    \"\"\"\n    try:\n        if isinstance(message, dict):\n            message = json.dumps(message)\n        await socket.send(message)\n    except Exception as e:\n        print(\"Error sending:\", e)`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function QueryParamsArgumentsDocs({ queryParams }) {\n  if (!queryParams) {\n    return null;\n  }\n\n  return queryParams.map((param) => {\n    const paramName = param[0];\n    const firstLine = `${paramName} (str, optional):`;\n    const secondLine = `If provided (or if ${paramName.toUpperCase()} environment variable is set), added as ?${paramName}=… to URL`;\n    return (\n      <Text indent={1}>\n        {firstLine}\n        {secondLine}\n      </Text>\n    );\n  });\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function InitSignature({ queryParams, serverUrl }) {\n  if (!queryParams) {\n    return (\n      <Text indent={2} newLines={2}>\n        {`def __init__(self, url: str = \"${serverUrl}\"):`}\n      </Text>\n    );\n  }\n  \n  const queryParamsArguments = queryParams?.map((param) => {\n    const paramName = param[0];\n    const paramDefaultValue = param[1];\n    const defaultValue = paramDefaultValue ? `\"${paramDefaultValue}\"` : 'None';\n    return `${paramName}: str = ${defaultValue}`;\n  }).join(', ');\n  \n  return (\n    <Text indent={2} newLines={2}>\n      {`def __init__(self, url: str = \"${serverUrl}\", ${queryParamsArguments}):`}\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\nimport { getMessageDiscriminatorsFromOperations } from '@asyncapi/generator-helpers';\n\n/**\n * Component that generates initialization code for receive operation discriminators.\n * \n * @param {Object} props - Component properties\n * @param {Array<Object>} [props.receiveOperations] - Receive operations from AsyncAPI document\n * @returns {React.Element|null} Rendered initialization code or null.\n */\nexport function ReceiveOperationsDiscriminators({ receiveOperations }) {\n  const hasOperations = Array.isArray(receiveOperations) && receiveOperations.length > 0;\n  if (!hasOperations) {\n    return null;\n  }\n\n  const operationDiscriminators = getMessageDiscriminatorsFromOperations(receiveOperations);\n  const serializedDiscriminators = JSON.stringify(operationDiscriminators);\n\n  return (\n    <Text indent={2} newLines={2}>\n      {`\n      self.receive_operation_handlers = {}\n      self.receive_operation_discriminators = ${serializedDiscriminators}`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\nimport { QueryParamsVariables } from '@asyncapi/generator-components';\nimport { QueryParamsArgumentsDocs } from './QueryParamsArgumentsDocs';\nimport { InitSignature } from './InitSignature';\nimport { ReceiveOperationsDiscriminators } from './ReceiveOperationsDiscriminators';\n\nexport function Constructor({ serverUrl, query, receiveOperations }) {\n  const queryParamsArray = query && Array.from(query.entries());\n\n  return (\n    <>\n      <InitSignature queryParams={queryParamsArray} serverUrl={serverUrl} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      Constructor to initialize the WebSocket client.\n\n      Args:\n          url (str, optional): The WebSocket server URL. Use it if the server URL is \n          different from the default one taken from the AsyncAPI document.`}\n      </Text>\n      <QueryParamsArgumentsDocs queryParams={queryParamsArray} />\n      <Text indent={2}>\n        {`\n      \"\"\"\n      self.ws_app = None  # Instance of WebSocketApp\n      self.message_handlers = []      # Callables for incoming messages\n      self.error_handlers = []        # Callables for errors\n      self.outgoing_processors = []   # Callables to process outgoing messages\n      self._stop_event = threading.Event()\n      ${ query ? 'params = {}' : ''}`\n        }\n      </Text>\n      <ReceiveOperationsDiscriminators receiveOperations={receiveOperations} />\n      <QueryParamsVariables\n        language=\"python\"\n        queryParams={queryParamsArray} \n      />\n      <Text newLines={2}>\n        {`\n        ${query ? 'qs = urlencode(params) if params else \"\"' : ''}\n        ${query ? 'self.url = f\"{url}{f\\'?{qs}\\' if qs else \\'\\'}\"' : 'self.url = url'}`}\n      </Text>\n    </>\n  );\n}","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function RegisterOutgoingProcessor() {\n  return (\n    <Text newLines={2} indent={2}>\n      {\n        `def register_outgoing_processor(self, processor):\n    \"\"\"\n    Register a callable that processes outgoing messages automatically.\n    These processors run in sequence before each message is sent.\n    \"\"\"\n    if callable(processor):\n        self.outgoing_processors.append(processor)\n    else:\n        print(\"Outgoing processor must be callable\")`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\n\nexport function HandleError() {\n  return (\n    <Text newLines={2} indent={2}>\n      {\n        `def handle_error(self, error):\n    \"\"\"Pass the error to all registered error handlers. Generic log message is printed if no handlers are registered.\"\"\"\n    if len(self.error_handlers) == 0:\n      print(\"\\\\033[91mError occurred:\\\\033[0m\", error)\n    else:\n      # Call custom error handlers\n      for handler in self.error_handlers:\n        handler(error)`\n      }\n    </Text>\n  );\n}\n","import { Text } from '@asyncapi/generator-react-sdk';\nimport { toSnakeCase } from '@asyncapi/generator-helpers';\n\n/**\n * Component for rendering receive operation handler registration methods\n * Generates one specific method per operation for better DX\n * \n* @param {Object} props - Component props\n* @param {Array} props.receiveOperations - Array of receive operations from AsyncAPI document\n* @returns {Array<React.Element>|null} Array of Text components containing Python handler registration methods, or null if no operations\n */\nexport function RegisterReceiveOperations({ receiveOperations }) {\n  if (!receiveOperations || receiveOperations.length === 0) {\n    return null;\n  }\n\n  return receiveOperations.map((operation) => {\n    const operationId = operation.id();\n    const methodName = `register_${toSnakeCase(operationId)}_handler`;\n\n    return (\n      <Text indent={2} newLines={2} key={operationId}>\n        {`def ${methodName}(self, handler, discriminator_key=None, discriminator_value=None):\n    \"\"\"\n    Register a handler for ${operationId} operation.\n\n    Args:\n        handler (callable): Handler function that receives raw_message as argument\n        discriminator_key (str): Message field to use for routing (e.g., 'type', 'event_type')\n        discriminator_value (str): Expected value for routing (e.g., 'hello', 'user_joined')\n    \"\"\"\n    if not callable(handler):\n        print(\"Handler must be callable\")\n        return\n\n    # Validate that either both discriminator_key and discriminator_value are provided or neither\n    if (discriminator_key is not None and discriminator_value is None) or (discriminator_key is None and discriminator_value is not None):\n        print(\"Error: Both discriminator_key and discriminator_value must be provided together\")\n        return\n\n    # Register handler\n    self.receive_operation_handlers[\"${operationId}\"] = handler\n\n    # Add discriminator entry to the list if both key and value are provided\n    if discriminator_key is not None and discriminator_value is not None:\n        discriminator_entry = {\n            \"key\": discriminator_key,\n            \"value\": discriminator_value,\n            \"operation_id\": \"${operationId}\"\n        }\n        \n        # Check if this discriminator already exists\n        exists = any(\n            d.get(\"key\") == discriminator_key and \n            d.get(\"value\") == discriminator_value and \n            d.get(\"operation_id\") == \"${operationId}\" \n            for d in self.receive_operation_discriminators\n        )\n\n        if not exists:\n          self.receive_operation_discriminators.append(discriminator_entry)`}\n      </Text>\n    );\n  });\n}","import { Text } from '@asyncapi/generator-react-sdk';\nimport { getClientName, getServerUrl, getServer, getQueryParams, getTitle } from '@asyncapi/generator-helpers';\nimport { Send } from './Send';\nimport { Constructor } from './Constructor';\nimport { CloseConnection, RegisterMessageHandler, RegisterErrorHandler, SendOperations, Connect, HandleMessage } from '@asyncapi/generator-components';\nimport { RegisterOutgoingProcessor } from './RegisterOutgoingProcessor';\nimport { HandleError } from './HandleError';\nimport { RegisterReceiveOperations } from './RegisterReceiveOperations';\n\nexport function ClientClass({ asyncapi, params }) {\n  const server = getServer(asyncapi.servers(), params.server);\n  const title = getTitle(asyncapi);\n  const queryParams = getQueryParams(asyncapi.channels());\n  const clientName = getClientName(asyncapi, params.appendClientSuffix, params.customClientName);\n  const serverUrl = getServerUrl(server);\n  const operations = asyncapi.operations();\n  const sendOperations = operations.filterBySend();\n  const receiveOperations = operations.filterByReceive();\n\n  return (\n    <Text>\n      <Text newLines={2}>\n        {`class ${clientName}:`}\n      </Text>\n      <Constructor serverUrl={serverUrl} query={queryParams} receiveOperations={receiveOperations} />\n      <Connect language=\"python\" title={title} />\n      <RegisterMessageHandler\n        language=\"python\"\n        methodName='register_message_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process incoming messages.\"\"\"'\n      />\n      <RegisterErrorHandler\n        language=\"python\"\n        methodName='register_error_handler'\n        methodParams={['self', 'handler']}\n        preExecutionCode='\"\"\"Register a callable to process errors.\"\"\"'\n      />\n      <RegisterOutgoingProcessor />\n      <HandleMessage\n        language=\"python\"\n        methodName='handle_message'\n        methodParams={['self', 'message']}\n        preExecutionCode='\"\"\"Pass the incoming message to all registered message handlers. \"\"\"'\n      />\n      <HandleError />\n      <RegisterReceiveOperations receiveOperations={receiveOperations} />\n      <SendOperations \n        language=\"python\"\n        sendOperations={sendOperations} \n        clientName={clientName} \n      />\n      <Send sendOperations={sendOperations} />\n      <CloseConnection \n        language=\"python\" \n        methodParams={['self']}\n        preExecutionCode='\"\"\"Cleanly close the WebSocket connection.\"\"\"'\n      />\n    </Text>\n  );\n}\n","import { File } from '@asyncapi/generator-react-sdk';\nimport { getServer, getQueryParams, getInfo } from '@asyncapi/generator-helpers';\nimport { FileHeaderInfo } from '@asyncapi/generator-components';\nimport { Requires } from '../components/Requires';\nimport { ClientClass } from '../components/ClientClass';\n\nexport default function ({ asyncapi, params }) {\n  const server = getServer(asyncapi.servers(), params.server);\n  const info = getInfo(asyncapi);\n  const queryParams = getQueryParams(asyncapi.channels());\n  return (\n    // The clientFileName default values can be found and modified under the .ageneratorrc\n    <File name={params.clientFileName}>\n      <FileHeaderInfo\n        info={info}\n        server={server}\n        language=\"python\"\n      />\n      <Requires query={queryParams} />\n      <ClientClass asyncapi={asyncapi} params={params} />\n    </File>\n  );\n}\n"],"names":["Requires","query","additionalDependencies","push","_jsx","DependencyProvider","language","Send","sendOperations","length","Text","newLines","indent","children","QueryParamsArgumentsDocs","queryParams","map","param","paramName","firstLine","secondLine","toUpperCase","_jsxs","InitSignature","serverUrl","queryParamsArguments","paramDefaultValue","defaultValue","join","ReceiveOperationsDiscriminators","receiveOperations","hasOperations","Array","isArray","operationDiscriminators","getMessageDiscriminatorsFromOperations","serializedDiscriminators","JSON","stringify","Constructor","queryParamsArray","from","entries","_Fragment","QueryParamsVariables","RegisterOutgoingProcessor","HandleError","RegisterReceiveOperations","operation","operationId","id","methodName","toSnakeCase","ClientClass","asyncapi","params","server","getServer","servers","title","getTitle","getQueryParams","channels","clientName","getClientName","appendClientSuffix","customClientName","getServerUrl","operations","filterBySend","filterByReceive","Connect","RegisterMessageHandler","methodParams","preExecutionCode","RegisterErrorHandler","HandleMessage","SendOperations","CloseConnection","info","getInfo","File","name","clientFileName","FileHeaderInfo"],"mappings":";;;;;;;;AAEO,SAASA,QAAQA,CAAC;AAAEC,EAAAA,KAAAA;AAAM,CAAC,EAAE;EAClC,MAAMC,sBAAsB,GAAG,EAAE,CAAA;AACjC,EAAA,IAAID,KAAK,EAAE;AACTC,IAAAA,sBAAsB,CAACC,IAAI,CAAC,WAAW,CAAC,CAAA;AACxCD,IAAAA,sBAAsB,CAACC,IAAI,CAAC,oCAAoC,CAAC,CAAA;AACnE,GAAA;EACA,oBACEC,cAAA,CAACC,sCAAkB,EAAA;AACjBC,IAAAA,QAAQ,EAAC,QAAQ;AACjBJ,IAAAA,sBAAsB,EAAEA,sBAAAA;AAAuB,GAChD,CAAC,CAAA;AAEN;;ACZO,SAASK,IAAIA,CAAC;AAAEC,EAAAA,cAAAA;AAAe,CAAC,EAAE;EACvC,IAAI,CAACA,cAAc,IAAIA,cAAc,CAACC,MAAM,KAAK,CAAC,EAAE;AAClD,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;EAEA,oBACEL,cAAA,CAACM,sBAAI,EAAA;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAACC,IAAAA,MAAM,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAEzB,CAAA;AACR;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,kCAAA,CAAA;AAAmC,GAEzB,CAAC,CAAA;AAEX;;AC9BO,SAASC,wBAAwBA,CAAC;AAAEC,EAAAA,WAAAA;AAAY,CAAC,EAAE;EACxD,IAAI,CAACA,WAAW,EAAE;AAChB,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;AAEA,EAAA,OAAOA,WAAW,CAACC,GAAG,CAAEC,KAAK,IAAK;AAChC,IAAA,MAAMC,SAAS,GAAGD,KAAK,CAAC,CAAC,CAAC,CAAA;AAC1B,IAAA,MAAME,SAAS,GAAG,CAAGD,EAAAA,SAAS,CAAmB,iBAAA,CAAA,CAAA;IACjD,MAAME,UAAU,GAAG,CAAA,mBAAA,EAAsBF,SAAS,CAACG,WAAW,EAAE,CAA4CH,yCAAAA,EAAAA,SAAS,CAAW,SAAA,CAAA,CAAA;IAChI,oBACEI,eAAA,CAACZ,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;MAAAC,QAAA,EAAA,CACbM,SAAS,EACTC,UAAU,CAAA;AAAA,KACP,CAAC,CAAA;AAEX,GAAC,CAAC,CAAA;AACJ;;AChBO,SAASG,aAAaA,CAAC;EAAER,WAAW;AAAES,EAAAA,SAAAA;AAAU,CAAC,EAAE;EACxD,IAAI,CAACT,WAAW,EAAE;IAChB,oBACEX,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAACD,MAAAA,QAAQ,EAAE,CAAE;MAAAE,QAAA,EAC1B,kCAAkCW,SAAS,CAAA,GAAA,CAAA;AAAK,KAC7C,CAAC,CAAA;AAEX,GAAA;EAEA,MAAMC,oBAAoB,GAAGV,WAAW,KAAXA,IAAAA,IAAAA,WAAW,KAAXA,KAAAA,CAAAA,GAAAA,KAAAA,CAAAA,GAAAA,WAAW,CAAEC,GAAG,CAAEC,KAAK,IAAK;AACvD,IAAA,MAAMC,SAAS,GAAGD,KAAK,CAAC,CAAC,CAAC,CAAA;AAC1B,IAAA,MAAMS,iBAAiB,GAAGT,KAAK,CAAC,CAAC,CAAC,CAAA;IAClC,MAAMU,YAAY,GAAGD,iBAAiB,GAAG,IAAIA,iBAAiB,CAAA,CAAA,CAAG,GAAG,MAAM,CAAA;AAC1E,IAAA,OAAO,CAAGR,EAAAA,SAAS,CAAWS,QAAAA,EAAAA,YAAY,CAAE,CAAA,CAAA;AAC9C,GAAC,CAAC,CAACC,IAAI,CAAC,IAAI,CAAC,CAAA;EAEb,oBACExB,cAAA,CAACM,sBAAI,EAAA;AAACE,IAAAA,MAAM,EAAE,CAAE;AAACD,IAAAA,QAAQ,EAAE,CAAE;AAAAE,IAAAA,QAAA,EAC1B,CAAA,+BAAA,EAAkCW,SAAS,CAAA,GAAA,EAAMC,oBAAoB,CAAA,EAAA,CAAA;AAAI,GACtE,CAAC,CAAA;AAEX;;ACbO,SAASI,+BAA+BA,CAAC;AAAEC,EAAAA,iBAAAA;AAAkB,CAAC,EAAE;AACrE,EAAA,MAAMC,aAAa,GAAGC,KAAK,CAACC,OAAO,CAACH,iBAAiB,CAAC,IAAIA,iBAAiB,CAACrB,MAAM,GAAG,CAAC,CAAA;EACtF,IAAI,CAACsB,aAAa,EAAE;AAClB,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;AAEA,EAAA,MAAMG,uBAAuB,GAAGC,uDAAsC,CAACL,iBAAiB,CAAC,CAAA;AACzF,EAAA,MAAMM,wBAAwB,GAAGC,IAAI,CAACC,SAAS,CAACJ,uBAAuB,CAAC,CAAA;EAExE,oBACE9B,cAAA,CAACM,sBAAI,EAAA;AAACE,IAAAA,MAAM,EAAE,CAAE;AAACD,IAAAA,QAAQ,EAAE,CAAE;AAAAE,IAAAA,QAAA,EAC1B,CAAA;AACP;AACA,8CAAA,EAAgDuB,wBAAwB,CAAA,CAAA;AAAE,GAEhE,CAAC,CAAA;AAEX;;ACrBO,SAASG,WAAWA,CAAC;EAAEf,SAAS;EAAEvB,KAAK;AAAE6B,EAAAA,iBAAAA;AAAkB,CAAC,EAAE;AACnE,EAAA,MAAMU,gBAAgB,GAAGvC,KAAK,IAAI+B,KAAK,CAACS,IAAI,CAACxC,KAAK,CAACyC,OAAO,EAAE,CAAC,CAAA;EAE7D,oBACEpB,eAAA,CAAAqB,mBAAA,EAAA;IAAA9B,QAAA,EAAA,cACET,cAAA,CAACmB,aAAa,EAAA;AAACR,MAAAA,WAAW,EAAEyB,gBAAiB;AAAChB,MAAAA,SAAS,EAAEA,SAAAA;AAAU,KAAE,CAAC,eACtEpB,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAAAC,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA,0EAAA,CAAA;AAA2E,KAC/D,CAAC,eACPT,cAAA,CAACU,wBAAwB,EAAA;AAACC,MAAAA,WAAW,EAAEyB,gBAAAA;AAAiB,KAAE,CAAC,eAC3DpC,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAAAC,MAAAA,QAAA,EACb,CAAA;AACT;AACA;AACA;AACA;AACA;AACA;AACA,MAAA,EAASZ,KAAK,GAAG,aAAa,GAAG,EAAE,CAAA,CAAA;AAAE,KAEzB,CAAC,eACPG,cAAA,CAACyB,+BAA+B,EAAA;AAACC,MAAAA,iBAAiB,EAAEA,iBAAAA;AAAkB,KAAE,CAAC,eACzE1B,cAAA,CAACwC,wCAAoB,EAAA;AACnBtC,MAAAA,QAAQ,EAAC,QAAQ;AACjBS,MAAAA,WAAW,EAAEyB,gBAAAA;AAAiB,KAC/B,CAAC,eACFpC,cAAA,CAACM,sBAAI,EAAA;AAACC,MAAAA,QAAQ,EAAE,CAAE;AAAAE,MAAAA,QAAA,EACf,CAAA;AACT,QAAA,EAAUZ,KAAK,GAAG,0CAA0C,GAAG,EAAE,CAAA;AACjE,QAAA,EAAUA,KAAK,GAAG,iDAAiD,GAAG,gBAAgB,CAAA,CAAA;AAAE,KAC5E,CAAC,CAAA;AAAA,GACP,CAAC,CAAA;AAEP;;AC3CO,SAAS4C,yBAAyBA,GAAG;EAC1C,oBACEzC,cAAA,CAACM,sBAAI,EAAA;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAACC,IAAAA,MAAM,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAEzB,CAAA;AACR;AACA;AACA;AACA;AACA;AACA;AACA;AACA,oDAAA,CAAA;AAAqD,GAE3C,CAAC,CAAA;AAEX;;AChBO,SAASiC,WAAWA,GAAG;EAC5B,oBACE1C,cAAA,CAACM,sBAAI,EAAA;AAACC,IAAAA,QAAQ,EAAE,CAAE;AAACC,IAAAA,MAAM,EAAE,CAAE;AAAAC,IAAAA,QAAA,EAEzB,CAAA;AACR;AACA;AACA;AACA;AACA;AACA;AACA,sBAAA,CAAA;AAAuB,GAEb,CAAC,CAAA;AAEX;;ACNO,SAASkC,yBAAyBA,CAAC;AAAEjB,EAAAA,iBAAAA;AAAkB,CAAC,EAAE;EAC/D,IAAI,CAACA,iBAAiB,IAAIA,iBAAiB,CAACrB,MAAM,KAAK,CAAC,EAAE;AACxD,IAAA,OAAO,IAAI,CAAA;AACb,GAAA;AAEA,EAAA,OAAOqB,iBAAiB,CAACd,GAAG,CAAEgC,SAAS,IAAK;AAC1C,IAAA,MAAMC,WAAW,GAAGD,SAAS,CAACE,EAAE,EAAE,CAAA;AAClC,IAAA,MAAMC,UAAU,GAAG,CAAA,SAAA,EAAYC,4BAAW,CAACH,WAAW,CAAC,CAAU,QAAA,CAAA,CAAA;IAEjE,oBACE7C,cAAA,CAACM,sBAAI,EAAA;AAACE,MAAAA,MAAM,EAAE,CAAE;AAACD,MAAAA,QAAQ,EAAE,CAAE;MAAAE,QAAA,EAC1B,OAAOsC,UAAU,CAAA;AAC1B;AACA,2BAAA,EAA6BF,WAAW,CAAA;AACxC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,qCAAA,EAAuCA,WAAW,CAAA;AAClD;AACA;AACA;AACA;AACA;AACA;AACA,6BAAA,EAA+BA,WAAW,CAAA;AAC1C;AACA;AACA;AACA;AACA;AACA;AACA,sCAAA,EAAwCA,WAAW,CAAA;AACnD;AACA;AACA;AACA;AACA,2EAAA,CAAA;AAA4E,KAAA,EAvCnCA,WAwC7B,CAAC,CAAA;AAEX,GAAC,CAAC,CAAA;AACJ;;ACvDO,SAASI,WAAWA,CAAC;EAAEC,QAAQ;AAAEC,EAAAA,MAAAA;AAAO,CAAC,EAAE;AAChD,EAAA,MAAMC,MAAM,GAAGC,0BAAS,CAACH,QAAQ,CAACI,OAAO,EAAE,EAAEH,MAAM,CAACC,MAAM,CAAC,CAAA;AAC3D,EAAA,MAAMG,KAAK,GAAGC,yBAAQ,CAACN,QAAQ,CAAC,CAAA;EAChC,MAAMvC,WAAW,GAAG8C,+BAAc,CAACP,QAAQ,CAACQ,QAAQ,EAAE,CAAC,CAAA;AACvD,EAAA,MAAMC,UAAU,GAAGC,8BAAa,CAACV,QAAQ,EAAEC,MAAM,CAACU,kBAAkB,EAAEV,MAAM,CAACW,gBAAgB,CAAC,CAAA;AAC9F,EAAA,MAAM1C,SAAS,GAAG2C,6BAAY,CAACX,MAAM,CAAC,CAAA;AACtC,EAAA,MAAMY,UAAU,GAAGd,QAAQ,CAACc,UAAU,EAAE,CAAA;AACxC,EAAA,MAAM5D,cAAc,GAAG4D,UAAU,CAACC,YAAY,EAAE,CAAA;AAChD,EAAA,MAAMvC,iBAAiB,GAAGsC,UAAU,CAACE,eAAe,EAAE,CAAA;EAEtD,oBACEhD,eAAA,CAACZ,sBAAI,EAAA;IAAAG,QAAA,EAAA,cACHT,cAAA,CAACM,sBAAI,EAAA;AAACC,MAAAA,QAAQ,EAAE,CAAE;MAAAE,QAAA,EACf,SAASkD,UAAU,CAAA,CAAA,CAAA;AAAG,KACnB,CAAC,eACP3D,cAAA,CAACmC,WAAW,EAAA;AAACf,MAAAA,SAAS,EAAEA,SAAU;AAACvB,MAAAA,KAAK,EAAEc,WAAY;AAACe,MAAAA,iBAAiB,EAAEA,iBAAAA;AAAkB,KAAE,CAAC,eAC/F1B,cAAA,CAACmE,2BAAO,EAAA;AAACjE,MAAAA,QAAQ,EAAC,QAAQ;AAACqD,MAAAA,KAAK,EAAEA,KAAAA;AAAM,KAAE,CAAC,eAC3CvD,cAAA,CAACoE,0CAAsB,EAAA;AACrBlE,MAAAA,QAAQ,EAAC,QAAQ;AACjB6C,MAAAA,UAAU,EAAC,0BAA0B;AACrCsB,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,+DAAA;AAAyD,KAC3E,CAAC,eACFtE,cAAA,CAACuE,wCAAoB,EAAA;AACnBrE,MAAAA,QAAQ,EAAC,QAAQ;AACjB6C,MAAAA,UAAU,EAAC,wBAAwB;AACnCsB,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,oDAAA;KAClB,CAAC,eACFtE,cAAA,CAACyC,yBAAyB,IAAE,CAAC,eAC7BzC,cAAA,CAACwE,iCAAa,EAAA;AACZtE,MAAAA,QAAQ,EAAC,QAAQ;AACjB6C,MAAAA,UAAU,EAAC,gBAAgB;AAC3BsB,MAAAA,YAAY,EAAE,CAAC,MAAM,EAAE,SAAS,CAAE;AAClCC,MAAAA,gBAAgB,EAAC,4EAAA;KAClB,CAAC,eACFtE,cAAA,CAAC0C,WAAW,IAAE,CAAC,eACf1C,cAAA,CAAC2C,yBAAyB,EAAA;AAACjB,MAAAA,iBAAiB,EAAEA,iBAAAA;AAAkB,KAAE,CAAC,eACnE1B,cAAA,CAACyE,kCAAc,EAAA;AACbvE,MAAAA,QAAQ,EAAC,QAAQ;AACjBE,MAAAA,cAAc,EAAEA,cAAe;AAC/BuD,MAAAA,UAAU,EAAEA,UAAAA;AAAW,KACxB,CAAC,eACF3D,cAAA,CAACG,IAAI,EAAA;AAACC,MAAAA,cAAc,EAAEA,cAAAA;AAAe,KAAE,CAAC,eACxCJ,cAAA,CAAC0E,mCAAe,EAAA;AACdxE,MAAAA,QAAQ,EAAC,QAAQ;MACjBmE,YAAY,EAAE,CAAC,MAAM,CAAE;AACvBC,MAAAA,gBAAgB,EAAC,qDAAA;AAA+C,KACjE,CAAC,CAAA;AAAA,GACE,CAAC,CAAA;AAEX;;ACtDe,kBAAU,EAAA;EAAEpB,QAAQ;AAAEC,EAAAA,MAAAA;AAAO,CAAC,EAAE;AAC7C,EAAA,MAAMC,MAAM,GAAGC,0BAAS,CAACH,QAAQ,CAACI,OAAO,EAAE,EAAEH,MAAM,CAACC,MAAM,CAAC,CAAA;AAC3D,EAAA,MAAMuB,IAAI,GAAGC,wBAAO,CAAC1B,QAAQ,CAAC,CAAA;EAC9B,MAAMvC,WAAW,GAAG8C,+BAAc,CAACP,QAAQ,CAACQ,QAAQ,EAAE,CAAC,CAAA;AACvD,EAAA;AAAA;AACE;AACAxC,IAAAA,eAAA,CAAC2D,sBAAI,EAAA;MAACC,IAAI,EAAE3B,MAAM,CAAC4B,cAAe;MAAAtE,QAAA,EAAA,cAChCT,cAAA,CAACgF,kCAAc,EAAA;AACbL,QAAAA,IAAI,EAAEA,IAAK;AACXvB,QAAAA,MAAM,EAAEA,MAAO;AACflD,QAAAA,QAAQ,EAAC,QAAA;AAAQ,OAClB,CAAC,eACFF,cAAA,CAACJ,QAAQ,EAAA;AAACC,QAAAA,KAAK,EAAEc,WAAAA;AAAY,OAAE,CAAC,eAChCX,cAAA,CAACiD,WAAW,EAAA;AAACC,QAAAA,QAAQ,EAAEA,QAAS;AAACC,QAAAA,MAAM,EAAEA,MAAAA;AAAO,OAAE,CAAC,CAAA;KAC/C,CAAA;AAAC,IAAA;AAEX;;;;"}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/components/ClientClass.js

@@ -5,6 +5,7 @@ import { Constructor } from './Constructor';
 import { CloseConnection, RegisterMessageHandler, RegisterErrorHandler, SendOperations, Connect, HandleMessage } from '@asyncapi/generator-components';
 import { RegisterOutgoingProcessor } from './RegisterOutgoingProcessor';
 import { HandleError } from './HandleError';
+import { RegisterReceiveOperations } from './RegisterReceiveOperations';
 
 export function ClientClass({ asyncapi, params }) {
   const server = getServer(asyncapi.servers(), params.server);
@@ -14,12 +15,14 @@ export function ClientClass({ asyncapi, params }) {
   const serverUrl = getServerUrl(server);
   const operations = asyncapi.operations();
   const sendOperations = operations.filterBySend();
+  const receiveOperations = operations.filterByReceive();
+
   return (
     <Text>
       <Text newLines={2}>
         {`class ${clientName}:`}
       </Text>
-      <Constructor serverUrl={serverUrl} query={queryParams} />
+      <Constructor serverUrl={serverUrl} query={queryParams} receiveOperations={receiveOperations} />
       <Connect language="python" title={title} />
       <RegisterMessageHandler
         language="python"
@@ -41,6 +44,7 @@ export function ClientClass({ asyncapi, params }) {
         preExecutionCode='"""Pass the incoming message to all registered message handlers. """'
       />
       <HandleError />
+      <RegisterReceiveOperations receiveOperations={receiveOperations} />
       <SendOperations 
         language="python"
         sendOperations={sendOperations} 

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/components/Constructor.js

@@ -2,10 +2,11 @@ import { Text } from '@asyncapi/generator-react-sdk';
 import { QueryParamsVariables } from '@asyncapi/generator-components';
 import { QueryParamsArgumentsDocs } from './QueryParamsArgumentsDocs';
 import { InitSignature } from './InitSignature';
+import { ReceiveOperationsDiscriminators } from './ReceiveOperationsDiscriminators';
 
-export function Constructor({ serverUrl, query}) {
+export function Constructor({ serverUrl, query, receiveOperations }) {
   const queryParamsArray = query && Array.from(query.entries());
-  
+
   return (
     <>
       <InitSignature queryParams={queryParamsArray} serverUrl={serverUrl} />
@@ -27,10 +28,10 @@ export function Constructor({ serverUrl, query}) {
       self.error_handlers = []        # Callables for errors
       self.outgoing_processors = []   # Callables to process outgoing messages
       self._stop_event = threading.Event()
-      
       ${ query ? 'params = {}' : ''}`
         }
       </Text>
+      <ReceiveOperationsDiscriminators receiveOperations={receiveOperations} />
       <QueryParamsVariables
         language="python"
         queryParams={queryParamsArray} 

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/components/ReceiveOperationsDiscriminators.js

@@ -0,0 +1,28 @@
+import { Text } from '@asyncapi/generator-react-sdk';
+import { getMessageDiscriminatorsFromOperations } from '@asyncapi/generator-helpers';
+
+/**
+ * Component that generates initialization code for receive operation discriminators.
+ * 
+ * @param {Object} props - Component properties
+ * @param {Array<Object>} [props.receiveOperations] - Receive operations from AsyncAPI document
+ * @returns {React.Element|null} Rendered initialization code or null.
+ */
+export function ReceiveOperationsDiscriminators({ receiveOperations }) {
+  const hasOperations = Array.isArray(receiveOperations) && receiveOperations.length > 0;
+  if (!hasOperations) {
+    return null;
+  }
+
+  const operationDiscriminators = getMessageDiscriminatorsFromOperations(receiveOperations);
+  const serializedDiscriminators = JSON.stringify(operationDiscriminators);
+
+  return (
+    <Text indent={2} newLines={2}>
+      {`
+      self.receive_operation_handlers = {}
+      self.receive_operation_discriminators = ${serializedDiscriminators}`
+      }
+    </Text>
+  );
+}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/components/RegisterReceiveOperations.js

@@ -0,0 +1,65 @@
+import { Text } from '@asyncapi/generator-react-sdk';
+import { toSnakeCase } from '@asyncapi/generator-helpers';
+
+/**
+ * Component for rendering receive operation handler registration methods
+ * Generates one specific method per operation for better DX
+ * 
+* @param {Object} props - Component props
+* @param {Array} props.receiveOperations - Array of receive operations from AsyncAPI document
+* @returns {Array<React.Element>|null} Array of Text components containing Python handler registration methods, or null if no operations
+ */
+export function RegisterReceiveOperations({ receiveOperations }) {
+  if (!receiveOperations || receiveOperations.length === 0) {
+    return null;
+  }
+
+  return receiveOperations.map((operation) => {
+    const operationId = operation.id();
+    const methodName = `register_${toSnakeCase(operationId)}_handler`;
+
+    return (
+      <Text indent={2} newLines={2} key={operationId}>
+        {`def ${methodName}(self, handler, discriminator_key=None, discriminator_value=None):
+    """
+    Register a handler for ${operationId} operation.
+
+    Args:
+        handler (callable): Handler function that receives raw_message as argument
+        discriminator_key (str): Message field to use for routing (e.g., 'type', 'event_type')
+        discriminator_value (str): Expected value for routing (e.g., 'hello', 'user_joined')
+    """
+    if not callable(handler):
+        print("Handler must be callable")
+        return
+
+    # Validate that either both discriminator_key and discriminator_value are provided or neither
+    if (discriminator_key is not None and discriminator_value is None) or (discriminator_key is None and discriminator_value is not None):
+        print("Error: Both discriminator_key and discriminator_value must be provided together")
+        return
+
+    # Register handler
+    self.receive_operation_handlers["${operationId}"] = handler
+
+    # Add discriminator entry to the list if both key and value are provided
+    if discriminator_key is not None and discriminator_value is not None:
+        discriminator_entry = {
+            "key": discriminator_key,
+            "value": discriminator_value,
+            "operation_id": "${operationId}"
+        }
+        
+        # Check if this discriminator already exists
+        exists = any(
+            d.get("key") == discriminator_key and 
+            d.get("value") == discriminator_value and 
+            d.get("operation_id") == "${operationId}" 
+            for d in self.receive_operation_discriminators
+        )
+
+        if not exists:
+          self.receive_operation_discriminators.append(discriminator_entry)`}
+      </Text>
+    );
+  });
+}

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/example-slack-with-routing.py

@@ -0,0 +1,41 @@
+import sys
+import os
+import time
+
+sys.path.append(os.path.join(os.path.dirname(__file__), 'test'))
+
+from temp.snapshotTestResult.client_slack.client import SlackWebsocketAPIClient
+
+
+def handle_hello_message(message):
+    print(f"[WebSocket][HELLO] Received hello message: {message}")
+
+
+def handle_event_message(message):
+    print(f"[WebSocket][EVENT] Received event message: {message}")
+
+
+def handle_disconnect_message(message):
+    print(f"[WebSocket][DISCONNECT] Received disconnect message: {message}")
+
+
+def handle_unrecognized_message(raw_message):
+    print(f"[WebSocket][UNRECOGNIZED] Received unrecognized message: {raw_message}")
+
+
+def main():
+    client = SlackWebsocketAPIClient()
+
+    client.register_on_hello_message_handler(handle_hello_message)
+    
+    client.register_on_event_handler(handle_event_message)
+    
+    client.register_on_disconnect_message_handler(handle_disconnect_message)
+
+    client.connect()
+
+    # Keep program alive for a while to allow message processing
+    time.sleep(1000)
+
+
+main()

package/lib/templates/bakedInTemplates/core-template-client-websocket-python/package.json

@@ -12,8 +12,8 @@
   "license": "Apache-2.0",
   "dependencies": {
     "@asyncapi/generator-react-sdk": "*",
-    "@asyncapi/generator-helpers": "1.0.1",
-    "@asyncapi/generator-components": "0.4.1"
+    "@asyncapi/generator-helpers": "1.1.0",
+    "@asyncapi/generator-components": "0.5.0"
   },
   "devDependencies": {
     "@asyncapi/parser": "^3.4.0",

package/lib/templates/config/loader.js

@@ -46,7 +46,7 @@ async function loadTemplateConfig(templateDir, templateParams) {
  */
 function loadDefaultValues(templateConfig, templateParams) {
     const parameters = templateConfig.parameters;
-    const defaultValues = Object.keys(parameters || {}).filter(key => parameters[key].default);
+    const defaultValues = Object.keys(parameters || {}).filter(key =>  Object.hasOwn(parameters[key], "default"));
 
     defaultValues.filter(dv => !Object.prototype.hasOwnProperty.call(templateParams, dv)).forEach(dv =>
         Object.defineProperty(templateParams, dv, {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@asyncapi/generator",
-  "version": "3.0.1",
+  "version": "3.1.1",
   "description": "The AsyncAPI generator. It can generate documentation, code, anything!",
   "main": "./lib/generator.js",
   "engines": {